Update app.py

app.py

@@ -104,21 +104,17 @@ def change_dress_color(img, color_prompt):
         print(f"Error: {e}")
         return img
 
-# Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("## 🎨 AI Dress Recolorer - Prompt Based")
-    gr.Markdown("Upload an image and type a color (e.g., 'lavender', 'light green', 'royal blue').")
-
-    with gr.Row():
-        with gr.Column():
-            input_image = gr.Image(type="pil", label="Upload Image")
-            color_input = gr.Textbox(label="Enter Dress Color", placeholder="e.g. crimson, lavender, sky blue")
-            recolor_btn = gr.Button("Apply New Color")
-
-        with gr.Column():
-            output_image = gr.Image(type="pil", label="Recolored Result")
-
-    recolor_btn.click(fn=change_dress_color, inputs=[input_image, color_input], outputs=output_image)
+# Fixed Gradio UI using Interface instead of Blocks
+demo = gr.Interface(
+    fn=change_dress_color,
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Textbox(label="Enter Dress Color", placeholder="e.g. crimson, lavender, sky blue")
+    ],
+    outputs=gr.Image(type="pil", label="Recolored Result"),
+    title="🎨 AI Dress Recolorer - Prompt Based",
+    description="Upload an image and type a color (e.g., 'lavender', 'light green', 'royal blue')."
+)
 
 if __name__ == "__main__":
     demo.launch()