Update app.py

app.py

@@ -309,14 +309,10 @@ def detect_forgery(file):
             return None, None, None, None, empty_html
         
         # Get file path
-        file_path = file.name if hasattr(file, 'name') else file
+        file_path = file if isinstance(file, str) else file
         
-        # Check if PDF
-        if file_path.lower().endswith('.pdf'):
-            original, overlay, gauge_dice, gauge_acc, results_html = detector.detect(file_path)
-        else:
-            image = Image.open(file_path)
-            original, overlay, gauge_dice, gauge_acc, results_html = detector.detect(image)
+        # Detect forgeries
+        original, overlay, gauge_dice, gauge_acc, results_html = detector.detect(file_path)
         
         return original, overlay, gauge_dice, gauge_acc, results_html
     
@@ -357,10 +353,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(
+            input_file = gr.Image(
                 label="Document (Image or PDF)",
-                file_types=["image", ".pdf"],
-                type="filepath"
+                type="filepath",
+                sources=["upload"]
             )
             
             with gr.Row():