Update app.py

app.py

@@ -8,62 +8,90 @@ 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")
+# 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)
 
-    # 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")
+#                 # Debugging: Print raw email output
+#                 st.write(f"Generated Email: {email}")
 
-    submit_button = st.button("Generate Cold 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.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")
 
     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)
+            data = clean_text(loader.load().pop().page_content)
+            portfolio.load_portfolio()
+            jobs = llm.extract_jobs(data)
 
             if not jobs:
-                st.warning("⚠️ No jobs found on the page. Please check the URL.")
+                st.error("❌ No job postings found on the page.")
                 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", [])
+            for job in jobs:
+                skills = job.get('skills', [])
                 links = portfolio.query_links(skills)
+                email = llm.write_mail(job, links)
 
-                # 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.")
+                st.subheader(f"📩 Cold Email for {job.get('role', 'Job')}")
+                st.code(email, language='markdown')
 
         except Exception as e:
-            st.error(f"❌ An Error Occurred: {str(e)}")
+            st.error(f"An Error Occurred: {e}")
 
 
-if __name__ == "__main__":
 
+if __name__ == "__main__":
+    st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
     create_streamlit_app(chain, portfolio, clean_text)