Update app.py

app.py

@@ -218,7 +218,7 @@ class ForgeryDetector:
         # Create HTML response
         results_html = self._create_html_report(results)
         
-        return original_image, overlay, gauge_dice, gauge_accuracy, results_html
+        return overlay, gauge_dice, gauge_accuracy, results_html
     
     def _create_overlay(self, image, results):
         """Create overlay visualization"""
@@ -312,15 +312,15 @@ def detect_forgery(file):
     try:
         if file is None:
             empty_html = "<div style='padding:12px; border:1px solid #d9534f; border-radius:8px;'>❌ <b>No file uploaded.</b></div>"
-            return None, None, None, None, empty_html
+            return None, None, None, empty_html
         
         # Get file path
         file_path = file if isinstance(file, str) else file
         
         # Detect forgeries
-        original, overlay, gauge_dice, gauge_acc, results_html = detector.detect(file_path)
+        overlay, gauge_dice, gauge_acc, results_html = detector.detect(file_path)
         
-        return original, overlay, gauge_dice, gauge_acc, results_html
+        return overlay, gauge_dice, gauge_acc, results_html
     
     except Exception as e:
         import traceback
@@ -331,7 +331,7 @@ def detect_forgery(file):
             ❌ <b>Error:</b> {str(e)}
         </div>
         """
-        return None, None, None, None, error_html
+        return None, None, None, error_html
 
 
 # Custom CSS - subtle styling
@@ -384,9 +384,7 @@ with gr.Blocks(css=custom_css) as demo:
         
         with gr.Column(scale=2):
             gr.Markdown("### Detection Results")
-            with gr.Row():
-                original_image = gr.Image(label="Original Document", type="numpy")
-                output_image = gr.Image(label="Detected Forgeries", type="numpy")
+            output_image = gr.Image(label="Detected Forgeries", type="numpy")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -414,13 +412,13 @@ with gr.Blocks(css=custom_css) as demo:
     analyze_btn.click(
         fn=detect_forgery,
         inputs=[input_file],
-        outputs=[original_image, output_image, gauge_dice, gauge_accuracy, output_html]
+        outputs=[output_image, gauge_dice, gauge_accuracy, 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, original_image, output_image, gauge_dice, gauge_accuracy, output_html]
+        outputs=[input_file, output_image, gauge_dice, gauge_accuracy, output_html]
     )