Update app.py

app.py

@@ -412,15 +412,18 @@ class ForgeryDetector:
 detector = ForgeryDetector()
 
 
-def detect_forgery(file):
-    """Gradio interface function - handles image and PDF uploads"""
+def detect_forgery(file, webcam):
+    """Gradio interface function - handles file uploads and webcam capture"""
     try:
-        if file is None:
-            empty_html = "<div style='padding:12px; border:1px solid #d9534f; border-radius:8px;'>❌ <b>No file uploaded.</b></div>"
+        # Use whichever input has data
+        source = file if file is not None else webcam
+        
+        if source is None:
+            empty_html = "<div style='padding:12px; border:1px solid #d9534f; border-radius:8px;'>❌ <b>No input provided.</b> Please upload a file or use webcam.</div>"
             return None, None, empty_html
         
         # Detect forgeries
-        overlay, metrics_gauge, results_html = detector.detect(file)
+        overlay, metrics_gauge, results_html = detector.detect(source)
         
         return overlay, metrics_gauge, results_html
     
@@ -463,11 +466,20 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.Column(scale=1):
             gr.Markdown("### Upload Document")
             
-            input_file = gr.File(
-                label="📤 Upload Image or PDF",
-                file_types=["image", ".pdf"],
-                type="filepath"
-            )
+            with gr.Tabs():
+                with gr.Tab("📤 Upload File"):
+                    input_file = gr.File(
+                        label="Upload Image or PDF",
+                        file_types=["image", ".pdf"],
+                        type="filepath"
+                    )
+                
+                with gr.Tab("📷 Webcam"):
+                    input_webcam = gr.Image(
+                        label="Capture from Webcam",
+                        type="filepath",
+                        sources=["webcam"]
+                    )
             
             with gr.Row():
                 clear_btn = gr.Button("🧹 Clear", elem_classes="clear-btn")
@@ -553,14 +565,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]
     )