code = """import streamlit as st
import sys
import os
import pandas as pd
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from ats.skill_extractor import extract_text_from_pdf, extract_skills
from ats.ats_score import calculate_ats_score
from cover_letter.generator import generate_cover_letter
from roadmap.roadmap_generator import generate_roadmap, generate_interview_questions
from agents.resume_agent import resume_agent
from agents.router_agent import router_agent
from rag.qdrant_store import create_collection, store_document
from rag.retriever import ask_with_rag
from cold_email.sender import load_recruiters, generate_email_body, send_cold_emails

st.set_page_config(page_title="AI Recruitment Copilot", page_icon="🤖", layout="wide")

st.sidebar.title("🤖 AI Recruitment Copilot")
page = st.sidebar.radio("Navigate", [
    "🏠 Dashboard",
    "📄 Resume Analysis",
    "✉️ Cover Letter",
    "❓ Interview Questions",
    "🗺️ Learning Roadmap",
    "📧 Cold Email Sender",
    "💬 AI Chatbot"
])

if "resume_text" not in st.session_state:
    st.session_state.resume_text = ""
if "jd_text" not in st.session_state:
    st.session_state.jd_text = ""
if "ats_result" not in st.session_state:
    st.session_state.ats_result = None
if "resume_skills" not in st.session_state:
    st.session_state.resume_skills = []

st.sidebar.markdown("---")
st.sidebar.subheader("Upload Files")
resume_file = st.sidebar.file_uploader("Upload Resume PDF", type=["pdf"])

if resume_file:
    with open("data/uploads/resume.pdf", "wb") as f:
        f.write(resume_file.read())
    st.session_state.resume_text = extract_text_from_pdf("data/uploads/resume.pdf")
    st.sidebar.success("Resume uploaded!")

st.sidebar.markdown("---")
st.sidebar.subheader("Job Description")
jd_input = st.sidebar.text_area("Paste Job Description here", height=200)
if st.sidebar.button("Save JD"):
    if jd_input.strip():
        st.session_state.jd_text = jd_input.strip()
        st.sidebar.success("JD saved!")
    else:
        st.sidebar.error("Please paste a job description first.")

if page == "🏠 Dashboard":
    st.title("🤖 AI Recruitment Copilot")
    st.markdown("### Your AI-powered job application assistant")
    st.markdown("---")
    if st.session_state.resume_text and st.session_state.jd_text:
        if st.button("Run Full ATS Analysis"):
            with st.spinner("Analyzing..."):
                resume_skills = extract_skills(st.session_state.resume_text, "resume")
                jd_skills = extract_skills(st.session_state.jd_text, "job description")
                result = calculate_ats_score(
                    resume_skills,
                    jd_skills,
                    resume_text=st.session_state.resume_text,
                    jd_text=st.session_state.jd_text
                )
                st.session_state.ats_result = result
                st.session_state.resume_skills = resume_skills

        if st.session_state.ats_result:
            result = st.session_state.ats_result
            st.markdown("### 📊 ATS Analysis Results")
            col1, col2, col3, col4, col5 = st.columns(5)
            col1.metric("🎯 Final Score", f"{result['ats_score']}%")
            col2.metric("🔑 Keywords", f"{result['keyword_score']}%")
            col3.metric("🧠 Semantic", f"{result['semantic_score']}%")
            col4.metric("💼 Experience", f"{result['experience_score']}%")
            col5.metric("🎓 Education", f"{result['education_score']}%")
            st.markdown("---")
            st.subheader("💼 Experience Analysis")
            st.info(result['experience_msg'])
            st.subheader("🎓 Education Analysis")
            st.info(result['education_msg'])
            st.subheader("📄 Resume Format Check")
            for issue in result['format_issues']:
                if "✅" in issue:
                    st.success(issue)
                else:
                    st.error(issue)
            if result['format_suggestions']:
                st.subheader("💡 Suggestions")
                for suggestion in result['format_suggestions']:
                    st.warning(suggestion)
            st.markdown("---")
            col4, col5 = st.columns(2)
            with col4:
                st.subheader("✅ Matched Skills")
                for skill in result['matched_skills']:
                    st.success(skill)
            with col5:
                st.subheader("❌ Missing Skills")
                for skill in result['missing_skills']:
                    st.error(skill)
    else:
        st.info("Upload your Resume and paste Job Description from the sidebar to get started!")

elif page == "📄 Resume Analysis":
    st.title("Resume Analysis")
    st.markdown("---")
    if st.session_state.resume_text:
        if st.button("Analyze My Resume"):
            with st.spinner("Analyzing resume..."):
                analysis = resume_agent(st.session_state.resume_text)
            st.markdown(analysis)
    else:
        st.warning("Please upload your resume from the sidebar first.")

elif page == "✉️ Cover Letter":
    st.title("Cover Letter Generator")
    st.markdown("---")
    if st.session_state.resume_text and st.session_state.jd_text:
        if st.button("Generate Cover Letter"):
            with st.spinner("Writing cover letter..."):
                letter = generate_cover_letter(
                    st.session_state.resume_text,
                    st.session_state.jd_text
                )
            st.text_area("Your Cover Letter", letter, height=400)
            st.download_button("Download", letter, file_name="cover_letter.txt")
    else:
        st.warning("Please upload Resume and paste JD from the sidebar first.")

elif page == "❓ Interview Questions":
    st.title("Interview Question Generator")
    st.markdown("---")
    if st.session_state.resume_text and st.session_state.jd_text:
        if st.button("Generate Questions"):
            with st.spinner("Generating questions..."):
                questions = generate_interview_questions(
                    st.session_state.resume_text,
                    st.session_state.jd_text
                )
            st.markdown(questions)
    else:
        st.warning("Please upload Resume and paste JD from the sidebar first.")

elif page == "🗺️ Learning Roadmap":
    st.title("Learning Roadmap Generator")
    st.markdown("---")
    if st.session_state.ats_result:
        missing = st.session_state.ats_result['missing_skills']
        st.subheader("Missing Skills")
        st.write(missing)
        if st.button("Generate Roadmap"):
            with st.spinner("Creating your roadmap..."):
                roadmap = generate_roadmap(missing)
            st.markdown(roadmap)
    else:
        st.warning("Please run ATS Analysis from the Dashboard first.")

elif page == "📧 Cold Email Sender":
    st.title("📧 Cold Email Sender")
    st.markdown("### Send personalized emails to multiple HRs automatically")
    st.markdown("---")

    col1, col2 = st.columns(2)
    with col1:
        sender_email = st.text_input("Your Gmail", value="nehab3099@gmail.com")
    with col2:
        app_password = st.text_input("Gmail App Password", type="password")

    excel_file = st.file_uploader("Upload HR Excel File", type=["xlsx"])
    delay = st.slider("Delay between emails (seconds)", min_value=10, max_value=60, value=30)

    if excel_file:
        with open("data/uploads/recruiters.xlsx", "wb") as f:
            f.write(excel_file.read())

        df = load_recruiters("data/uploads/recruiters.xlsx")
        st.success(f"Found {len(df)} valid HR emails!")
        st.dataframe(df[["Company Name", "HR / Contact Person", "Email ID"]])

        st.markdown("---")
        st.subheader("📝 Email Preview")
        if len(df) > 0:
            sample_row = df.iloc[0]
            preview = generate_email_body(
                sample_row["HR / Contact Person"],
                sample_row["Company Name"],
                st.session_state.resume_skills
            )
            st.text_area("Sample Email", preview, height=300)

        st.markdown("---")
        if not st.session_state.resume_text:
            st.warning("Please upload your resume from the sidebar first!")
        elif not app_password:
            st.warning("Please enter your Gmail App Password!")
        else:
            if st.button(f"🚀 Send Emails to All {len(df)} HRs"):
                with st.spinner(f"Sending emails... This will take {len(df) * delay} seconds"):
                    results, total = send_cold_emails(
                        "data/uploads/recruiters.xlsx",
                        "data/uploads/resume.pdf",
                        sender_email,
                        app_password,
                        st.session_state.resume_skills,
                        delay
                    )

                st.success(f"Done! Processed {total} emails.")
                st.markdown("### 📊 Results")
                for r in results:
                    if "✅" in r['status']:
                        st.success(f"{r['status']} → {r['hr_name']} | {r['company']} | {r['email']}")
                    else:
                        st.error(f"{r['status']} → {r['email']}")

elif page == "💬 AI Chatbot":
    st.title("AI Career Chatbot")
    st.markdown("---")
    create_collection()
    if st.session_state.resume_text:
        store_document(st.session_state.resume_text, {"type": "resume"})
    if st.session_state.jd_text:
        store_document(st.session_state.jd_text, {"type": "jd"})
    user_query = st.text_input("Ask anything about your resume or job...")
    if st.button("Ask"):
        if user_query:
            with st.spinner("Thinking..."):
                answer = router_agent(
                    user_query,
                    resume_text=st.session_state.resume_text,
                    jd_text=st.session_state.jd_text,
                    missing_skills=st.session_state.ats_result['missing_skills'] if st.session_state.ats_result else []
                )
            st.markdown(answer)
"""

