Update app.py

app.py

@@ -6,60 +6,59 @@ from gradio_client import Client, handle_file
 HF_TOKEN = os.getenv("HF_TOKEN")
 TARGET_SPACE = "selfit-camera/omni-image-editor"
 
-def bridge_api(image, prompt):
-    if not image:
-        return None, "Error: No image uploaded."
+def process_via_api(image_path, prompt):
+    if not image_path:
+        return None, "Status: Please upload an image."
     
     try:
+        # Connect to the target space
         client = Client(TARGET_SPACE, hf_token=HF_TOKEN)
+        
+        # Call the prediction API
         result = client.predict(
-            image=handle_file(image),
+            image=handle_file(image_path),
             prompt=prompt,
             api_name="/predict"
         )
-        return result, "Status: Processed successfully"
+        return result, "Status: Processed successfully via API"
     except Exception as e:
         return None, f"Status Error: {str(e)}"
 
-# แก้ไขส่วน Theme ตรงนี้ให้เป็นมาตรฐานของ Gradio 5
-my_theme = gr.themes.Default(primary_hue="indigo")
-
-with gr.Blocks(theme=my_theme) as demo:
-    gr.HTML("<h1 style='text-align: center;'>🎨 Omni Editor 2.0</h1>")
-    gr.HTML("<p style='text-align: center; background: #6366f1; color: white; padding: 12px; border-radius: 50px; font-weight: bold;'>Proxy Interface Mode (Free Tier)</p>")
+# UI Construction (Minimalist & Stable for Python 3.13)
+with gr.Blocks() as demo:
+    gr.HTML("<h1 style='text-align: center;'>🎨 Omni Editor 2.0 (Remote)</h1>")
+    gr.HTML("<div style='text-align: center; background: #6366f1; color: white; padding: 10px; border-radius: 10px; margin-bottom: 20px;'>Powered by Omni-Image-Editor API | Running on Free Tier</div>")
     
     with gr.Tabs():
-        with gr.TabItem("🖼️ Single Image Edit"):
+        with gr.TabItem("🖼️ Image Editing"):
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("### 📥 Upload Image")
-                    with gr.Group():
-                        input_img = gr.Image(type="filepath", label="Select image")
-                        input_prompt = gr.Textbox(
-                            label="Instruction", 
-                            placeholder="e.g., 'Change background to a snowy mountain'",
-                            lines=3
-                        )
-                        submit_btn = gr.Button("Execute", variant="primary")
+                    input_img = gr.Image(type="filepath", label="Input Image")
+                    input_prompt = gr.Textbox(
+                        label="Editing Instruction", 
+                        placeholder="e.g. 'add a red hat to the dog'",
+                        lines=3
+                    )
+                    submit_btn = gr.Button("Execute", variant="primary")
                 
                 with gr.Column(scale=1):
-                    gr.Markdown("### 🎯 Editing Result")
-                    with gr.Group():
-                        output_img = gr.Image(label="Result")
-                        use_as_input = gr.Button("🔄 Use as Input")
-                    
-                    status_box = gr.Textbox(label="Processing Status", interactive=False)
+                    output_img = gr.Image(label="Output Result")
+                    status_box = gr.Textbox(label="System Status", interactive=False)
+                    use_as_input = gr.Button("🔄 Use Result as New Input")
 
+    # Event Handlers
     submit_btn.click(
-        fn=bridge_api,
+        fn=process_via_api,
         inputs=[input_img, input_prompt],
         outputs=[output_img, status_box]
     )
     
+    # Loop back output to input
     use_as_input.click(
         fn=lambda x: x,
         inputs=output_img,
         outputs=input_img
     )
 
-demo.queue(default_concurrency_limit=20).launch(server_name="0.0.0.0")
+if __name__ == "__main__":
+    demo.queue(max_size=20).launch(server_name="0.0.0.0")