Update app.py

app.py

@@ -146,17 +146,30 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
     with gr.Tabs():
         with gr.Tab("📂 Upload File"):
             file_input = gr.File(label="Upload Document", file_types=[".pdf", ".docx", ".png", ".jpg", ".jpeg"])
-            preview_box = gr.Component(label="📄 File Preview", elem_id="preview-box")
+            
+            # Use both preview options: one for text, one for image
+            preview_text = gr.Textbox(label="📄 File/Text Preview", lines=10, elem_id="preview-box")
+            preview_image = gr.Image(label="🖼️ Image Preview", elem_id="preview-box")
+
             report_box = gr.Textbox(label="Verification Report", lines=20, elem_id="report-box")
             verify_btn_file = gr.Button("🔍 Verify Document")
-            verify_btn_file.click(fn=verify_document, inputs=file_input, outputs=[preview_box, report_box])
+
+            def handle_file(file):
+                preview, report = verify_document(file)
+                if isinstance(preview, Image.Image):  # image case
+                    return None, preview, report
+                else:  # text case
+                    return preview, None, report
+
+            verify_btn_file.click(fn=handle_file, inputs=file_input, outputs=[preview_text, preview_image, report_box])
 
         with gr.Tab("📝 Paste Text"):
             text_input = gr.Textbox(label="Paste Notification Text", lines=10, placeholder="Paste text here...")
-            preview_text = gr.Textbox(label="Text Preview", lines=10, elem_id="preview-box")
+            preview_text2 = gr.Textbox(label="Text Preview", lines=10, elem_id="preview-box")
             report_box_text = gr.Textbox(label="Verification Report", lines=20, elem_id="report-box")
             verify_btn_text = gr.Button("🔍 Verify Text")
-            verify_btn_text.click(fn=process_text_input, inputs=text_input, outputs=[preview_text, report_box_text])
+
+            verify_btn_text.click(fn=process_text_input, inputs=text_input, outputs=[preview_text2, report_box_text])
 
 # ------------------------
 # Launch