Update app.py

app.py

@@ -299,7 +299,7 @@ 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)
+            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,15 +412,18 @@ class ForgeryDetector:
 detector = ForgeryDetector()
 
 
-def detect_forgery(file):
-    """Gradio interface function"""
+def detect_forgery(file, webcam):
+    """Gradio interface function - accepts file upload or webcam capture"""
     try:
-        if file is None:
+        # Prioritize file upload, fallback to webcam
+        input_source = file if file is not None else webcam
+        
+        if input_source is None:
             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
+        file_path = input_source if isinstance(input_source, str) else input_source
         
         # Detect forgeries
         overlay, metrics_gauge, results_html = detector.detect(file_path)
@@ -465,10 +468,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_file = gr.File(
+                label="Upload Document (Image or PDF)",
+                file_types=["image", ".pdf"],
+                type="filepath"
+            )
+            input_webcam = gr.Image(
+                label="Or Capture from Webcam",
                 type="filepath",
-                sources=["upload", "webcam"]
+                sources=["webcam"],
+                visible=True
             )
             
             with gr.Row():
@@ -555,14 +564,14 @@ with gr.Blocks(css=custom_css) as demo:
     # Event handlers
     analyze_btn.click(
         fn=detect_forgery,
-        inputs=[input_file],
+        inputs=[input_file, input_webcam],
         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_file, input_webcam, output_image, metrics_gauge, output_html]
     )