Update app.py

app.py

@@ -8,90 +8,65 @@ from utils import clean_text
 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 Input
-#     url_input = st.text_input("Enter a Job URL:", value="https://jobs.nike.com/job/R-33460")
-#     username = st.text_input("Your Name (Default: Computer Engineering Student)", value="Computer Engineering Student")
-#     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 jobs found on the page. Please check the URL.")
-#                 return
-
-#             # Process extracted jobs
-#             for idx, job in enumerate(jobs, start=1):
-#                 st.subheader(f"Job {idx}: {job.get('role', 'Unknown Role')}")
-#                 st.write(f"**Experience Required:** {job.get('experience', 'Not Specified')}")
-#                 st.write(f"**Skills:** {', '.join(job.get('skills', [])) if job.get('skills') else 'Not Specified'}")
-#                 st.write(f"**Description:** {job.get('description', 'No description available')}")
-#                 st.divider()
-
-#                 # Get relevant portfolio links based on job skills
-#                 skills = job.get("skills", [])
-#                 links = portfolio.query_links(skills)
-
-#                 # Generate cold email
-#                 email = llm.write_mail(job, links, username, client_name)
-
-#                 # Debugging: Print raw email output
-#                 st.write(f"Generated Email: {email}")
-
-#                 # Display email
-#                 if email:
-#                     st.code(email, language="markdown")
-#                 else:
-#                     st.error("⚠️ Failed to generate an email.")
-
-#         except Exception as e:
-#             st.error(f"❌ An Error Occurred: {str(e)}")
-
 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")
-    url_input = st.text_input("Enter a URL:", value="https://jobs.nike.com/job/R-33460")
-    submit_button = st.button("Submit")
+
+    # User Inputs
+    url_input = st.text_input("Enter a Job URL:", value="https://jobs.nike.com/job/R-33460")
+    username = st.text_input("Your Name (Default: Computer Engineering Student)", value="Computer Engineering Student")
+    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])
-            data = clean_text(loader.load().pop().page_content)
-            portfolio.load_portfolio()
-            jobs = llm.extract_jobs(data)
+            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.error("❌ No job postings found on the page.")
+                st.warning("⚠️ No job postings found. Please check the URL.")
                 return
 
-            for job in jobs:
-                skills = job.get('skills', [])
-                links = portfolio.query_links(skills)
-                email = llm.write_mail(job, links)
-
-                st.subheader(f"📩 Cold Email for {job.get('role', 'Job')}")
-                st.code(email, language='markdown')
+            # 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: {e}")
-
+            st.error(f"❌ An Error Occurred: {str(e)}")
 
 
 if __name__ == "__main__":
-    st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
     create_streamlit_app(chain, portfolio, clean_text)
-