Spaces:
Sleeping
Sleeping
File size: 3,240 Bytes
6eb7feb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import streamlit as st
import pandas as pd
import traceback
import backend
from lib.common.database_support import save_email_to_excel
import time
def render_gen_email():
try:
st.title("Generate Email")
excel_file = st.session_state.get("excel_file", "Error Not Found")
if excel_file == "Error Not Found":
raise Exception("Error Not Found")
if not st.session_state.get("user_id"):
st.warning("Please fill in your profile information first.")
return
split_screen_col1, split_screen_col2 = st.columns(2)
# Input fields
with split_screen_col1:
industry_col, recipient_col = st.columns(2)
with industry_col:
industry = st.text_input("Industry")
with recipient_col:
role = st.text_input("Role")
details_col1, details_col2 = st.columns(2)
with details_col1:
sender_name = st.text_input("Senders Name")
with details_col2:
rec_name = st.text_input("Receivers Name")
rec_designation = st.text_input("Receivers Designation")
rec_company_name = st.text_input("Receivers Company Name")
sender_details = {
"sender_name": sender_name,
}
rec_details = {
"rec_name": rec_name,
"rec_designation": rec_designation,
"rec_company_name": rec_company_name
}
tone = st.selectbox("Tone", ["Formal", "Casual", "Friendly", "Professional"])
context = st.text_area("Context")
is_generate_clicked = st.button("Generate Email")
with split_screen_col2:
if is_generate_clicked:
generated_email = backend.collect_context_for_email(industry, role, tone, context, sender_details,rec_details )
st.session_state.generated_email = generated_email
st.text_area("Generated Email", value=generated_email, height=300, key="editable_email")
st.session_state.is_generate_clicked = True
else:
st.text_area("Generated Email", height=300, disabled=True)
# Save to templates button
if st.button("Save to templates"):
if st.session_state.get("generated_email"):
edited_email = st.session_state.editable_email
timestamp = int(time.time())
new_template = {
"Name": f"Generated Email {timestamp}",
"Subject": "New Generated Email",
"Body": edited_email
}
st.session_state.new_template = new_template # Store the new template in session state
st.session_state.show_new_template = True
st.session_state.page = "Template Emails"
st.rerun()
else:
st.warning("Please generate an email first.")
st.markdown("---")
except Exception as err:
traceback.print_exc()
print(err)
|