with open("streamlit_app/app.py", "w", encoding="utf-8") as f:
    f.write(code)

print("app.py updated with Cold Email Sender!")


# code = """import streamlit as st
# import sys
# import os
# sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

# from ats.skill_extractor import extract_text_from_pdf, extract_skills
# from ats.ats_score import calculate_ats_score
# from cover_letter.generator import generate_cover_letter
# from roadmap.roadmap_generator import generate_roadmap, generate_interview_questions
# from agents.resume_agent import resume_agent
# from agents.router_agent import router_agent
# from rag.qdrant_store import create_collection, store_document
# from rag.retriever import ask_with_rag

# st.set_page_config(page_title="AI Recruitment Copilot", page_icon="🤖", layout="wide")

# st.sidebar.title("🤖 AI Recruitment Copilot")
# page = st.sidebar.radio("Navigate", [
#     "🏠 Dashboard",
#     "📄 Resume Analysis",
#     "✉️ Cover Letter",
#     "❓ Interview Questions",
#     "🗺️ Learning Roadmap",
#     "💬 AI Chatbot"
# ])

# if "resume_text" not in st.session_state:
#     st.session_state.resume_text = ""
# if "jd_text" not in st.session_state:
#     st.session_state.jd_text = ""
# if "ats_result" not in st.session_state:
#     st.session_state.ats_result = None

