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 file upload (image/PDF) 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
@@ -441,6 +441,31 @@ def detect_forgery(image_input, pdf_input):
         return None, None, error_html
 
 
+def update_preview(file_path):
+    """Update preview when file is uploaded"""
+    if file_path is None:
+        return None
+    
+    # Check if it's an image file
+    if file_path.lower().endswith(('.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.webp', '.gif')):
+        return file_path
+    elif file_path.lower().endswith('.pdf'):
+        # For PDFs, try to render first page as preview
+        try:
+            import fitz
+            pdf = fitz.open(file_path)
+            page = pdf[0]
+            pix = page.get_pixmap(matrix=fitz.Matrix(1, 1))
+            img = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.height, pix.width, pix.n)
+            if pix.n == 4:
+                img = cv2.cvtColor(img, cv2.COLOR_RGBA2RGB)
+            pdf.close()
+            return img
+        except:
+            return None
+    return None
+
+
 # Custom CSS - subtle styling
 custom_css = """
 .predict-btn {
@@ -467,16 +492,22 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Upload Document")
-            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"],
+            input_file = gr.File(
+                label="Upload Image or PDF",
+                file_types=["image", ".pdf"],
                 type="filepath"
             )
+            file_preview = gr.Image(
+                label="Preview",
+                type="numpy",
+                interactive=False
+            )
+            input_webcam = gr.Image(
+                label="Or capture from webcam",
+                type="filepath",
+                sources=["webcam"],
+                height=150
+            )
             
             with gr.Row():
                 clear_btn = gr.Button("🧹 Clear", elem_classes="clear-btn")
@@ -560,16 +591,23 @@ with gr.Blocks(css=custom_css) as demo:
             )
     
     # Event handlers
+    # Update preview when file is uploaded
+    input_file.change(
+        fn=update_preview,
+        inputs=[input_file],
+        outputs=[file_preview]
+    )
+    
     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, None, 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, file_preview, input_webcam, output_image, metrics_gauge, output_html]
     )