Update module_ocr.py

module_ocr.py

@@ -106,7 +106,9 @@ def process(
     
     return output_text, output_pdf
 
-
+#
+# Preview the document (image or PDF)
+#
 def preview_file(file):
     if file is None:
         return None, None
@@ -131,12 +133,9 @@ def preview_file(file):
             
             if pages:
                 # Save the first page as a temporary image
-                base_filename = os.path.basename(file_path)
-                base_filename, _ = os.path.splitext(base_filename)
-                output_path = f"{base_filename}_{uuid.uuid4()}.png"
-                output_path = os.path.join(output_dir, output_path)
-                pages[0].save(output_path.name, 'PNG')
-                return output_path.name.name, f"PDF Preview: {os.path.basename(file_path)}"
+                with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmp_file:
+                    pages[0].save(tmp_file.name, 'PNG')
+                    return tmp_file.name, f"PDF Preview: {os.path.basename(file_path)}"
             else:
                 return None, "<p>Could not convert PDF to image</p>"
                 
@@ -214,11 +213,11 @@ with gr.Blocks() as demo:
         """)
 
     # Update preview when file is uploaded
-    #input_file.change(
-    #    fn=preview_file,
-    #    inputs=[input_file],
-    #    outputs=[preview_image, preview_text]
-    #)
+    input_file.change(
+        fn=preview_file,
+        inputs=[input_file],
+        outputs=[preview_image, preview_text]
+    )
     
     # Functions
     ocr_btn.click(