Update file_module.py

file_module.py

@@ -9,10 +9,13 @@ from file_utils import (
 # === FILE-BASED LEARNING LAYOUT WITH NAVIGATION ===
 def file_dashboard():
     with gr.Blocks(elem_id="file-learning-layout") as file_panel:
+        uploaded_files = gr.State([])
+        file_sections = gr.State({})
+
         with gr.Row():
             # === LEFT SIDEBAR ===
             with gr.Column(scale=1, min_width=220):
-                gr.Markdown("""### 🎓 Educare"", elem_id="sidebar-title")
+                gr.Markdown("### 🎓 Educare", elem_id="sidebar-title")
                 gr.Markdown("---")
 
                 with gr.Accordion("🧭 Learning Modes", open=True):
@@ -21,33 +24,69 @@ def file_dashboard():
 
                     with gr.Accordion("📄 File-Based Learning", open=True):
                         btn_study_guides = gr.Button("📘 Study Guides")
+                        file_guide_list = gr.Radio(label="Your Study Guides", choices=[], interactive=True)
                         btn_file_view = gr.Button("📁 Files")
 
                 gr.Markdown("---")
                 gr.Markdown("""**Ankush Bansal**  
 ankban@wharton.upenn.edu""", elem_id="sidebar-user")
 
-            # === MAIN CONTENT AREA ===
+            # === CENTER + RIGHT PANEL ===
             with gr.Column(scale=4, elem_id="main-column") as main_content:
+                mode_view = gr.State("upload")
                 file_upload = gr.File(label="📤 Upload File", file_types=[".pdf", ".txt", ".md", ".pptx"])
-                content_output = gr.Markdown("Upload a file to get started.", elem_id="file-main-area")
+                study_guide_output = gr.Markdown("Upload a file to generate a study guide.")
+                file_chat_output = gr.Row(visible=False)
+
+                with file_chat_output:
+                    file_preview = gr.Textbox(label="📄 File Viewer (Simulated)", lines=15, interactive=False)
+                    chat_output = gr.Textbox(label="💬 Chat with Document (Simulated)", lines=15, interactive=False)
 
-        # === File Upload Handler ===
-        def parse_sections(file):
+        # === Upload Handler ===
+        def store_file_and_generate_guide(file, current_files, section_map):
             if file is None:
-                return "No file uploaded."
+                return current_files, section_map, "No file uploaded.", gr.update(choices=[])
 
             text = extract_text_from_file(file)
-            prompt = f"Break this content into 4–6 learning sections with a title and short explanation.\n\n{text}"
-            response = call_llm(prompt, system_message="You are an education assistant.")
+            filename = os.path.basename(file.name)
+
+            prompt = f"Break the content into 4–6 logical learning sections. Format: 'Section Title: Short summary'\n\n{text}"
+            response = call_llm(prompt, system_message="You are a helpful assistant for students studying documents.")
+
+            # Save file name and content
+            updated_files = current_files + [filename] if filename not in current_files else current_files
+            section_map[filename] = response
+
+            return updated_files, section_map, f"### {filename} Study Guide\n\n{response}", gr.update(choices=updated_files)
+
+        # === Select Study Guide ===
+        def load_study_guide(selected_filename, section_map):
+            if not selected_filename or selected_filename not in section_map:
+                return "No study guide available."
+            return f"### {selected_filename} Study Guide\n\n{section_map[selected_filename]}"
 
-            return f"### 📑 Auto-Generated Study Guide\n\n{response}"
+        # === Switch to Chat/File Viewer ===
+        def open_file_viewer():
+            return gr.update(visible=True), "Simulated PDF viewer content...", "Ask me anything about this file."
 
         file_upload.change(
-            fn=parse_sections,
-            inputs=[file_upload],
-            outputs=[content_output],
+            fn=store_file_and_generate_guide,
+            inputs=[file_upload, uploaded_files, file_sections],
+            outputs=[uploaded_files, file_sections, study_guide_output, file_guide_list],
             show_progress="minimal"
         )
 
-        return file_panel
+        file_guide_list.change(
+            fn=load_study_guide,
+            inputs=[file_guide_list, file_sections],
+            outputs=[study_guide_output],
+            show_progress="minimal"
+        )
+
+        btn_file_view.click(
+            fn=open_file_viewer,
+            inputs=[],
+            outputs=[file_chat_output, file_preview, chat_output]
+        )
+
+        return file_panel