Update app.py

app.py

@@ -48,4 +48,27 @@ def process_image_with_doctr(input_image_pil):
     final_image_bgr = cv2.cvtColor(final_image_rgb, cv2.COLOR_RGB_BGR)
     
     # Save to a temporary file for the download button
-    with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as
+    with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp_file:
+        cv2.imwrite(tmp_file.name, final_image_bgr)
+        # Return the RGB image for Gradio display and the file path for download
+        return final_image_rgb, tmp_file.name
+
+# ==============================================================================
+# Gradio Interface
+# ==============================================================================
+demo = gr.Interface(
+    fn=process_image_with_doctr,
+    inputs=gr.Image(type="pil", label="Upload Document Photo"),
+    outputs=[
+        gr.Image(type="numpy", label="Corrected Document"),
+        gr.File(label="Download as JPG")
+    ],
+    title="📄 AI-Powered Document Scanner",
+    description="A robust, universal document scanner powered by the Doctr deep learning library. This tool correctly handles complex photos with skew, rotation, and varied backgrounds.",
+    # The `allow_flagging` parameter is deprecated. Using `flagging_options=None` is the modern equivalent.
+    # We remove the examples to prevent startup errors from network issues.
+    flagging_options=None 
+)
+
+if __name__ == "__main__":
+    demo.launch()