# st.sidebar.markdown("---")
# st.sidebar.subheader("Upload Files")
# resume_file = st.sidebar.file_uploader("Upload Resume PDF", type=["pdf"])

# if resume_file:
#     with open("data/uploads/resume.pdf", "wb") as f:
#         f.write(resume_file.read())
#     st.session_state.resume_text = extract_text_from_pdf("data/uploads/resume.pdf")
#     st.sidebar.success("Resume uploaded!")

# st.sidebar.markdown("---")
# st.sidebar.subheader("Job Description")
# jd_input = st.sidebar.text_area("Paste Job Description here", height=200)
# if st.sidebar.button("Save JD"):
#     if jd_input.strip():
#         st.session_state.jd_text = jd_input.strip()
#         st.sidebar.success("JD saved!")
#     else:
#         st.sidebar.error("Please paste a job description first.")

# if page == "🏠 Dashboard":
#     st.title("🤖 AI Recruitment Copilot")
#     st.markdown("### Your AI-powered job application assistant")
#     st.markdown("---")
#     if st.session_state.resume_text and st.session_state.jd_text:
#         if st.button("Run Full ATS Analysis"):
#             with st.spinner("Analyzing..."):
#                 resume_skills = extract_skills(st.session_state.resume_text, "resume")
#                 jd_skills = extract_skills(st.session_state.jd_text, "job description")
#                 result = calculate_ats_score(
#                     resume_skills,
#                     jd_skills,
#                     resume_text=st.session_state.resume_text,
#                     jd_text=st.session_state.jd_text
#                 )
#                 st.session_state.ats_result = result

#         if st.session_state.ats_result:
#             result = st.session_state.ats_result
#             st.markdown("### 📊 ATS Analysis Results")

#             col1, col2, col3, col4, col5 = st.columns(5)
#             col1.metric("🎯 Final Score", f"{result['ats_score']}%")
#             col2.metric("🔑 Keywords", f"{result['keyword_score']}%")
#             col3.metric("🧠 Semantic", f"{result['semantic_score']}%")
#             col4.metric("💼 Experience", f"{result['experience_score']}%")
#             col5.metric("🎓 Education", f"{result['education_score']}%")

