Update app.py

app.py

@@ -1,91 +1,65 @@
 import os
-import sys
 import gradio as gr
 from gradio_client import Client, handle_file
 
-# Configuration from Source
+# Configuration
 HF_TOKEN = os.getenv("HF_TOKEN")
 TARGET_SPACE = "selfit-camera/omni-image-editor"
 
 def bridge_api(image, prompt):
     if not image:
-        return None, "Please upload an image first."
+        return None, "Error: No image uploaded."
     
     try:
-        # Connect to the target API using your token
         client = Client(TARGET_SPACE, hf_token=HF_TOKEN)
-        
-        # Request processing
         result = client.predict(
             image=handle_file(image),
             prompt=prompt,
             api_name="/predict"
         )
-        return result, "Process Completed Successfully"
+        return result, "Status: Processed successfully"
     except Exception as e:
-        return None, f"Status: {str(e)}"
+        return None, f"Status Error: {str(e)}"
 
-# Building UI following Omni Editor 2.0 Layout
-with gr.Blocks(theme=gr.themes.Default(primary_hue="indigo", secondary_hue="slate")) as demo:
+# แก้ไขส่วน 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;'>Please give us a ❤️ if you find it helpful. Free trials refreshed every few days.</p>")
+    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>")
     
-    with gr.Tabs() as tabs:
+    with gr.Tabs():
         with gr.TabItem("🖼️ Single Image Edit"):
             with gr.Row():
-                # Left Column: Inputs
                 with gr.Column(scale=1):
                     gr.Markdown("### 📥 Upload Image")
                     with gr.Group():
-                        input_img = gr.Image(type="filepath", label="Select image to edit")
+                        input_img = gr.Image(type="filepath", label="Select image")
                         input_prompt = gr.Textbox(
                             label="Instruction", 
-                            placeholder="Describe your changes here...",
+                            placeholder="e.g., 'Change background to a snowy mountain'",
                             lines=3
                         )
-                        submit_btn = gr.Button("Execute", variant="primary", size="lg")
+                        submit_btn = gr.Button("Execute", variant="primary")
                 
-                # Right Column: Outputs & Status
                 with gr.Column(scale=1):
                     gr.Markdown("### 🎯 Editing Result")
                     with gr.Group():
-                        output_img = gr.Image(label="Result Display")
-                        use_as_input = gr.Button("🔄 Use as Input", variant="secondary")
+                        output_img = gr.Image(label="Result")
+                        use_as_input = gr.Button("🔄 Use as Input")
                     
-                    gr.Markdown("### 📋 Processing Status")
-                    status_box = gr.Textbox(show_label=False, interactive=False, placeholder="Waiting for action...")
-
-        # Placeholder for other tabs based on the image
-        with gr.TabItem("🖼️🖼️ Multi-Image Edit"):
-            gr.Markdown("Module locked in Proxy Mode")
-        with gr.TabItem("✨ Text to Image"):
-            gr.Markdown("Module locked in Proxy Mode")
-        with gr.TabItem("🔍 Image Upscale"):
-            gr.Markdown("Module locked in Proxy Mode")
-        with gr.TabItem("🏷️ Remove Watermark"):
-            gr.Markdown("Module locked in Proxy Mode")
+                    status_box = gr.Textbox(label="Processing Status", interactive=False)
 
-    # Button Logic
     submit_btn.click(
         fn=bridge_api,
         inputs=[input_img, input_prompt],
         outputs=[output_img, status_box]
     )
     
-    # Feedback loop: Set output as new input
     use_as_input.click(
         fn=lambda x: x,
         inputs=output_img,
         outputs=input_img
     )
 
-# Launch Settings derived from the source loader
-demo.queue(
-    default_concurrency_limit=20,
-    max_size=50,
-    api_open=False
-).launch(
-    server_name="0.0.0.0",
-    show_error=True,
-    height=800
-)
+demo.queue(default_concurrency_limit=20).launch(server_name="0.0.0.0")