Update app.py

app.py

@@ -7,26 +7,26 @@ from agents import AnalyzerAgent, ForecastAgent, CoachAgent
 with open("questions.json", "r", encoding="utf-8") as f:
     QUESTION_BANK = json.load(f)
 
-# Agents
+# Agents (all powered by GLM-4.5 API)
 analyzer = AnalyzerAgent()
 forecast_agent = ForecastAgent()
 coach_agent = CoachAgent()
 
-# State
+# Start exam
 def start_exam(level, subject, num_questions=5):
-    # Filter questions
+    # Filter question bank
     questions = [q for q in QUESTION_BANK if q["subject"] == f"{level}_{subject}"]
     selected = random.sample(questions, min(num_questions, len(questions)))
 
-    # Keep only necessary fields for UI
+    # Format for UI
     exam_data = [
         {"id": q["id"], "text": q["text"], "choices": q.get("choices", [])}
         for q in selected
     ]
     return exam_data, gr.update(visible=True), gr.update(visible=False)
 
+# Submit exam
 def submit_exam(answers, exam_data, level, subject):
-    # Score
     correct = 0
     per_question = {}
     for q in exam_data:
@@ -39,13 +39,9 @@ def submit_exam(answers, exam_data, level, subject):
 
     score = round(100 * correct / len(exam_data), 2)
 
-    # Run Analyzer
-    analysis = analyzer.analyze(per_question, QUESTION_BANK)
-
-    # Call Forecast Agent (GLM-4.5)
+    # GLM-4.5 Agents
+    analysis = analyzer.analyze(per_question, exam_data)
     forecast = forecast_agent.forecast(level, subject)
-
-    # Call Coach Agent (GLM-4.5)
     coach = coach_agent.coach(analysis, level, subject)
 
     return (
@@ -59,45 +55,52 @@ def submit_exam(answers, exam_data, level, subject):
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 📘 Pay Fong Exam Simulator (AI-Powered)")
-    
+    gr.Markdown("## 📘 Pay Fong Exam Simulator (AI-Powered with GLM-4.5)")
+
     with gr.Row():
-        level = gr.Dropdown(["Form2", "Form3", "Higher1", "Higher2", "Higher3"], label="Select Level")
-        subject = gr.Dropdown(["Math", "Science", "Physics", "Chemistry", "Biology"], label="Select Subject")
+        level = gr.Dropdown(
+            ["Form2", "Form3", "Higher1", "Higher2", "Higher3"], label="Select Level"
+        )
+        subject = gr.Dropdown(
+            ["Math", "Science", "Physics", "Chemistry", "Biology"], label="Select Subject"
+        )
         start_btn = gr.Button("Start Exam")
-    
+
     exam_output = gr.State()
-    answers_state = gr.State()
 
     exam_area = gr.Column(visible=False)
     with exam_area:
         gr.Markdown("### Exam Questions")
         exam_display = gr.JSON(label="Questions")
-        answers_box = gr.JSON(label="Your Answers (edit as JSON: {id: 'answer'})")
+        answers_box = gr.JSON(
+            label="Your Answers (edit as JSON: {id: 'answer'})"
+        )
         submit_btn = gr.Button("Submit Exam")
 
     results_area = gr.Column(visible=False)
     with results_area:
         score_text = gr.Textbox(label="Score")
-        analysis_json = gr.JSON(label="Weakness Analysis")
-        forecast_json = gr.JSON(label="Forecast (AI)")
-        coach_json = gr.JSON(label="Study Coach")
+        with gr.Tab("Weakness Analysis"):
+            analysis_json = gr.JSON()
+        with gr.Tab("Forecast (AI)"):
+            forecast_json = gr.JSON()
+        with gr.Tab("Study Coach"):
+            coach_json = gr.JSON()
 
     # Button actions
     start_btn.click(
         start_exam,
         inputs=[level, subject],
-        outputs=[exam_display, exam_area, results_area]
-    ).then(
-        lambda exam: exam, outputs=exam_output
-    )
+        outputs=[exam_display, exam_area, results_area],
+    ).then(lambda exam: exam, outputs=exam_output)
 
     submit_btn.click(
         submit_exam,
         inputs=[answers_box, exam_output, level, subject],
-        outputs=[score_text, analysis_json, forecast_json, coach_json, exam_area, results_area]
+        outputs=[score_text, analysis_json, forecast_json, coach_json, exam_area, results_area],
     )
 
 if __name__ == "__main__":
     demo.launch()
 
+