Create app.py

app.py

@@ -0,0 +1,77 @@
+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/51326")
+    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)
+
+    # 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")
+
+    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 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("⚠️ 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)