Update app.py

app.py

@@ -412,14 +412,14 @@ class ForgeryDetector:
 detector = ForgeryDetector()
 
 
-def detect_forgery(file_input, webcam_input):
-    """Gradio interface function - handles file upload (image/PDF) and webcam"""
+def detect_forgery(image_input, pdf_input):
+    """Gradio interface function - handles image/webcam and PDF inputs"""
     try:
-        # 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
+        # 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
         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,31 +441,6 @@ def detect_forgery(file_input, webcam_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 {
@@ -476,6 +451,16 @@ 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
@@ -492,22 +477,20 @@ 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 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.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"
+                )
             
             with gr.Row():
                 clear_btn = gr.Button("🧹 Clear", elem_classes="clear-btn")
@@ -591,23 +574,16 @@ 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_file, input_webcam],
+        inputs=[input_image, input_pdf],
         outputs=[output_image, metrics_gauge, output_html]
     )
     
     clear_btn.click(
-        fn=lambda: (None, None, 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, file_preview, input_webcam, output_image, metrics_gauge, output_html]
+        outputs=[input_image, input_pdf, output_image, metrics_gauge, output_html]
     )