Error

app.py

@@ -14,8 +14,8 @@ def remove_folds(image, intensity=0.5):
     l, a, b = cv2.split(img_lab)
 
     # --- (1) Smooth illumination correction ---
-    ksize = int(49 + intensity * 100)  # kernel size changes with intensity
-    ksize = ksize + 1 if ksize % 2 == 0 else ksize  # must be odd
+    ksize = int(49 + intensity * 100)
+    ksize = ksize + 1 if ksize % 2 == 0 else ksize
     l_blur = cv2.GaussianBlur(l, (ksize, ksize), 0)
     l_equal = cv2.divide(l, l_blur, scale=128)
 
@@ -25,7 +25,7 @@ def remove_folds(image, intensity=0.5):
     rows, cols = l_equal.shape
     crow, ccol = rows // 2, cols // 2
     mask = np.ones((rows, cols), np.uint8)
-    r = int(10 + intensity * 30)  # notch size grows with intensity
+    r = int(10 + intensity * 30)
     mask[crow - r:crow + r, ccol - r:ccol + r] = 0
     freqshift = freqshift * mask
     ishift = np.fft.ifftshift(freqshift)
@@ -38,35 +38,32 @@ def remove_folds(image, intensity=0.5):
     return Image.fromarray(result)
 
 # -------------------------------------------------
-# 🎛️ GRADIO UI
+# 🎛️ GRADIO BLOCKS UI (with slider throttle)
 # -------------------------------------------------
-title = "🪄 Saree Fold Remover (Lighting Only, No Distortion)"
-description = """
-Upload a flat saree or fabric image.<br>
-Adjust the **Fold Intensity** slider — higher values remove deeper folds.<br>
-The process equalizes lighting only, keeping weave and motifs intact.
-"""
+with gr.Blocks(title="🪄 Saree Fold Remover (Lighting Only, No Distortion)") as demo:
+    gr.Markdown(
+        """
+        ## 🪄 Saree Fold Remover  
+        Upload a flat saree or fabric image.<br>
+        Adjust the **Fold Intensity** slider — higher values remove deeper folds.<br>
+        Lighting only correction, no geometric distortion.
+        """
+    )
 
-iface = gr.Interface(
-    fn=remove_folds,
-    inputs=[
-        gr.Image(label="Upload Saree Image", type="pil"),
-        gr.Slider(
-            0.0,
-            1.0,
-            value=0.5,
-            step=0.05,
+    with gr.Row():
+        inp = gr.Image(label="Upload Saree Image", type="pil")
+        intensity = gr.Slider(
+            0.0, 1.0, value=0.5, step=0.05,
             label="Fold Intensity",
-            info="Higher = stronger flattening (lighting only)",
-            interactive=True
-        ),
-    ],
-    outputs=gr.Image(label="Flat, Fold-Free Output"),
-    title=title,
-    description=description,
-    live=False,        # prevents rerun while moving
-    throttle=0.8       # waits 0.8 s before triggering
-)
+            info="Higher = stronger flattening (lighting only)"
+        )
 
-if __name__ == "__main__":
-    iface.launch()
+    out = gr.Image(label="Flat, Fold-Free Output")
+
+    # Button to trigger with throttle behavior
+    run_btn = gr.Button("✨ Remove Folds")
+
+    # Run when button is pressed (instead of live refresh)
+    run_btn.click(fn=remove_folds, inputs=[inp, intensity], outputs=out)
+
+demo.launch()