Update file_study_guide_panel.py

file_study_guide_panel.py

@@ -1,84 +1,93 @@
 import gradio as gr
-import os
 import json
-from file_utils import (
-    extract_text_from_file,
-    call_llm,
-    save_study_guide,
-    fetch_study_guides
-)
-
-def file_study_guide_dashboard(nickname):
-    uploaded_files = gr.State([])
-    file_sections = gr.State({})
-    file_texts = gr.State({})
-
-    with gr.Column(scale=4, elem_id="main-column") as guide_content:
-        nickname_input = nickname
-        file_upload = gr.File(label="📤 Upload File for Study Guide", file_types=[".pdf", ".txt", ".md", ".pptx"])
-        study_guide_output = gr.Textbox(label="📘 Generated Study Guide", lines=10, interactive=False)
-        save_guide_btn = gr.Button("📌 Save Study Guide")
-
-        saved_guides_dropdown = gr.Dropdown(label="📚 Your Saved Guides", choices=[], interactive=True)
-        saved_guide_display = gr.Textbox(label="📖 Selected Guide Content", lines=10, interactive=False)
-
-        def generate_study_guide(file, current_files, section_map, text_map):
-            if file is None:
-                return current_files, section_map, text_map, "No file uploaded."
+from file_utils import extract_text_from_file, call_llm
+
+def file_study_guide_dashboard(nickname_input):
+    with gr.Column() as panel:
+        file_input = gr.File(label="📤 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"])
+        section_output = gr.Column()
+        file_text_state = gr.State("")
 
+        def generate_study_guide(file):
+            if file is None:
+                return "No file uploaded."
             text = extract_text_from_file(file)
-            filename = os.path.basename(file.name)
-
-            prompt = f"Divide this document into 4–6 learning sections. For each section, include a bold title and 2–3 line explanation.\n\n{text}"
-            response = call_llm(prompt, system_message="You are an AI tutor creating study guides.")
-
-            updated_files = current_files + [filename] if filename not in current_files else current_files
-            section_map[filename] = response
-            text_map[filename] = text
-
-            return updated_files, section_map, text_map, response
-
-        def save_guide_to_db(user, uploaded_files, section_map):
-            if not uploaded_files or user.strip() == "":
-                return
-            filename = uploaded_files[-1]
-            guide = section_map.get(filename, "")
-            if guide:
-                save_study_guide(user=user, filename=filename, guide=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):
-                    return g.guide
-            return ""
-
-        file_upload.change(
-            fn=generate_study_guide,
-            inputs=[file_upload, uploaded_files, file_sections, file_texts],
-            outputs=[uploaded_files, file_sections, file_texts, study_guide_output]
-        )
-
-        save_guide_btn.click(
-            fn=save_guide_to_db,
-            inputs=[nickname_input, uploaded_files, file_sections],
-            outputs=[]
-        )
-
-        nickname_input.change(
-            fn=load_user_guides,
-            inputs=[nickname_input],
-            outputs=[saved_guides_dropdown]
-        )
-
-        saved_guides_dropdown.change(
-            fn=show_saved_guide,
-            inputs=[nickname_input, saved_guides_dropdown],
-            outputs=[saved_guide_display]
-        )
-
-    return guide_content
+
+            prompt = f"""
+You are an AI tutor. Divide the following content into 5–6 logical sections.
+For each section:
+- Provide a 2–3 line summary (overview)
+- Write a detailed explanation of the topic
+- Create 5 multiple choice questions (MCQs), each with:
+  - A question
+  - 4 options
+  - 1 correct answer
+Return as JSON in this format:
+[
+  {
+    "section_title": "...",
+    "overview": "...",
+    "explanation": "...",
+    "quiz": [
+      {"question": "...", "options": ["A", "B", "C", "D"], "answer": "B"},
+      ...
+    ]
+  },
+  ...
+]
+
+Text:
+{text}
+"""
+            response = call_llm(prompt)
+            try:
+                study_guide = json.loads(response)
+                return render_sections(study_guide)
+            except:
+                return gr.Textbox(value="❌ Error generating structured study guide. Please try with a clearer file.", interactive=False)
+
+        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_section(index, sec):
+            section_ui = gr.Column()
+
+            with section_ui:
+                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)
+                question_index = 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):
+                        correct = q['answer']
+                        return "✅ Correct!" if choice == correct else f"❌ Correct Answer: {correct}"
+
+                    option_radio.change(fn=grade, inputs=[option_radio], outputs=[quiz_output])
+                    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)
+
+            return section_ui
+
+        file_input.change(fn=generate_study_guide, inputs=[file_input], outputs=[section_output])
+
+    return panel