Spaces:
Build error
Build error
| 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) |