Update app.py

app.py

@@ -4,24 +4,37 @@ import traceback
 from model import screen_resumes_backend
 
 
-def process(job_description, files, model_option, threshold):
-    try:
-        if not files:
-            return "❌ Upload CV files first", None, None
+def process(job_description, files, model_option, threshold, gmail, password):
 
-        if not job_description or len(job_description.strip()) < 20:
+    try:
+        if not job_description or len(job_description) < 20:
             return "❌ Job description too short", None, None
 
-        results, summary, _, zip_path = screen_resumes_backend(
-            job_description,
-            files,
-            model_option,
-            threshold
-        )
-
-        df = pd.DataFrame(results) if results else None
-
-        return summary, df, zip_path
+        # if email provided → use Gmail mode
+        if gmail and password:
+            results, report, zip_path = screen_resumes_backend(
+                job_description,
+                files=None,
+                model_name=model_option,
+                threshold=threshold,
+                gmail=gmail,
+                password=password
+            )
+        else:
+            # file upload mode
+            if not files:
+                return "❌ Upload files or provide email", None, None
+
+            results, report, zip_path = screen_resumes_backend(
+                job_description,
+                files,
+                model_option,
+                threshold
+            )
+
+        df = pd.DataFrame(results)
+
+        return "✅ Process Completed", df, zip_path
 
     except Exception as e:
         return f"❌ Error: {str(e)}\n{traceback.format_exc()}", None, None
@@ -30,42 +43,37 @@ def process(job_description, files, model_option, threshold):
 # ================= UI =================
 with gr.Blocks(title="AI Resume Screener") as demo:
 
-    gr.HTML("""
-    <h1 style="text-align:center; color:#2563eb;">
-    🧠 AI Resume Screening System
-    </h1>
-    """)
+    gr.Markdown("# 🧠 AI Resume Screening System")
 
-    job_desc = gr.Textbox(
-        label="Job Description",
-        lines=8,
-        placeholder="Enter job requirements..."
-    )
+    job_desc = gr.Textbox(label="Job Description", lines=6)
 
-    model_choice = gr.Dropdown(
+    model = gr.Dropdown(
         ["Fast (MiniLM)", "Balanced (Recommended)", "High Accuracy"],
-        value="Balanced (Recommended)",
-        label="Model"
+        value="Balanced (Recommended)"
     )
 
-    threshold = gr.Slider(0.4, 0.9, value=0.6, step=0.05, label="Threshold")
+    threshold = gr.Slider(0.4, 0.9, value=0.6)
 
-    file_input = gr.File(
-        label="Upload CVs (PDF/DOCX)",
-        file_count="multiple"
-    )
+    # 🔥 NEW: EMAIL SECTION
+    gr.Markdown("## 📧 Email Integration (Optional)")
+
+    gmail = gr.Textbox(label="Gmail Address")
+    password = gr.Textbox(label="App Password (NOT normal password)", type="password")
+
+    gr.Markdown("OR Upload Files Below 👇")
+
+    files = gr.File(file_count="multiple", label="Upload CVs (PDF/DOCX)")
 
     btn = gr.Button("Process")
 
     output_text = gr.Markdown()
     output_table = gr.DataFrame()
-    download_file = gr.File()
+    download_zip = gr.File()
 
     btn.click(
         process,
-        inputs=[job_desc, file_input, model_choice, threshold],
-        outputs=[output_text, output_table, download_file]
+        inputs=[job_desc, files, model, threshold, gmail, password],
+        outputs=[output_text, output_table, download_zip]
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()