Update file_module.py

file_module.py

@@ -11,6 +11,7 @@ def file_dashboard():
     with gr.Blocks(elem_id="file-learning-layout") as file_panel:
         uploaded_files = gr.State([])
         file_sections = gr.State({})
+        file_texts = gr.State({})
 
         with gr.Row():
             # === LEFT SIDEBAR ===
@@ -40,12 +41,14 @@ ankban@wharton.upenn.edu""", elem_id="sidebar-user")
 
                 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)
+                    chat_output = gr.Textbox(label="💬 Chat Response", lines=8, interactive=False)
+
+                chat_input = gr.Textbox(label="Ask something about this file")
 
         # === Upload Handler ===
-        def store_file_and_generate_guide(file, current_files, section_map):
+        def store_file_and_generate_guide(file, current_files, section_map, text_map):
             if file is None:
-                return current_files, section_map, "No file uploaded.", gr.update(choices=[])
+                return current_files, section_map, text_map, "No file uploaded.", gr.update(choices=[])
 
             text = extract_text_from_file(file)
             filename = os.path.basename(file.name)
@@ -53,11 +56,11 @@ ankban@wharton.upenn.edu""", elem_id="sidebar-user")
             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
+            text_map[filename] = text
 
-            return updated_files, section_map, f"### {filename} Study Guide\n\n{response}", gr.update(choices=updated_files)
+            return updated_files, section_map, text_map, f"### {filename} Study Guide\n\n{response}", gr.update(choices=updated_files)
 
         # === Select Study Guide ===
         def load_study_guide(selected_filename, section_map):
@@ -65,14 +68,23 @@ ankban@wharton.upenn.edu""", elem_id="sidebar-user")
                 return "No study guide available."
             return f"### {selected_filename} Study Guide\n\n{section_map[selected_filename]}"
 
-        # === Switch to Chat/File Viewer ===
+        # === Show File Viewer + Chat ===
         def open_file_viewer():
-            return gr.update(visible=True), "Simulated PDF viewer content...", "Ask me anything about this file."
+            return gr.update(visible=True), "Simulated PDF viewer content...", ""
+
+        # === GPT Chat Handler ===
+        def chat_with_file(question, selected_file, text_map):
+            if not selected_file or selected_file not in text_map:
+                return "Please select a file first."
+
+            text = text_map[selected_file]
+            prompt = f"Answer the following question using only this document:\n\n{text}\n\nQuestion: {question}"
+            return call_llm(prompt, system_message="You are a tutor helping a student understand their document.")
 
         file_upload.change(
             fn=store_file_and_generate_guide,
-            inputs=[file_upload, uploaded_files, file_sections],
-            outputs=[uploaded_files, file_sections, study_guide_output, file_guide_list],
+            inputs=[file_upload, uploaded_files, file_sections, file_texts],
+            outputs=[uploaded_files, file_sections, file_texts, study_guide_output, file_guide_list],
             show_progress="minimal"
         )
 
@@ -89,4 +101,11 @@ ankban@wharton.upenn.edu""", elem_id="sidebar-user")
             outputs=[file_chat_output, file_preview, chat_output]
         )
 
+        chat_input.change(
+            fn=chat_with_file,
+            inputs=[chat_input, file_guide_list, file_texts],
+            outputs=[chat_output],
+            show_progress="minimal"
+        )
+
         return file_panel