Update app.py

app.py

@@ -4,7 +4,7 @@ import re
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoModelForSeq2SeqLM
 from groq import Groq
-import gradio as gr
+import streamlit as st
 from docxtpl import DocxTemplate
 from datetime import datetime
 
@@ -170,46 +170,29 @@ def generate_offer_letter(template_file, candidate_name, role, start_date, hours
     # Return the file object
     return open(docx_file_path, 'rb')
 
-# --- Gradio Interface ---
-demo = gr.Interface(
-    fn=analyze_resume,
-    inputs=[
-        gr.File(label="Upload Resume (PDF or TXT)"),
-        gr.File(label="Upload Job Description (TXT)"),
-    ],
-    outputs=[
-        gr.Textbox(label="Similarity Score"),
-        gr.Textbox(label="Communication Response"),
-        gr.Textbox(label="Sentiment Analysis"),
-        gr.Textbox(label="Resume Skills"),
-        gr.Textbox(label="Job Description Skills"),
-        gr.Textbox(label="Enhanced Resume", lines=20),
-        gr.Textbox(label="Enhanced Job Description", lines=10),
-    ],
-    title="Resume and Job Description Analyzer",
-    description="Upload your resume (PDF or TXT) and job description (TXT) to analyze their similarity, extract skills, and generate a communication response.",
-)
-
-offer_demo = gr.Interface(
-    fn=generate_offer_letter,
-    inputs=[
-        gr.File(label="Upload Offer Letter Template (DOCX)"),
-        gr.Textbox(label="Candidate Name"),
-        gr.Textbox(label="Role"),
-        gr.Textbox(label="Start Date (YYYY-MM-DD)"),  # Use Textbox for date
-        gr.Number(label="Hours per Week"),
-    ],
-    outputs=gr.File(label="Offer Letter"),  # Change to gr.File
-    title="Offer Letter Generator",
-    description="Upload an offer letter template and enter candidate information to generate an offer letter.",
-)
-
-# Combine the interfaces using a Tabbed interface
-demo = gr.TabbedInterface(
-    [demo, offer_demo],
-    ["Resume Analyzer", "Offer Letter Generator"],
-    title="HR Assistant",
-)
-
-if __name__ == '__main__':
-    demo.launch(share=True)
+# --- Streamlit Interface ---
+st.title("HR Assistant")
+
+tab1, tab2 = st.tabs(["Resume Analyzer", "Offer Letter Generator"])
+
+with tab1:
+    st.header("Resume and Job Description Analyzer")
+    resume_file = st.file_uploader("Upload Resume (PDF or TXT)", type=['pdf', 'txt'])
+    job_description_file = st.file_uploader("Upload Job Description (TXT)", type=['txt'])
+
+    if resume_file is not None and job_description_file is not None:
+        analysis_results = analyze_resume(resume_file, job_description_file)
+        for result in analysis_results:
+            st.markdown(result)
+
+with tab2:
+    st.header("Offer Letter Generator")
+    template_file = st.file_uploader("Upload Offer Letter Template (DOCX)", type=['docx'])
+    candidate_name = st.text_input("Candidate Name")
+    role = st.text_input("Role")
+    start_date = st.text_input("Start Date (YYYY-MM-DD)")
+    hours = st.number_input("Hours per Week")
+
+    if template_file is not None and candidate_name and role and start_date and hours:
+        offer_letter = generate_offer_letter(template_file, candidate_name, role, start_date, hours)
+        st.download_button("Download Offer Letter", offer_letter, file_name=f"{candidate_name}_offer_letter.docx")