Update app.py

app.py

@@ -8,20 +8,20 @@ from utils import clean_text
 chain = Chain()
 portfolio = Portfolio()
 
-# Define available email styles
-EMAIL_STYLES = ["Formal", "Casual", "Persuasive"]
-
 def create_streamlit_app(llm, portfolio, clean_text):
     st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
     st.title("📧 Cold Mail Generator")
 
     # User Inputs
     url_input = st.text_input("Enter a Job URL:", value="https://atliq.keka.com/careers/jobdetails/70996")
-    username = st.text_input("Your Name:", value="John")
-    client_name = st.text_input("Recipient's Name:", value="Hiring Manager")
+    username = st.text_input("Your Name (Default: John)", value="John")
+    client_name = st.text_input("Recipient's Name (Default: Hiring Manager)", value="Hiring Manager")
+
+    # Dropdown to select number of emails to generate
+    num_emails = st.selectbox("Select Number of Emails to Generate", options=[1, 2, 3], index=0)
 
-    # Select number of email variations
-    num_emails = st.slider("How many email variations do you want?", 1, 3, 1)
+    # Dropdown to select email style
+    email_style = st.selectbox("Choose Email Style", options=["Formal", "Casual", "Persuasive"], index=0)
 
     submit_button = st.button("Generate Cold Emails")
 
@@ -59,17 +59,15 @@ def create_streamlit_app(llm, portfolio, clean_text):
                 skills_list = job.get("skills", [])
                 portfolio_links = [link for link in portfolio.query_links(skills_list) if link]
 
-                # Generate cold emails based on user selection
-                selected_styles = EMAIL_STYLES[:num_emails]
-                
-                for style in selected_styles:
-                    st.subheader(f"📩 {style} Email")
-                    email = llm.write_mail(job, portfolio_links, username, client_name, style)
-                    
+                # Generate multiple cold emails based on user selection
+                for i in range(num_emails):
+                    st.subheader(f"📩 Email {i+1} ({email_style} Style)")
+                    email = llm.write_mail(job, portfolio_links, username, client_name, email_style)
+
                     if email:
                         st.code(email, language="markdown")
                     else:
-                        st.error(f"⚠️ Failed to generate {style} email.")
+                        st.error("⚠️ Email generation failed.")
 
         except Exception as e:
             st.error(f"❌ An Error Occurred: {str(e)}")
@@ -77,77 +75,3 @@ def create_streamlit_app(llm, portfolio, clean_text):
 
 if __name__ == "__main__":
     create_streamlit_app(chain, portfolio, clean_text)
-
-
-# import streamlit as st
-# from langchain_community.document_loaders import WebBaseLoader
-# from chains import Chain
-# from portfolio import Portfolio
-# from utils import clean_text
-
-# # Initialize backend services
-# chain = Chain()
-# portfolio = Portfolio()
-
-# def create_streamlit_app(llm, portfolio, clean_text):
-#     st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
-#     st.title("📧 Cold Mail Generator")
-
-#     # User Inputs
-#     url_input = st.text_input("Enter a Job URL:", value="https://atliq.keka.com/careers/jobdetails/70996")
-#     username = st.text_input("Your Name (Default: John)", value="John")
-#     client_name = st.text_input("Recipient's Name (Default: Hiring Manager)", value="Hiring Manager")
-
-#     submit_button = st.button("Generate Cold Email")
-
-#     if submit_button:
-#         try:
-#             st.info("🔄 Extracting job details...")
-
-#             # Load and clean job description
-#             loader = WebBaseLoader([url_input])
-#             page_content = loader.load().pop().page_content
-#             cleaned_data = clean_text(page_content)
-
-#             # Extract job details
-#             jobs = llm.extract_jobs(cleaned_data)
-
-#             if not jobs:
-#                 st.warning("⚠️ No job postings found. Please check the URL.")
-#                 return
-
-#             # Process extracted jobs
-#             for idx, job in enumerate(jobs, start=1):
-#                 job_title = job.get("role", "Unknown Role")
-#                 experience = job.get("experience", "Not Specified")
-#                 skills = ", ".join(job.get("skills", [])) if job.get("skills") else "Not Specified"
-#                 job_description = job.get("description", "No description available.")
-
-#                 # Display job details
-#                 st.subheader(f"📌 Job {idx}: {job_title}")
-#                 st.write(f"**Experience Required:** {experience}")
-#                 st.write(f"**Skills:** {skills}")
-#                 st.write(f"**Description:** {job_description}")
-#                 st.divider()
-
-#                 # Fetch relevant portfolio links
-#                 skills_list = job.get("skills", [])
-#                 portfolio_links = [link for link in portfolio.query_links(skills_list) if link]
-
-#                 # Generate cold email
-#                 email = llm.write_mail(job, portfolio_links, username, client_name)
-
-#                 # Debugging: Show raw email output
-#                 st.write(f"**Generated Email:**")
-                
-#                 if email:
-#                     st.code(email, language="markdown")
-#                 else:
-#                     st.error("⚠️ Email generation failed.")
-
-#         except Exception as e:
-#             st.error(f"❌ An Error Occurred: {str(e)}")
-
-
-# if __name__ == "__main__":
-#     create_streamlit_app(chain, portfolio, clean_text)