Update app.py

app.py

@@ -128,6 +128,7 @@ def process_image_page(img):
 def process_file_and_create_pdf(file):
     """
     Main function to process a file (image or PDF) and return a path to a new PDF.
+    This function will ensure the temporary output directory is cleaned up safely.
     The @GPU decorator ensures this function is run on the GPU.
     """
     if file is None:
@@ -135,7 +136,6 @@ def process_file_and_create_pdf(file):
 
     temp_output_dir = tempfile.mkdtemp()
     output_pdf_path = os.path.join(temp_output_dir, "ocr_results.pdf")
-    
     input_image_for_display = None
 
     try:
@@ -203,7 +203,6 @@ def process_file_and_create_pdf(file):
             
             temp_img_path = os.path.join(temp_output_dir, "original_image.png")
             original_image.save(temp_img_path)
-            # Corrected the typo: `T` changed to `True`
             c.drawImage(temp_img_path, 50, height - 300, width=200, preserveAspectRatio=True)
             os.remove(temp_img_path)
             
@@ -218,20 +217,28 @@ def process_file_and_create_pdf(file):
                     y = height - 50
             c.save()
 
+        print(f"Generated PDF path: {output_pdf_path}")
         return output_pdf_path, input_image_for_display
     
     except Exception as e:
         print(f"An error occurred: {e}")
+        # Return None, None on error
+        return None, None
+    finally:
+        # Ensure temporary directory is cleaned up after the function returns
         if os.path.exists(temp_output_dir):
+            print(f"Cleaning up temporary directory: {temp_output_dir}")
             shutil.rmtree(temp_output_dir)
-        return None, None
 
 # Gradio Interface
 @GPU
 def process_file_for_gradio(file):
+    """
+    Wrapper function for Gradio interface.
+    This function calls the main processing logic and returns the outputs
+    in the format required by the gr.Interface.
+    """
     output_path, input_image = process_file_and_create_pdf(file)
-    if output_path is None:
-        return None, None
     return output_path, input_image
 
 demo = gr.Interface(