Update file_module.py

file_module.py

@@ -12,6 +12,7 @@ def file_dashboard():
         uploaded_files = gr.State([])
         file_sections = gr.State({})
         file_texts = gr.State({})
+        chat_history = gr.State({})
 
         with gr.Row():
             # === LEFT SIDEBAR ===
@@ -40,8 +41,8 @@ ankban@wharton.upenn.edu""", elem_id="sidebar-user")
                 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 Response", lines=8, interactive=False)
+                    file_preview = gr.Textbox(label="📄 File Content Preview", lines=15, interactive=False)
+                    chat_output = gr.Chatbot(label="💬 Chat with Document")
 
                 chat_input = gr.Textbox(label="Ask something about this file")
 
@@ -69,17 +70,25 @@ ankban@wharton.upenn.edu""", elem_id="sidebar-user")
             return f"### {selected_filename} Study Guide\n\n{section_map[selected_filename]}"
 
         # === Show File Viewer + Chat ===
-        def open_file_viewer():
-            return gr.update(visible=True), "Simulated PDF viewer content...", ""
+        def open_file_viewer(selected_file, text_map):
+            content = text_map.get(selected_file, "No file content available.")
+            return gr.update(visible=True), content, []
 
         # === GPT Chat Handler ===
-        def chat_with_file(question, selected_file, text_map):
+        def chat_with_file(question, selected_file, text_map, history_map):
             if not selected_file or selected_file not in text_map:
-                return "Please select a file first."
+                return [], history_map
 
             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.")
+            history = history_map.get(selected_file, [])
+
+            prompt = f"You are a helpful tutor. Answer this question using the document content below.\n\n{text}\n\nQuestion: {question}"
+            answer = call_llm(prompt)
+
+            new_history = history + [(question, answer)]
+            history_map[selected_file] = new_history
+
+            return new_history, history_map
 
         file_upload.change(
             fn=store_file_and_generate_guide,
@@ -97,14 +106,14 @@ ankban@wharton.upenn.edu""", elem_id="sidebar-user")
 
         btn_file_view.click(
             fn=open_file_viewer,
-            inputs=[],
+            inputs=[file_guide_list, file_texts],
             outputs=[file_chat_output, file_preview, chat_output]
         )
 
-        chat_input.change(
+        chat_input.submit(
             fn=chat_with_file,
-            inputs=[chat_input, file_guide_list, file_texts],
-            outputs=[chat_output],
+            inputs=[chat_input, file_guide_list, file_texts, chat_history],
+            outputs=[chat_output, chat_history],
             show_progress="minimal"
         )