#             st.markdown("---")
#             st.subheader("💼 Experience Analysis")
#             st.info(result['experience_msg'])

#             st.subheader("🎓 Education Analysis")
#             st.info(result['education_msg'])

#             st.subheader("📄 Resume Format Check")
#             for issue in result['format_issues']:
#                 if "✅" in issue:
#                     st.success(issue)
#                 else:
#                     st.error(issue)
#             if result['format_suggestions']:
#                 st.subheader("💡 Suggestions")
#                 for suggestion in result['format_suggestions']:
#                     st.warning(suggestion)

#             st.markdown("---")
#             col4, col5 = st.columns(2)
#             with col4:
#                 st.subheader("✅ Matched Skills")
#                 for skill in result['matched_skills']:
#                     st.success(skill)
#             with col5:
#                 st.subheader("❌ Missing Skills")
#                 for skill in result['missing_skills']:
#                     st.error(skill)
#     else:
#         st.info("Upload your Resume and paste Job Description from the sidebar to get started!")

# elif page == "📄 Resume Analysis":
#     st.title("Resume Analysis")
#     st.markdown("---")
#     if st.session_state.resume_text:
#         if st.button("Analyze My Resume"):
#             with st.spinner("Analyzing resume..."):
#                 analysis = resume_agent(st.session_state.resume_text)
#             st.markdown(analysis)
#     else:
#         st.warning("Please upload your resume from the sidebar first.")

# elif page == "✉️ Cover Letter":
#     st.title("Cover Letter Generator")
#     st.markdown("---")
#     if st.session_state.resume_text and st.session_state.jd_text:
#         if st.button("Generate Cover Letter"):
#             with st.spinner("Writing cover letter..."):
#                 letter = generate_cover_letter(
#                     st.session_state.resume_text,
#                     st.session_state.jd_text
#                 )
#             st.text_area("Your Cover Letter", letter, height=400)
#             st.download_button("Download", letter, file_name="cover_letter.txt")
#     else:
#         st.warning("Please upload Resume and paste JD from the sidebar first.")

# elif page == "❓ Interview Questions":
#     st.title("Interview Question Generator")
#     st.markdown("---")
#     if st.session_state.resume_text and st.session_state.jd_text:
#         if st.button("Generate Questions"):
#             with st.spinner("Generating questions..."):
#                 questions = generate_interview_questions(
#                     st.session_state.resume_text,
#                     st.session_state.jd_text
#                 )
#             st.markdown(questions)
#     else:
#         st.warning("Please upload Resume and paste JD from the sidebar first.")

# elif page == "🗺️ Learning Roadmap":
#     st.title("Learning Roadmap Generator")
#     st.markdown("---")
#     if st.session_state.ats_result:
#         missing = st.session_state.ats_result['missing_skills']
#         st.subheader("Missing Skills")
#         st.write(missing)
#         if st.button("Generate Roadmap"):
#             with st.spinner("Creating your roadmap..."):
#                 roadmap = generate_roadmap(missing)
#             st.markdown(roadmap)
#     else:
#         st.warning("Please run ATS Analysis from the Dashboard first.")

# elif page == "💬 AI Chatbot":
#     st.title("AI Career Chatbot")
#     st.markdown("---")
#     create_collection()
#     if st.session_state.resume_text:
#         store_document(st.session_state.resume_text, {"type": "resume"})
#     if st.session_state.jd_text:
#         store_document(st.session_state.jd_text, {"type": "jd"})
#     user_query = st.text_input("Ask anything about your resume or job...")
#     if st.button("Ask"):
#         if user_query:
#             with st.spinner("Thinking..."):
#                 answer = router_agent(
#                     user_query,
#                     resume_text=st.session_state.resume_text,
#                     jd_text=st.session_state.jd_text,
#                     missing_skills=st.session_state.ats_result['missing_skills'] if st.session_state.ats_result else []
#                 )
#             st.markdown(answer)
# """

# with open("streamlit_app/app.py", "w", encoding="utf-8") as f:
#     f.write(code)

# print("app.py updated successfully!")