Update app.py

app.py

@@ -412,14 +412,14 @@ class ForgeryDetector:
 detector = ForgeryDetector()
 
 
-def detect_forgery(image_input, pdf_input):
-    """Gradio interface function - handles image/webcam and PDF inputs"""
+def detect_forgery(file_input, webcam_input):
+    """Gradio interface function - handles unified file upload and webcam"""
     try:
-        # 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
+        # Prioritize file upload, fallback to webcam
+        if file_input is not None:
+            file_path = file_input
+        elif webcam_input is not None:
+            file_path = webcam_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
@@ -451,16 +451,6 @@ custom_css = """
     background-color: #6A89A7 !important;
     color: white !important;
 }
-.upload-container {
-    border: 1px solid #444 !important;
-    border-radius: 10px !important;
-    padding: 10px !important;
-}
-.pdf-btn {
-    background-color: #d9534f !important;
-    color: white !important;
-    font-size: 0.9em !important;
-}
 """
 
 # Create Gradio interface
@@ -477,20 +467,25 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Upload Document")
-            with gr.Group(elem_classes="upload-container"):
-                input_image = gr.Image(
-                    label="Upload Image or Use Webcam",
-                    type="filepath",
-                    sources=["upload", "webcam"]
-                )
-                with gr.Row():
-                    gr.Markdown("<center><b>or</b></center>")
-                input_pdf = gr.File(
-                    label="📄 Upload PDF",
-                    file_types=[".pdf"],
-                    type="filepath",
-                    elem_classes="pdf-btn"
-                )
+            
+            # Single unified input that accepts images and PDFs
+            input_file = gr.File(
+                label="📤 Upload Image/PDF or 📷 Use Webcam",
+                file_types=["image", ".pdf"],
+                type="filepath"
+            )
+            
+            # Hidden webcam component for capture functionality
+            input_webcam = gr.Image(
+                label="Webcam Capture",
+                type="filepath",
+                sources=["webcam"],
+                visible=False
+            )
+            
+            # Button to trigger webcam
+            with gr.Row():
+                webcam_btn = gr.Button("📷 Open Webcam", size="sm")
             
             with gr.Row():
                 clear_btn = gr.Button("🧹 Clear", elem_classes="clear-btn")
@@ -574,16 +569,23 @@ with gr.Blocks(css=custom_css) as demo:
             )
     
     # Event handlers
+    # Toggle webcam visibility
+    webcam_btn.click(
+        fn=lambda: gr.update(visible=True),
+        inputs=None,
+        outputs=[input_webcam]
+    )
+    
     analyze_btn.click(
         fn=detect_forgery,
-        inputs=[input_image, input_pdf],
+        inputs=[input_file, input_webcam],
         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, gr.update(visible=False), None, None, "<i>No analysis yet. Upload a document and click Analyze.</i>"),
         inputs=None,
-        outputs=[input_image, input_pdf, output_image, metrics_gauge, output_html]
+        outputs=[input_file, input_webcam, input_webcam, output_image, metrics_gauge, output_html]
     )