Update file_study_guide_panel.py

file_study_guide_panel.py

@@ -1,16 +1,20 @@
 import gradio as gr
 import json
-from file_utils import extract_text_from_file, call_llm
+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()
+
     with gr.Column() as panel:
         file_input = gr.File(label="📤 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"])
-        section_output = gr.Column()
+        save_btn = gr.Button("💾 Save Guide")
+        guide_dropdown = gr.Dropdown(label="📚 Your Saved Guides", choices=[])
         file_text_state = gr.State("")
+        guide_state = gr.State([])
 
         def generate_study_guide(file):
             if file is None:
-                return "No file uploaded."
+                return gr.update(visible=True), "No file uploaded.", [], []
             text = extract_text_from_file(file)
 
             prompt = f"""
@@ -42,9 +46,11 @@ Text:
             response = call_llm(prompt)
             try:
                 study_guide = json.loads(response)
-                return render_sections(study_guide)
+                section_output.children = render_sections(study_guide)
+                return text, study_guide
             except:
-                return gr.Textbox(value="❌ Error generating structured study guide. Please try with a clearer file.", interactive=False)
+                section_output.children = [gr.Markdown("❌ Error generating structured study guide.")]
+                return text, []
 
         def render_sections(study_guide):
             sections_ui = []
@@ -63,31 +69,50 @@ Text:
 
                 gr.Markdown("### 🧠 Quiz")
                 quiz_output = gr.Textbox(label="Answer Feedback", interactive=False)
-                question_index = gr.State(0)
+                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")
 
-                    def grade(choice):
+                    def grade(choice, score):
                         correct = q['answer']
-                        return "✅ Correct!" if choice == correct else f"❌ Correct Answer: {correct}"
+                        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
 
-                    option_radio.change(fn=grade, inputs=[option_radio], outputs=[quiz_output])
+                    option_radio.change(fn=grade, inputs=[option_radio, score_state], outputs=[quiz_output, score_display, score_state])
                     return [question_box, option_radio]
 
-                # Display first question initially
                 quiz_elements = render_question(0)
 
-                nav_buttons = []
                 for i in range(5):
-                    btn = gr.Button(str(i + 1), size="sm")
-                    btn.click(fn=lambda idx=i: render_question(idx), inputs=[], outputs=quiz_elements)
-                    nav_buttons.append(btn)
+                    gr.Button(str(i + 1), size="sm").click(fn=lambda idx=i: render_question(idx), inputs=[], outputs=quiz_elements)
 
             return section_ui
 
-        file_input.change(fn=generate_study_guide, inputs=[file_input], outputs=[section_output])
+        def save_to_user(nickname, filename, guide):
+            if not guide or not filename:
+                return
+            save_study_guide(nickname, filename.name, json.dumps(guide))
+
+        def load_user_guides(user):
+            guides = fetch_study_guides(user)
+            return [f"{g.filename} ({g.timestamp})" for g in guides]
+
+        def show_saved_guide(user, label):
+            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.")]
+
+        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=[])
 
     return panel