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_resized > 0.7).astype(np.uint8)
+            high_conf_mask = (prob_map > 0.7).astype(np.uint8)
             predicted_positive = np.sum(refined_mask > 0)
             high_conf_positive = np.sum(high_conf_mask > 0)
             
@@ -412,18 +412,15 @@ class ForgeryDetector:
 detector = ForgeryDetector()
 
 
-def detect_forgery(file, webcam):
-    """Gradio interface function - accepts file upload or webcam capture"""
+def detect_forgery(file):
+    """Gradio interface function"""
     try:
-        # Prioritize file upload, fallback to webcam
-        input_source = file if file is not None else webcam
-        
-        if input_source is None:
+        if file 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 = input_source if isinstance(input_source, str) else input_source
+        file_path = file if isinstance(file, str) else file
         
         # Detect forgeries
         overlay, metrics_gauge, results_html = detector.detect(file_path)
@@ -468,16 +465,10 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Upload Document")
-            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",
+            input_file = gr.Image(
+                label="Document (Image or PDF)",
                 type="filepath",
-                sources=["webcam"],
-                visible=True
+                sources=["upload", "webcam"]
             )
             
             with gr.Row():
@@ -564,14 +555,14 @@ with gr.Blocks(css=custom_css) as demo:
     # Event handlers
     analyze_btn.click(
         fn=detect_forgery,
-        inputs=[input_file, input_webcam],
+        inputs=[input_file],
         outputs=[output_image, metrics_gauge, output_html]
     )
     
     clear_btn.click(
-        fn=lambda: (None, None, None, None, "<i>No analysis yet. Upload a document and click Analyze.</i>"),
+        fn=lambda: (None, None, None, "<i>No analysis yet. Upload a document and click Analyze.</i>"),
         inputs=None,
-        outputs=[input_file, input_webcam, output_image, metrics_gauge, output_html]
+        outputs=[input_file, output_image, metrics_gauge, output_html]
     )