Update app.py

app.py

@@ -124,16 +124,22 @@ class ForgeryDetector:
             gauge_accuracy: Accuracy gauge
             results_html: Detection results as HTML
         """
-        # Handle PDF files
-        if isinstance(image, str) and image.lower().endswith('.pdf'):
-            import fitz  # PyMuPDF
-            pdf_document = fitz.open(image)
-            page = pdf_document[0]
-            pix = page.get_pixmap(matrix=fitz.Matrix(2, 2))
-            image = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.height, pix.width, pix.n)
-            if pix.n == 4:
-                image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
-            pdf_document.close()
+        # Handle file path input (from gr.Image with type="filepath")
+        if isinstance(image, str):
+            if image.lower().endswith('.pdf'):
+                # Handle PDF files
+                import fitz  # PyMuPDF
+                pdf_document = fitz.open(image)
+                page = pdf_document[0]
+                pix = page.get_pixmap(matrix=fitz.Matrix(2, 2))
+                image = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.height, pix.width, pix.n)
+                if pix.n == 4:
+                    image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
+                pdf_document.close()
+            else:
+                # Load image file
+                image = Image.open(image)
+                image = np.array(image)
         
         # Convert PIL to numpy
         if isinstance(image, Image.Image):