Update file_viewer_panel.py

file_viewer_panel.py

@@ -2,6 +2,8 @@ import gradio as gr
 import os
 import shutil
 import json
+import tempfile
+from gradio import processing_utils
 from pdf2image import convert_from_path
 from file_utils import extract_text_from_file, call_llm, save_user_file, fetch_user_files
 
@@ -23,20 +25,31 @@ def file_viewer_dashboard(nickname):
 
         file_text_state = gr.State("")
         chat_state = gr.State([])
+        
 
         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(filename, dpi=150, fmt="png", output_folder=output_dir)
+                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.save(os.path.join(output_dir, f"page_{i}.png"), "PNG")
-                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>"
+                    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):
             if not file or not nickname: