Update file_study_guide_panel.py

file_study_guide_panel.py

@@ -3,18 +3,20 @@ 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()
+    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=[])
-        file_text_state = gr.State("")
-        guide_state = gr.State([])
+
+        guide_state = gr.State([])  # Stores current study guide
+        filename_state = gr.State("")
 
         def generate_study_guide(file):
             if file is None:
-                return gr.update(visible=True), "No file uploaded.", [], []
+                return [gr.Markdown("❌ No file uploaded.")], "", []
+
             text = extract_text_from_file(file)
 
             prompt = f"""
@@ -28,15 +30,15 @@ For each section:
   - 1 correct answer
 Return as JSON in this format:
 [
-  {
+  {{
     "section_title": "...",
     "overview": "...",
     "explanation": "...",
     "quiz": [
-      {"question": "...", "options": ["A", "B", "C", "D"], "answer": "B"},
+      {{"question": "...", "options": ["A", "B", "C", "D"], "answer": "B"}},
       ...
     ]
-  },
+  }},
   ...
 ]
 
@@ -45,58 +47,64 @@ Text:
 """
             response = call_llm(prompt)
             try:
-                study_guide = json.loads(response)
-                section_output.children = render_sections(study_guide)
-                return text, study_guide
+                guide = json.loads(response)
+                return render_sections(guide), text, guide
             except:
-                section_output.children = [gr.Markdown("❌ Error generating structured study guide.")]
-                return text, []
+                return [gr.Markdown("❌ Failed to parse structured study guide.")], text, []
 
-        def render_sections(study_guide):
-            sections_ui = []
-            for i, sec in enumerate(study_guide):
-                section_ui = render_section(i + 1, sec)
-                sections_ui.append(section_ui)
-            return sections_ui
+        def render_sections(guide_data):
+            components = []
+            for i, section in enumerate(guide_data):
+                components.append(render_section(i + 1, section))
+            return components
 
         def render_section(index, sec):
-            section_ui = gr.Column()
+            section_block = gr.Column()
 
-            with section_ui:
+            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)
 
-                def render_question(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")
+                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']
-                        new_score = score + 1 if choice == correct else score
-                        feedback = "✅ Correct!" if choice == correct else f"❌ Correct Answer: {correct}"
-                        return feedback, f"Score: {new_score}/5", new_score
+                        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
 
-                    option_radio.change(fn=grade, inputs=[option_radio, score_state], outputs=[quiz_output, score_display, score_state])
-                    return [question_box, option_radio]
+                    q_radio.change(fn=grade, inputs=[q_radio, score_state], outputs=[quiz_output, score_display, score_state])
+                    return [q_box, q_radio]
 
-                quiz_elements = render_question(0)
+                # Load first question by default
+                question_area.children = render_question(0)
 
-                for i in range(5):
-                    gr.Button(str(i + 1), size="sm").click(fn=lambda idx=i: render_question(idx), inputs=[], outputs=quiz_elements)
+                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_ui
+            return section_block
 
         def save_to_user(nickname, filename, guide):
             if not guide or not filename:
                 return
-            save_study_guide(nickname, filename.name, json.dumps(guide))
+            save_study_guide(nickname, filename, json.dumps(guide))
 
         def load_user_guides(user):
             guides = fetch_study_guides(user)
@@ -106,13 +114,18 @@ Text:
             guides = fetch_study_guides(user)
             for g in guides:
                 if label.startswith(g.filename):
-                    section_output.children = render_sections(json.loads(g.guide))
-                    return
-            section_output.children = [gr.Markdown("No content found.")]
+                    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=[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=[])
+        guide_dropdown.change(fn=show_saved_guide, inputs=[nickname_input, guide_dropdown], outputs=[section_output])
 
-    return panel
+    return panel