Update file_viewer_panel.py

file_viewer_panel.py

@@ -3,19 +3,17 @@ import os
 import shutil
 import json
 import tempfile
-from gradio import processing_utils
 from pdf2image import convert_from_path
+from gradio.processing_utils import save_temp_file
 from file_utils import extract_text_from_file, call_llm, save_user_file, fetch_user_files
 
 def file_viewer_dashboard(nickname):
     with gr.Column(scale=4, elem_id="main-column") as file_panel:
-        nickname_input = nickname
-
         with gr.Row():
             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")
+                file_upload = gr.File(label="📂 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"])
+                image_viewer = gr.HTML(label="📄 PDF Viewer")
+                load_more_btn = gr.Button("➕ Load More Pages", visible=False)
 
             with gr.Column(scale=2):
                 generate_quiz_btn = gr.Button("🧠 Practice Questions")
@@ -25,54 +23,60 @@ def file_viewer_dashboard(nickname):
 
         file_text_state = gr.State("")
         chat_state = gr.State([])
-        
+        all_image_paths = gr.State([])
+        loaded_page_count = gr.State(0)
 
-        def render_file_as_html(filename, text):
-            ext = os.path.splitext(filename)[1].lower()
-            
-            if ext == ".pdf":
-                from pdf2image import convert_from_path
-        
-                temp_dir = tempfile.mkdtemp()
-                images = convert_from_path(filename, dpi=150, fmt="png", output_folder=temp_dir)
-        
-                img_tags = []
-                for i, img in enumerate(images):
-                    img_path = os.path.join(temp_dir, f"page_{i}.png")
-                    img.save(img_path, "PNG")
-        
-                    # Copy to Gradio's accessible location and get file URL
-                    gradio_img_path = processing_utils.save_temp_file(img_path)
-                    img_tags.append(f"<img src='file={gradio_img_path}' style='width:100%; margin-bottom:10px;'/>")
-        
-                return "".join(img_tags)
-        
-            else:
-                return f"<pre style='padding:12px; background:#f9f9f9; white-space:pre-wrap;'>{text}</pre>"
-
-
-        def handle_file_upload(file, nickname):
+        def prepare_pdf_images(file):
             if not file or not nickname:
-                return "", "", [], []
+                return "", [], 0, gr.update(visible=False)
 
             filename = os.path.basename(file.name)
             saved_path = f"/tmp/{filename}"
             shutil.copy(file.name, saved_path)
-
             save_user_file(nickname, filename, saved_path)
-            text = extract_text_from_file(file)
-            html = render_file_as_html(file.name, text)
 
-            user_files = fetch_user_files(nickname)
-            file_choices = [f"{f.filename} ({f.timestamp})" for f in user_files]
+            text = extract_text_from_file(file)
 
-            return html, text, file_choices, []
+            ext = os.path.splitext(filename)[1].lower()
+            if ext != ".pdf":
+                html = f"<pre style='padding:12px; background:#f9f9f9; white-space:pre-wrap;'>{text}</pre>"
+                return html, [], 0, gr.update(visible=False)
+
+            temp_dir = tempfile.mkdtemp()
+            images = convert_from_path(saved_path, dpi=150, fmt="png", output_folder=temp_dir)
+
+            paths = []
+            for i, img in enumerate(images):
+                img_path = os.path.join(temp_dir, f"page_{i}.png")
+                img.save(img_path, "PNG")
+                gr_img_path = save_temp_file(img_path)
+                paths.append(gr_img_path)
+
+            initial_html = "".join([
+                f"<img src='file={paths[i]}' style='width:100%; margin-bottom:12px; border-radius:8px;'/>"
+                for i in range(min(3, len(paths)))
+            ])
+            return initial_html, paths, min(3, len(paths)), gr.update(visible=(len(paths) > 3))
+
+        def load_more_pages(paths, current_count):
+            next_count = current_count + 3
+            total = len(paths)
+
+            new_html = "".join([
+                f"<img src='file={paths[i]}' style='width:100%; margin-bottom:12px; border-radius:8px;'/>"
+                for i in range(current_count, min(next_count, total))
+            ])
+
+            show_more = next_count < total
+            return gr.update(value=new_html, append=True), next_count, gr.update(visible=show_more)
 
         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}"})
+            messages = [
+                {"role": "system", "content": "You are a tutor. Only use the document below."},
+                {"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})
@@ -104,15 +108,21 @@ def file_viewer_dashboard(nickname):
             except:
                 return gr.update(visible=False), []
 
-        def load_user_files(nickname):
-            files = fetch_user_files(nickname)
-            return [f"{f.filename} ({f.timestamp})" for f in files]
+        file_upload.change(
+            fn=prepare_pdf_images,
+            inputs=[file_upload],
+            outputs=[image_viewer, all_image_paths, loaded_page_count, load_more_btn]
+        )
+
+        load_more_btn.click(
+            fn=load_more_pages,
+            inputs=[all_image_paths, loaded_page_count],
+            outputs=[image_viewer, loaded_page_count, load_more_btn]
+        )
 
-        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
+    return file_panel