Update app.py

app.py

@@ -1569,26 +1569,30 @@ def process_image(
     try:
         if file is None:
             return (
+                None,
                 None,
                 "Please upload an ID image."
             )
-        image = read_gradio_image(
-            file
-        )
+
+        image = read_gradio_image(file)
         if image is None:
             return (
+                None,
                 None,
                 "Could not decode image."
             )
+
         cropped = crop_id_card(
             image,
             brightness
         )
         if cropped is None:
             return (
+                None,
                 None,
                 "❌ ID card was not detected."
             )
+
         # ----------------------------------------------------
         # USER ROTATION
         # ----------------------------------------------------
@@ -1608,12 +1612,20 @@ def process_image(
                 cropped,
                 cv2.ROTATE_90_COUNTERCLOCKWISE
             )
+
         # ----------------------------------------------------
-        # ALWAYS SAVE SINGLE-ID OUTPUT AS A REAL JPG FILE
+        # PREVIEW
+        # ----------------------------------------------------
+        # Show the processed crop immediately in the ID section.
+        # This is independent from the downloadable JPG file.
+        cropped_rgb = cv2.cvtColor(
+            cropped,
+            cv2.COLOR_BGR2RGB
+        )
+
+        # ----------------------------------------------------
+        # SAVE AS REAL JPG
         # ----------------------------------------------------
-        # Returning a NumPy/PIL image to Gradio can cause Gradio
-        # to re-encode it as WEBP. Save the bytes ourselves and
-        # return the actual .jpg filepath instead.
         fd, jpg_path = tempfile.mkstemp(
             prefix="id_card_crop_",
             suffix=".jpg"
@@ -1635,23 +1647,28 @@ def process_image(
             except OSError:
                 pass
             return (
+                cropped_rgb,
                 None,
                 "❌ Could not save cropped ID card as JPG."
             )
 
         return (
+            cropped_rgb,
             jpg_path,
-            "✅ ID card cropped successfully. JPG saved."
+            "✅ ID card cropped successfully. Preview shown and JPG saved."
         )
+
     except Exception as e:
         print(
             "PROCESS ERROR:",
             repr(e)
         )
         return (
+            None,
             None,
             "❌ Error: " + str(e)
         )
+
 # ============================================================
 # ZIP PROCESSING
 # ============================================================
@@ -1874,18 +1891,114 @@ with gr.Blocks(title="Certificate & Egyptian ID Card Cropper") as interface:
     gr.Markdown("## 🇪🇬 Egyptian ID Card Cropper")
     gr.Markdown("Upload an Egyptian ID photograph or a ZIP containing multiple images.")
 
-    gr.Markdown("### Single Image")
-    input_image = gr.Image(type="filepath", label="Upload ID Image")
-    process_button = gr.Button("Crop ID Card", variant="primary")
-    output_image = gr.File(label="Download Cropped ID Card (JPG)")
-    gr.Markdown("### Image Controls")
-    brightness_slider = gr.Slider(minimum=-50, maximum=50, value=10, step=1, label="Brightness", info="0 = original. Positive values make the ID brighter.")
-    rotation_dropdown = gr.Dropdown(choices=[0, 90, 180, 270], value=0, label="Rotate Image", info="Rotation is applied clockwise.")
-    status = gr.Textbox(label="Status", interactive=False)
-    process_button.click(process_image, [input_image, brightness_slider, rotation_dropdown], [output_image, status])
-    input_image.change(process_image, [input_image, brightness_slider, rotation_dropdown], [output_image, status])
-    brightness_slider.change(process_image, [input_image, brightness_slider, rotation_dropdown], [output_image, status])
-    rotation_dropdown.change(process_image, [input_image, brightness_slider, rotation_dropdown], [output_image, status])
+def process_image(
+    file,
+    brightness,
+    rotation
+):
+    try:
+        if file is None:
+            return (
+                None,
+                None,
+                "Please upload an ID image."
+            )
+
+        image = read_gradio_image(file)
+        if image is None:
+            return (
+                None,
+                None,
+                "Could not decode image."
+            )
+
+        cropped = crop_id_card(
+            image,
+            brightness
+        )
+        if cropped is None:
+            return (
+                None,
+                None,
+                "❌ ID card was not detected."
+            )
+
+        # ----------------------------------------------------
+        # USER ROTATION
+        # ----------------------------------------------------
+        rotation = int(rotation)
+        if rotation == 90:
+            cropped = cv2.rotate(
+                cropped,
+                cv2.ROTATE_90_CLOCKWISE
+            )
+        elif rotation == 180:
+            cropped = cv2.rotate(
+                cropped,
+                cv2.ROTATE_180
+            )
+        elif rotation == 270:
+            cropped = cv2.rotate(
+                cropped,
+                cv2.ROTATE_90_COUNTERCLOCKWISE
+            )
+
+        # ----------------------------------------------------
+        # PREVIEW
+        # ----------------------------------------------------
+        # Show the processed crop immediately in the ID section.
+        # This is independent from the downloadable JPG file.
+        cropped_rgb = cv2.cvtColor(
+            cropped,
+            cv2.COLOR_BGR2RGB
+        )
+
+        # ----------------------------------------------------
+        # SAVE AS REAL JPG
+        # ----------------------------------------------------
+        fd, jpg_path = tempfile.mkstemp(
+            prefix="id_card_crop_",
+            suffix=".jpg"
+        )
+        os.close(fd)
+
+        success = cv2.imwrite(
+            jpg_path,
+            cropped,
+            [
+                int(cv2.IMWRITE_JPEG_QUALITY),
+                95
+            ]
+        )
+
+        if not success:
+            try:
+                os.remove(jpg_path)
+            except OSError:
+                pass
+            return (
+                cropped_rgb,
+                None,
+                "❌ Could not save cropped ID card as JPG."
+            )
+
+        return (
+            cropped_rgb,
+            jpg_path,
+            "✅ ID card cropped successfully. Preview shown and JPG saved."
+        )
+
+    except Exception as e:
+        print(
+            "PROCESS ERROR:",
+            repr(e)
+        )
+        return (
+            None,
+            None,
+            "❌ Error: " + str(e)
+        )
+
 
     gr.Markdown("---")
     gr.Markdown("### 📦 Process a ZIP")