Update app.py

app.py

@@ -68,24 +68,21 @@ def process_image(input_img, brightness_threshold=150):
         cv2.imwrite(temp_img_file.name, annotated_img)
 
         # Save PDF report temporarily
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf_file:
-            create_pdf_report(coverage_percent, extracted_texts, temp_img_file.name, temp_pdf_file.name)
-
-            # Read PDF bytes for download
-            with open(temp_pdf_file.name, "rb") as f:
-                pdf_bytes = f.read()
+        temp_pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+        temp_pdf_file.close()  # Close so FPDF can write to it
+        create_pdf_report(coverage_percent, extracted_texts, temp_img_file.name, temp_pdf_file.name)
 
     # Convert annotated image back to RGB for display
     annotated_img_rgb = cv2.cvtColor(annotated_img, cv2.COLOR_BGR2RGB)
 
-    # Clean up temp files
-    os.unlink(temp_img_file.name)
-    os.unlink(temp_pdf_file.name)
-
     report_text = f"UV Sterilization Coverage: {coverage_percent:.2f}%\n\nExtracted Texts:\n"
     report_text += "\n".join(extracted_texts) if extracted_texts else "No text detected."
 
-    return annotated_img_rgb, report_text, pdf_bytes
+    # Delete annotated image file to avoid temp buildup
+    os.unlink(temp_img_file.name)
+
+    # Return annotated image, report text, and PDF file path
+    return annotated_img_rgb, report_text, temp_pdf_file.name
 
 iface = gr.Interface(
     fn=process_image,