Update file_viewer_panel.py

file_viewer_panel.py

@@ -1,45 +1,59 @@
 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
-import os, json
 
 def file_viewer_dashboard():
     with gr.Row():
-        with gr.Column(scale=3):  # Main viewer
+        with gr.Column(scale=3):
             file_upload = gr.File(label="📂 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"])
             file_viewer = gr.HTML(label="📄 File Preview")
-        
-        with gr.Column(scale=2):  # Right: Chat + Quiz
+
+        with gr.Column(scale=2):
             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 ChatEDU a question...")
+            chat_input = gr.Textbox(label="", placeholder="Ask Chatter a question...")
 
         file_text_state = gr.State("")
 
         def render_file(file):
             if not file:
                 return "", ""
+
             ext = os.path.splitext(file.name)[1].lower()
-            file_path = file.name
             text = extract_text_from_file(file)
 
             if ext == ".pdf":
-                viewer_html = f"<iframe src='/file={file_path}' width='100%' height='700px' style='border:none; border-radius:8px;'></iframe>"
-            elif ext in [".txt", ".md", ".pptx"]:
-                viewer_html = f"<pre style='padding:12px; background:#f9f9f9;'>{text}</pre>"
+                # 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 = []
+                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;'/>")
+
+                viewer_html = "".join(img_tags)
+
             else:
-                viewer_html = "<p>Unsupported file type.</p>"
+                # Fallback for text/markdown/pptx
+                viewer_html = f"<pre style='padding:12px; background:#f9f9f9; white-space:pre-wrap;'>{text}</pre>"
 
             return viewer_html, text
 
         def ask_question(q, text):
-            prompt = f"Answer the following question based only on this document:\n\n{text}\n\nQuestion: {q}"
+            prompt = f"Based on the document below, answer the user's question.\n\n{text}\n\nQuestion: {q}"
             return [(q, call_llm(prompt))]
 
         def generate_quiz(text):
             prompt = (
                 "Generate 5 multiple choice questions from this document. "
-                "Return a JSON array of questions, each with:\n- question\n- options (list)\n- answer (string)"
+                "Return a JSON list where each item has:\n- question\n- options (list)\n- answer (correct one)"
             )
             try:
                 result = call_llm(prompt)
@@ -58,7 +72,6 @@ def file_viewer_dashboard():
                     components.extend([q_md, radio, feedback])
 
                 return gr.update(visible=True), components
-
             except Exception as e:
                 return gr.update(visible=False), []
 
@@ -66,4 +79,4 @@ def file_viewer_dashboard():
         chat_input.submit(fn=ask_question, inputs=[chat_input, file_text_state], outputs=[chat_output])
         generate_quiz_btn.click(fn=generate_quiz, inputs=[file_text_state], outputs=[quiz_container])
 
-    return file_viewer_dashboard
+    return file_panel