Update app.py

app.py

@@ -8,32 +8,34 @@ from utils import clean_text
 chain = Chain()
 portfolio = Portfolio()
 
-def create_streamlit_app():
+def create_streamlit_app(llm, portfolio, clean_text):
     st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
-    st.title("📧 Cold Email Generator")
+    st.title("📧 Cold Mail Generator")
 
-    # Input fields
+    # 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("Extract Job Details")
+    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 = chain.extract_jobs(cleaned_data)
+            jobs = llm.extract_jobs(cleaned_data)
 
             if not jobs:
                 st.warning("⚠️ No jobs found on the page. Please check the URL.")
                 return
 
-            # Display extracted job details
+            # 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')}")
@@ -41,20 +43,29 @@ def create_streamlit_app():
                 st.write(f"**Description:** {job.get('description', 'No description available')}")
                 st.divider()
 
-            # Email generation button
-            if st.button("Generate Cold Email"):
-                for job in jobs:
-                    # Get relevant portfolio links based on job skills
-                    skills = job.get("skills", [])
-                    links = portfolio.query_links(skills)
+                # Get relevant portfolio links based on job skills
+                skills = job.get("skills", [])
+                links = portfolio.query_links(skills)
+
+                # Debugging: Print fetched data
+                st.write(f"**Debug:** Skills: {skills}")
+                st.write(f"**Debug:** Portfolio Links: {links}")
+
+                # Generate cold email
+                email = llm.write_mail(job, links, username, client_name)
+
+                # Debugging: Print raw email output
+                st.write(f"**Debug:** Generated Email: {email}")
 
-                    # Generate cold email
-                    email = chain.write_mail(job, links, username, client_name)
+                # 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)}")
 
 
 if __name__ == "__main__":
-    create_streamlit_app()
+    create_streamlit_app(chain, portfolio, clean_text)