Update app.py

app.py

@@ -298,8 +298,8 @@ class ForgeryDetector:
         
         # Calculate IoU, Precision, Recall from the refined mask and probability map
         if num_detections > 0:
-            # Use high-confidence areas from probability map as "predicted positive"
-            high_conf_mask = (prob_map > 0.7).astype(np.uint8)
+            # Use resized prob_map to match refined_mask dimensions
+            high_conf_mask = (prob_map_resized > 0.7).astype(np.uint8)
             predicted_positive = np.sum(refined_mask > 0)
             high_conf_positive = np.sum(high_conf_mask > 0)
             
@@ -412,16 +412,18 @@ class ForgeryDetector:
 detector = ForgeryDetector()
 
 
-def detect_forgery(file):
-    """Gradio interface function"""
+def detect_forgery(image_input, pdf_input):
+    """Gradio interface function - handles image/webcam and PDF inputs"""
     try:
-        if file is None:
+        # Prioritize image input, fallback to PDF
+        if image_input is not None:
+            file_path = image_input
+        elif pdf_input is not None:
+            file_path = pdf_input
+        else:
             empty_html = "<div style='padding:12px; border:1px solid #d9534f; border-radius:8px;'>❌ <b>No file uploaded.</b></div>"
             return None, None, empty_html
         
-        # Get file path
-        file_path = file if isinstance(file, str) else file
-        
         # Detect forgeries
         overlay, metrics_gauge, results_html = detector.detect(file_path)
         
@@ -465,11 +467,16 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Upload Document")
-            input_file = gr.Image(
-                label="Document (Image or PDF)",
+            input_image = gr.Image(
+                label="Image (with preview) or Webcam",
                 type="filepath",
                 sources=["upload", "webcam"]
             )
+            input_pdf = gr.File(
+                label="Or upload PDF here",
+                file_types=[".pdf"],
+                type="filepath"
+            )
             
             with gr.Row():
                 clear_btn = gr.Button("🧹 Clear", elem_classes="clear-btn")
@@ -555,14 +562,14 @@ with gr.Blocks(css=custom_css) as demo:
     # Event handlers
     analyze_btn.click(
         fn=detect_forgery,
-        inputs=[input_file],
+        inputs=[input_image, input_pdf],
         outputs=[output_image, metrics_gauge, output_html]
     )
     
     clear_btn.click(
-        fn=lambda: (None, None, None, "<i>No analysis yet. Upload a document and click Analyze.</i>"),
+        fn=lambda: (None, None, None, None, "<i>No analysis yet. Upload a document and click Analyze.</i>"),
         inputs=None,
-        outputs=[input_file, output_image, metrics_gauge, output_html]
+        outputs=[input_image, input_pdf, output_image, metrics_gauge, output_html]
     )