Update app.py

app.py

@@ -1,50 +1,71 @@
+import gradio as gr
+from modules.kyc_processor import extract_text_from_id
 from modules.profile_matcher import match_profiles
 from modules.interview_scheduler import suggest_slots
-from modules.kyc_processor import extract_text_from_id
 from modules.onboarding_trainer import generate_quiz
 from modules.performance_tracker import analyze_performance
 from modules.agent_motivation import personalized_nudge
-import gradio as gr
 
-def kyc_interface(uploaded_image):
+def run_kyc(image):
+    return extract_text_from_id(image)
+
+def run_profile_match(resume, job_role):
+    return match_profiles(resume, job_role)
+
+def run_schedule(availability):
+    return suggest_slots(eval(availability))  # input as a dict
+
+def run_quiz(topic):
+    return generate_quiz(topic)
+
+def run_performance(agent_data):
     try:
-        return extract_text_from_id(uploaded_image)
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-    # 1. Profile Matching
-    print("\n[1] Profile Matching:")
-    resume = "Experienced insurance agent with proven sales..."
-    job = "Insurance Sales Agent"
-    print("Match:", match_profiles(resume, job))
-
-    # 2. Interview Scheduling
-    print("\n[2] Interview Scheduling:")
-    slots = suggest_slots({'2025-06-20': ['10:00 AM', '3:00 PM']})
-    print("Suggested:", slots)
-
-    # 4. Onboarding Quiz
-    print("\n[4] AI Onboarding Quiz:")
-    quiz = generate_quiz("Insurance Policy")
-    print("Quiz:", quiz)
-
-    # 5. Performance Tracking
-    print("\n[5] Agent Performance Alerts:")
-    agents = [{'name': 'Ravi', 'sales': 3}, {'name': 'Priya', 'sales': 8}]
-    print("Alerts:", analyze_performance(agents))
-
-    # 6. Motivation
-    print("\n[6] Motivation Nudges:")
-    print(personalized_nudge("Ravi", "10 client meetings"))
-    
-def main():
-    gr.Interface(
-        fn=kyc_interface,
-        inputs=gr.Image(type="filepath", label="Upload ID Document"),
-        outputs="text",
-        title="KYC Document OCR",
-        description="Upload an image of an ID card for OCR-based text extraction."
-    ).launch()
-    
-if __name__ == "__main__":
-    main()
+        agents = eval(agent_data)
+        return "\n".join(analyze_performance(agents))
+    except:
+        return "Invalid data format. Use: [{'name': 'A', 'sales': 5}, ...]"
+
+def run_motivation(name, goal):
+    return personalized_nudge(name, goal)
+
+with gr.Blocks(title="Agent Recruitment & Management Suite") as demo:
+
+    with gr.Tab("KYC OCR"):
+        gr.Markdown("### Upload an ID Document for OCR")
+        image_input = gr.Image(type="filepath")
+        ocr_output = gr.Textbox(label="Extracted Text")
+        gr.Button("Extract").click(run_kyc, inputs=image_input, outputs=ocr_output)
+
+    with gr.Tab("Agent Profile Matcher"):
+        gr.Markdown("### Compare Resume with Job Role")
+        resume_input = gr.Textbox(label="Resume Text")
+        job_input = gr.Textbox(label="Job Role Description")
+        match_output = gr.Textbox(label="Match Result")
+        gr.Button("Check Match").click(run_profile_match, [resume_input, job_input], match_output)
+
+    with gr.Tab("Interview Scheduler"):
+        gr.Markdown("### Suggest Interview Slots")
+        schedule_input = gr.Textbox(label="Availability Dict (e.g. {'2025-06-21': ['10AM', '2PM']})")
+        schedule_output = gr.Textbox(label="Suggested Slots")
+        gr.Button("Suggest").click(run_schedule, schedule_input, schedule_output)
+
+    with gr.Tab("Training Quiz"):
+        gr.Markdown("### Generate Quiz for a Topic")
+        topic_input = gr.Textbox(label="Training Topic")
+        quiz_output = gr.JSON(label="Generated Quiz")
+        gr.Button("Generate Quiz").click(run_quiz, topic_input, quiz_output)
+
+    with gr.Tab("Performance Tracker"):
+        gr.Markdown("### Enter Agent Sales Data")
+        performance_input = gr.Textbox(label="Agent List (e.g. [{'name': 'A', 'sales': 3}])")
+        performance_output = gr.Textbox(label="Alerts")
+        gr.Button("Analyze").click(run_performance, performance_input, performance_output)
+
+    with gr.Tab("Motivational Nudge"):
+        gr.Markdown("### Send Motivation to Agent")
+        name_input = gr.Textbox(label="Agent Name")
+        goal_input = gr.Textbox(label="Goal (e.g. 10 meetings)")
+        nudge_output = gr.Textbox(label="Generated Nudge")
+        gr.Button("Motivate").click(run_motivation, [name_input, goal_input], nudge_output)
+
+demo.launch()