Update file_module.py

file_module.py

@@ -28,7 +28,8 @@ def file_dashboard():
                         file_guide_list = gr.Radio(label="Your Study Guides", choices=[], interactive=True)
                         btn_file_view = gr.Button("📁 Files")
                 gr.Markdown("---")
-                gr.Markdown("""Ankush Bansal""", elem_id="sidebar-user")
+                gr.Markdown("""**Ankush Bansal**  
+ankban@wharton.upenn.edu""", elem_id="sidebar-user")
 
             with gr.Column(scale=4, elem_id="main-column") as main_content:
                 mode_view = gr.State("upload")
@@ -42,7 +43,7 @@ def file_dashboard():
 
                 chat_input = gr.Textbox(label="Ask something about this file")
                 generate_quiz_btn = gr.Button("🧠 Generate Quiz from File")
-                quiz_display = gr.Column(visible=False)
+                quiz_container = gr.Column(visible=False)
                 quiz_score = gr.Textbox(label="Your Score", interactive=False, visible=False)
 
         def store_file_and_generate_guide(file, current_files, section_map, text_map):
@@ -77,29 +78,37 @@ def file_dashboard():
             history_map[selected_file] = new_history
             return new_history, history_map
 
-        def generate_quiz(selected_file, text_map):
+        def generate_quiz_ui(selected_file, text_map):
             if not selected_file or selected_file not in text_map:
-                return gr.update(visible=True), [], {}, gr.update(visible=True, value="")
+                return gr.update(visible=True), gr.update(visible=False, value="")
+
             text = text_map[selected_file]
             prompt = f"Create 5 multiple-choice questions based on the content below.\nReturn as JSON list with 'question', 'options' (list), and 'answer'.\n\n{text}"
             response = call_llm(prompt)
+
             try:
                 questions = json.loads(response)
-                score_state = {"score": 0, "total": len(questions)}
-                with quiz_display:
-                    for i, q in enumerate(questions):
-                        q_text = gr.Markdown(f"**Q{i+1}. {q['question']}**")
-                        q_radio = gr.Radio(choices=q['options'], label="Select an answer")
-                        q_feedback = gr.Textbox(label="Result", interactive=False)
-                        def handle_answer(choice, ans=q['answer'], score_state=score_state):
-                            if choice == ans:
-                                score_state["score"] += 1
-                                return "✅ Correct!"
-                            return f"❌ Incorrect. Correct answer: {ans}"
-                        q_radio.change(handle_answer, inputs=[q_radio], outputs=[q_feedback])
-                return gr.update(visible=True), questions, {selected_file: questions}, gr.update(visible=True, value="Score: 0/5")
-            except:
-                return gr.update(visible=True), [], {}, gr.update(visible=True, value="Error generating quiz.")
+                quiz_elements = []
+                score = {"correct": 0, "total": len(questions)}
+
+                for i, q in enumerate(questions):
+                    question_md = gr.Markdown(f"**Q{i+1}. {q['question']}**")
+                    options = gr.Radio(choices=q['options'], label=f"Your Answer to Q{i+1}", interactive=True)
+                    feedback = gr.Textbox(label="Result", interactive=False)
+
+                    def evaluate(ans):
+                        def check(choice):
+                            correct = choice == ans
+                            return "✅ Correct!" if correct else f"❌ Incorrect. Correct answer: {ans}"
+                        return check
+
+                    options.change(fn=evaluate(q['answer']), inputs=[options], outputs=[feedback])
+                    quiz_elements.extend([question_md, options, feedback])
+
+                return gr.update(visible=True), gr.update(visible=True, value=f"Score: 0/{score['total']}")
+
+            except Exception as e:
+                return gr.update(visible=True), gr.update(visible=True, value="Error generating quiz.")
 
         file_upload.change(
             fn=store_file_and_generate_guide,
@@ -129,9 +138,9 @@ def file_dashboard():
         )
 
         generate_quiz_btn.click(
-            fn=generate_quiz,
+            fn=generate_quiz_ui,
             inputs=[file_guide_list, file_texts],
-            outputs=[quiz_display, quiz_display, quiz_data, quiz_score],
+            outputs=[quiz_container, quiz_score],
             show_progress="minimal"
         )