Update app.py

app.py

@@ -1,71 +1,60 @@
-import os
-from langchain_groq import ChatGroq
-from langchain_core.prompts import PromptTemplate
-from langchain_core.output_parsers import JsonOutputParser
-from langchain_core.exceptions import OutputParserException
-from dotenv import load_dotenv
+import streamlit as st
+from langchain_community.document_loaders import WebBaseLoader
+from chains import Chain
+from portfolio import Portfolio
+from utils import clean_text
 
-load_dotenv()
+# Initialize backend services
+chain = Chain()
+portfolio = Portfolio()
 
-class Chain:
-    def __init__(self):
-        self.llm = ChatGroq(
-            temperature=0, 
-            groq_api_key=os.getenv("GROQ_API_KEY"), 
-            model_name="llama-3.3-70b-versatile"
-        )
+def create_streamlit_app():
+    st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
+    st.title("📧 Cold Email Generator")
 
-    def extract_jobs(self, cleaned_text):
-        prompt_extract = PromptTemplate.from_template(
-            """
-            ### SCRAPED TEXT FROM WEBSITE:
-            {page_data}
-            ### INSTRUCTION:
-            The scraped text is from the career's page of a website.
-            Your job is to extract the job postings and return them in JSON format containing the following keys: `role`, `experience`, `skills`, and `description`.
-            Only return the valid JSON.
-            ### VALID JSON (NO PREAMBLE):
-            """
-        )
-        chain_extract = prompt_extract | self.llm
-        res = chain_extract.invoke(input={"page_data": cleaned_text})
-        
+    # Input fields
+    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")
+
+    if submit_button:
         try:
-            json_parser = JsonOutputParser()
-            res = json_parser.parse(res.content)
-        except OutputParserException:
-            raise OutputParserException("Context too big. Unable to parse jobs.")
-        
-        return res if isinstance(res, list) else [res]
+            # 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)
+
+            if not jobs:
+                st.warning("⚠️ No jobs found on the page. Please check the URL.")
+                return
+
+            # Display extracted job details
+            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()
 
-    def write_mail(self, job, links, username="Computer Engineering Student", client_name="Hiring Manager"):
-        prompt_email = PromptTemplate.from_template(
-            """
-            ### JOB DESCRIPTION:
-            {job_description}
+            # 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)
 
-            ### INSTRUCTION:
-            You are {username}, a dedicated third-year Computer Engineering student.
-            You actively engage in projects that enhance your technical skills and have good communication skills.
-            Your task is to write a cold email to {client_name} regarding the job mentioned above, showcasing your capabilities and how your academic pursuits align with their needs.
-            Highlight your interests in innovative solutions and user engagement, as well as your commitment to creating impactful projects.
-            Also, add the most relevant ones from the following links to showcase your portfolio: {link_list}
-            Maintain a professional tone and avoid providing a preamble.
-            ### EMAIL (NO PREAMBLE):
-            """
-        )
+                    # Generate cold email
+                    email = chain.write_mail(job, links, username, client_name)
+                    st.code(email, language="markdown")
 
-        chain_email = prompt_email | self.llm
-        
-        # ✅ Ensure `username` and `client_name` are correctly passed
-        res = chain_email.invoke({
-            "job_description": str(job), 
-            "link_list": links,
-            "username": username,
-            "client_name": client_name
-        })
+        except Exception as e:
+            st.error(f"❌ An Error Occurred: {str(e)}")
 
-        return res.content
 
 if __name__ == "__main__":
-    print(os.getenv("GROQ_API_KEY"))
+    create_streamlit_app()