Update file_viewer_panel.py

file_viewer_panel.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import os
-import shutil
 import json
 from pdf2image import convert_from_path
 from file_utils import extract_text_from_file, call_llm
@@ -15,40 +14,53 @@ def file_viewer_dashboard():
             generate_quiz_btn = gr.Button("🧠 Practice Questions")
             quiz_container = gr.Column(visible=False)
             chat_output = gr.Chatbot(label="Ask a question to learn more about the file")
-            chat_input = gr.Textbox(label="", placeholder="Ask Chatter a question...")
+            chat_input = gr.Textbox(label="", placeholder="Ask ChatEDU a question...")
 
         file_text_state = gr.State("")
+        chat_state = gr.State([])
 
         def render_file(file):
             if not file:
-                return "", ""
+                return "", "", []
 
             ext = os.path.splitext(file.name)[1].lower()
             text = extract_text_from_file(file)
 
             if ext == ".pdf":
-                # Convert to images
                 output_dir = "/tmp/pdf_images"
                 os.makedirs(output_dir, exist_ok=True)
                 images = convert_from_path(file.name, dpi=150, fmt="png", output_folder=output_dir)
-
-                img_tags = []
+                img_tags = [
+                    f"<img src='/file={os.path.join(output_dir, f'page_{i}.png')}' style='width:100%; margin-bottom: 10px; border-radius: 8px;'/>"
+                    for i in range(len(images))
+                ]
                 for i, img in enumerate(images):
-                    img_path = os.path.join(output_dir, f"page_{i}.png")
-                    img.save(img_path, "PNG")
-                    img_tags.append(f"<img src='/file={img_path}' style='width:100%; margin-bottom: 10px; border-radius: 8px;'/>")
-
+                    img.save(os.path.join(output_dir, f"page_{i}.png"), "PNG")
                 viewer_html = "".join(img_tags)
-
             else:
-                # Fallback for text/markdown/pptx
                 viewer_html = f"<pre style='padding:12px; background:#f9f9f9; white-space:pre-wrap;'>{text}</pre>"
 
-            return viewer_html, text
+            return viewer_html, text, []  # Reset chat on new file
+
+        def ask_question_with_context(user_input, text, history):
+            if not user_input.strip():
+                return history, history
+
+            messages = [{"role": "system", "content": "You are a helpful tutor. Answer only using the content provided."}]
+            if len(text) > 8000:
+                text = text[:8000] + "\n\n[Truncated due to token limits]"
+
+            messages.append({"role": "user", "content": f"Document Content:\n{text}"})
+
+            for user_msg, bot_reply in history:
+                messages.append({"role": "user", "content": user_msg})
+                messages.append({"role": "assistant", "content": bot_reply})
+
+            messages.append({"role": "user", "content": user_input})
+            response = call_llm(messages=messages)
 
-        def ask_question(q, text):
-            prompt = f"Based on the document below, answer the user's question.\n\n{text}\n\nQuestion: {q}"
-            return [(q, call_llm(prompt))]
+            history.append((user_input, response))
+            return history, history
 
         def generate_quiz(text):
             prompt = (
@@ -56,7 +68,7 @@ def file_viewer_dashboard():
                 "Return a JSON list where each item has:\n- question\n- options (list)\n- answer (correct one)"
             )
             try:
-                result = call_llm(prompt)
+                result = call_llm(prompt=prompt)
                 questions = json.loads(result)
 
                 components = []
@@ -72,11 +84,12 @@ def file_viewer_dashboard():
                     components.extend([q_md, radio, feedback])
 
                 return gr.update(visible=True), components
-            except Exception as e:
+            except:
                 return gr.update(visible=False), []
 
-        file_upload.change(fn=render_file, inputs=[file_upload], outputs=[file_viewer, file_text_state])
-        chat_input.submit(fn=ask_question, inputs=[chat_input, file_text_state], outputs=[chat_output])
+        # Hook up logic
+        file_upload.change(fn=render_file, inputs=[file_upload], outputs=[file_viewer, file_text_state, chat_output])
+        chat_input.submit(fn=ask_question_with_context, inputs=[chat_input, file_text_state, chat_state], outputs=[chat_output, chat_state])
         generate_quiz_btn.click(fn=generate_quiz, inputs=[file_text_state], outputs=[quiz_container])
 
     return file_panel