Update file_viewer_panel.py

file_viewer_panel.py

@@ -1,76 +1,81 @@
 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
+from file_utils import extract_text_from_file, call_llm, save_user_file, fetch_user_files
 
 def file_viewer_dashboard():
-    with gr.Column(scale=4, elem_id="main-column") as file_panel:   # ✅ define this
+    with gr.Column(scale=4, elem_id="main-column") as file_panel:
+        nickname_input = gr.Textbox(label="👤 Your Nickname", placeholder="Enter your name")
+
         with gr.Row():
-            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=3):
+                user_file_list = gr.Radio(label="📁 Your Uploaded Files", choices=[], interactive=True)
+                file_upload = gr.File(label="📂 Upload a New File", file_types=[".pdf", ".txt", ".md", ".pptx"])
+                file_viewer = gr.HTML(label="📄 File Preview")
 
-        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...")
+            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 about the file")
+                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 "", "", []
-
-            ext = os.path.splitext(file.name)[1].lower()
-            text = extract_text_from_file(file)
-
+        def render_file_as_html(filename, text):
+            ext = os.path.splitext(filename)[1].lower()
             if ext == ".pdf":
                 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 = [
-                    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))
-                ]
+                images = convert_from_path(filename, dpi=150, fmt="png", output_folder=output_dir)
                 for i, img in enumerate(images):
                     img.save(os.path.join(output_dir, f"page_{i}.png"), "PNG")
-                viewer_html = "".join(img_tags)
-            else:
-                viewer_html = f"<pre style='padding:12px; background:#f9f9f9; white-space:pre-wrap;'>{text}</pre>"
+                return "".join([
+                    f"<img src='/file={os.path.join(output_dir, f'page_{i}.png')}' style='width:100%; margin-bottom:10px;'/>"
+                    for i in range(len(images))
+                ])
+            return f"<pre style='padding:12px; background:#f9f9f9; white-space:pre-wrap;'>{text}</pre>"
 
-            return viewer_html, text, []  # Reset chat on new file
+        def handle_file_upload(file, nickname):
+            if not file or not nickname:
+                return "", "", [], []
 
-        def ask_question_with_context(user_input, text, history):
-            if not user_input.strip():
-                return history, history
+            filename = os.path.basename(file.name)
+            saved_path = f"/tmp/{filename}"
+            shutil.copy(file.name, saved_path)
 
-            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]"
+            save_user_file(nickname, filename, saved_path)
+            text = extract_text_from_file(file)
+            html = render_file_as_html(file.name, text)
 
-            messages.append({"role": "user", "content": f"Document Content:\n{text}"})
+            user_files = fetch_user_files(nickname)
+            file_choices = [f"{f.filename} ({f.timestamp})" for f in user_files]
 
+            return html, text, file_choices, []
+
+        def ask_question(user_input, text, history):
+            if not user_input.strip():
+                return history, history
+            messages = [{"role": "system", "content": "You are a tutor. Only use the document below."}]
+            messages.append({"role": "user", "content": f"Document:\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)
-
             history.append((user_input, response))
             return history, history
 
         def generate_quiz(text):
             prompt = (
-                "Generate 5 multiple choice questions from this document. "
-                "Return a JSON list where each item has:\n- question\n- options (list)\n- answer (correct one)"
+                "Generate 5 multiple choice questions from the document. "
+                "Each question should have:\n- question\n- options (list)\n- answer (string). Return JSON."
             )
             try:
                 result = call_llm(prompt=prompt)
                 questions = json.loads(result)
-
                 components = []
                 for idx, q in enumerate(questions):
                     q_md = gr.Markdown(f"**Q{idx+1}. {q['question']}**")
@@ -82,14 +87,19 @@ def file_viewer_dashboard():
 
                     radio.change(fn=check, inputs=[radio], outputs=[feedback])
                     components.extend([q_md, radio, feedback])
-
                 return gr.update(visible=True), components
             except:
                 return gr.update(visible=False), []
 
-        # 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])
+        def load_user_files(nickname):
+            files = fetch_user_files(nickname)
+            return [f"{f.filename} ({f.timestamp})" for f in files]
+
+        nickname_input.change(fn=load_user_files, inputs=[nickname_input], outputs=[user_file_list])
+        file_upload.change(fn=handle_file_upload, inputs=[file_upload, nickname_input],
+                           outputs=[file_viewer, file_text_state, user_file_list, chat_output])
+        chat_input.submit(fn=ask_question, 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