Update file_study_guide_panel.py

file_study_guide_panel.py

@@ -3,15 +3,13 @@ import json
 from file_utils import extract_text_from_file, call_llm, save_study_guide, fetch_study_guides
 
 def file_study_guide_dashboard(nickname_input):
-    section_output = gr.Column(visible=True)
-
     with gr.Column() as panel:
         file_input = gr.File(label="📤 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"])
         save_btn = gr.Button("💾 Save Guide")
         guide_dropdown = gr.Dropdown(label="📚 Your Saved Guides", choices=[])
-
-        guide_state = gr.State([])  # Stores current study guide
-        filename_state = gr.State("")
+        section_output = gr.Column()
+        file_text_state = gr.State("")
+        guide_state = gr.State([])
 
         def generate_study_guide(file):
             if file is None:
@@ -53,58 +51,53 @@ Text:
                 return [gr.Markdown("❌ Failed to parse structured study guide.")], text, []
 
         def render_sections(guide_data):
-            components = []
-            for i, section in enumerate(guide_data):
-                components.append(render_section(i + 1, section))
-            return components
+            section_blocks = []
+            for i, sec in enumerate(guide_data):
+                section_blocks.append(render_section(i + 1, sec))
+            return section_blocks
 
         def render_section(index, sec):
-            section_block = gr.Column()
-
-            with section_block:
-                gr.Markdown(f"## 📘 Section {index}: {sec['section_title']}")
-                gr.Markdown(f"**Overview:** {sec['overview']}")
-                gr.Markdown(f"**Explanation:**\n\n{sec['explanation']}")
-                gr.Markdown("### 🧠 Quiz")
-
-                quiz_output = gr.Textbox(label="Answer Feedback", interactive=False)
-                score_display = gr.Textbox(label="Score", interactive=False, value="Score: 0/5")
-                score_state = gr.State(0)
-
-                question_area = gr.Column()
-                question_index = gr.State(0)
-
-                def render_question(idx):
-                    q = sec['quiz'][idx]
-                    q_box = gr.Markdown(f"**Q{idx+1}.** {q['question']}")
-                    q_radio = gr.Radio(q['options'], label="Choose an answer")
-
-                    def grade(choice, score):
-                        correct = q['answer']
-                        updated_score = score + 1 if choice == correct else score
-                        feedback = "✅ Correct!" if choice == correct else f"❌ Correct: {correct}"
-                        return feedback, f"Score: {updated_score}/5", updated_score
-
-                    q_radio.change(fn=grade, inputs=[q_radio, score_state], outputs=[quiz_output, score_display, score_state])
-                    return [q_box, q_radio]
-
-                # Load first question by default
-                question_area.children = render_question(0)
-
-                with gr.Row():
-                    for i in range(5):
-                        gr.Button(str(i + 1), size="sm").click(
-                            fn=lambda idx=i: render_question(idx),
-                            inputs=[],
-                            outputs=question_area
-                        )
-
-            return section_block
+            quiz_area = gr.Column()
+            nav_buttons = gr.Row()
+            section_ui = gr.Column(
+                children=[
+                    gr.Markdown(f"## 📘 Section {index}: {sec['section_title']}"),
+                    gr.Markdown(f"**Overview:** {sec['overview']}"),
+                    gr.Markdown(f"**Explanation:**\n\n{sec['explanation']}"),
+                    gr.Markdown("### 🧠 Quiz"),
+                    quiz_area,
+                    nav_buttons,
+                    gr.Textbox(label="Answer Feedback", interactive=False, elem_id=f"feedback_{index}"),
+                    gr.Textbox(label="Score", interactive=False, value="Score: 0/5", elem_id=f"score_{index}")
+                ]
+            )
+
+            def make_question_ui(q_idx):
+                q = sec['quiz'][q_idx]
+                question_box = gr.Markdown(f"**Q{q_idx+1}.** {q['question']}")
+                option_radio = gr.Radio(q['options'], label="Choose an answer")
+
+                def grade(choice):
+                    correct = q['answer']
+                    return "✅ Correct!" if choice == correct else f"❌ Correct: {correct}"
+
+                option_radio.change(fn=grade, inputs=[option_radio], outputs=[f"feedback_{index}"])
+                return [question_box, option_radio]
+
+            # Initialize quiz with question 1
+            quiz_area.children = make_question_ui(0)
+
+            for i in range(5):
+                q_button = gr.Button(str(i + 1), size="sm")
+                q_button.click(fn=lambda idx=i: make_question_ui(idx), inputs=[], outputs=quiz_area)
+                nav_buttons.children.append(q_button)
+
+            return section_ui
 
         def save_to_user(nickname, filename, guide):
             if not guide or not filename:
                 return
-            save_study_guide(nickname, filename, json.dumps(guide))
+            save_study_guide(nickname, filename.name, json.dumps(guide))
 
         def load_user_guides(user):
             guides = fetch_study_guides(user)
@@ -117,15 +110,9 @@ Text:
                     return render_sections(json.loads(g.guide))
             return [gr.Markdown("❌ Guide not found.")]
 
-        file_input.change(
-            fn=generate_study_guide,
-            inputs=[file_input],
-            outputs=[section_output, filename_state, guide_state]
-        )
-
-        save_btn.click(fn=save_to_user, inputs=[nickname_input, filename_state, guide_state], outputs=[])
-
+        file_input.change(fn=generate_study_guide, inputs=[file_input], outputs=[section_output, file_text_state, guide_state])
+        save_btn.click(fn=save_to_user, inputs=[nickname_input, file_input, guide_state], outputs=[])
         nickname_input.change(fn=load_user_guides, inputs=[nickname_input], outputs=[guide_dropdown])
         guide_dropdown.change(fn=show_saved_guide, inputs=[nickname_input, guide_dropdown], outputs=[section_output])
 
-    return panel
+    return panel