Update app.py

app.py

@@ -1,35 +1,50 @@
-import streamlit as st
-from langchain_community.document_loaders import WebBaseLoader
-
-from chains import Chain
-from portfolio import Portfolio
-from utils import clean_text
-
-
-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:
-            loader = WebBaseLoader([url_input])
-            data = clean_text(loader.load().pop().page_content)
-            portfolio.load_portfolio()
-            jobs = llm.extract_jobs(data)
-            for job in jobs:
-                skills = job.get('skills', [])
-                links = portfolio.query_links(skills)
-                email = llm.write_mail(job, links)
-                st.code(email, language='markdown')
-        except Exception as e:
-            st.error(f"An Error Occurred: {e}")
-
-
-if __name__ == "__main__":
-    chain = Chain()
-    portfolio = Portfolio()
-    st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
-    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 components
+chain = Chain()
+portfolio = Portfolio()
+
+def create_streamlit_app():
+    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 posting URL:", value="https://jobs.nike.com/job/R-33460")
+    username = st.text_input("Enter your name:", value="Computer Engineering Student")
+    client_name = st.text_input("Enter the recipient name:", value="Hiring Manager")
+    
+    submit_button = st.button("Generate Email")
+
+    if submit_button:
+        with st.spinner("Processing... Please wait."):
+            try:
+                # Load job description from URL
+                loader = WebBaseLoader([url_input])
+                data = clean_text(loader.load().pop().page_content)
+
+                # Load portfolio data
+                portfolio.load_portfolio()
+
+                # Extract jobs from the job posting page
+                jobs = chain.extract_jobs(data)
+
+                if not jobs:
+                    st.error("No job postings were found on the provided URL.")
+                    return
+
+                for job in jobs:
+                    skills = job.get('skills', [])
+                    links = portfolio.query_links(skills)
+                    email = chain.write_mail(job, links, username=username, client_name=client_name)
+                    st.subheader(f"Generated Cold Email for {job.get('role', 'Unknown Role')}")
+                    st.code(email, language="markdown")
+
+            except Exception as e:
+                st.error(f"An Error Occurred: {e}")
+
+if __name__ == "__main__":
+    create_streamlit_app()