Update app.py

app.py

@@ -1,72 +1,91 @@
 import os
+import sys
 import gradio as gr
 from gradio_client import Client, handle_file
 
-# Configuration
+# Configuration from Source
 HF_TOKEN = os.getenv("HF_TOKEN")
 TARGET_SPACE = "selfit-camera/omni-image-editor"
 
-def process_api(image, prompt, mode):
+def bridge_api(image, prompt):
     if not image:
-        return None, "Please upload an image."
+        return None, "Please upload an image first."
     
     try:
+        # Connect to the target API using your token
         client = Client(TARGET_SPACE, hf_token=HF_TOKEN)
-        # Mapping api_name based on UI tabs if necessary
-        # Defaulting to /predict as discussed
+        
+        # Request processing
         result = client.predict(
             image=handle_file(image),
             prompt=prompt,
             api_name="/predict"
         )
-        return result, "Success"
+        return result, "Process Completed Successfully"
     except Exception as e:
-        return None, f"Error: {str(e)}"
+        return None, f"Status: {str(e)}"
 
-# UI Layout mimicking Omni Editor 2.0
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+# Building UI following Omni Editor 2.0 Layout
+with gr.Blocks(theme=gr.themes.Default(primary_hue="indigo", secondary_hue="slate")) 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: 10px; border-radius: 20px;'>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;'>Please give us a ❤️ if you find it helpful. Free trials refreshed every few days.</p>")
     
     with gr.Tabs() as tabs:
-        with gr.TabItem("🖼️ Single Image Edit", id=0):
+        with gr.TabItem("🖼️ Single Image Edit"):
             with gr.Row():
-                with gr.Column():
+                # Left Column: Inputs
+                with gr.Column(scale=1):
                     gr.Markdown("### 📥 Upload Image")
-                    input_img = gr.Image(type="filepath", label="Select image to edit")
-                    input_prompt = gr.Textbox(label="Instruction", placeholder="Describe what to edit...")
-                    submit_btn = gr.Button("Execute", variant="primary")
-                    
-                with gr.Column():
+                    with gr.Group():
+                        input_img = gr.Image(type="filepath", label="Select image to edit")
+                        input_prompt = gr.Textbox(
+                            label="Instruction", 
+                            placeholder="Describe your changes here...",
+                            lines=3
+                        )
+                        submit_btn = gr.Button("Execute", variant="primary", size="lg")
+                
+                # Right Column: Outputs & Status
+                with gr.Column(scale=1):
                     gr.Markdown("### 🎯 Editing Result")
-                    output_img = gr.Image(label="Result")
-                    use_as_input = gr.Button("🔄 Use as Input")
+                    with gr.Group():
+                        output_img = gr.Image(label="Result Display")
+                        use_as_input = gr.Button("🔄 Use as Input", variant="secondary")
                     
                     gr.Markdown("### 📋 Processing Status")
-                    status_box = gr.Textbox(label="", interactive=False)
+                    status_box = gr.Textbox(show_label=False, interactive=False, placeholder="Waiting for action...")
 
-        with gr.TabItem("🖼️🖼️ Multi-Image Edit", id=1):
-            gr.Markdown("Feature coming soon...")
-        with gr.TabItem("✨ Text to Image", id=2):
-            gr.Markdown("Feature coming soon...")
-        with gr.TabItem("🔍 Image Upscale", id=3):
-            gr.Markdown("Feature coming soon...")
-        with gr.TabItem("🏷️ Remove Watermark", id=4):
-            gr.Markdown("Feature coming soon...")
+        # 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")
 
-    # Event Logic
+    # Button Logic
     submit_btn.click(
-        fn=process_api,
+        fn=bridge_api,
         inputs=[input_img, input_prompt],
         outputs=[output_img, status_box]
     )
     
-    # Logic for 'Use as Input'
+    # Feedback loop: Set output as new input
     use_as_input.click(
         fn=lambda x: x,
         inputs=output_img,
         outputs=input_img
     )
 
-if __name__ == "__main__":
-    demo.launch()
+# 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
+)