Update app.py

app.py

@@ -6,84 +6,78 @@ from gradio_client import Client, handle_file
 HF_TOKEN = os.getenv("HF_TOKEN")
 TARGET = "selfit-camera/omni-image-editor"
 
-def call_remote_api(image, prompt, mode_api_name):
-    if image is None and mode_api_name != "/t2i": # T2I ไม่ต้องใช้รูป
-        return None, "Error: Please provide necessary input."
-    
+def call_remote_api(images, prompt, mode_api_name):
     try:
         client = Client(TARGET, hf_token=HF_TOKEN)
         
-        # จัดการ input ตามแต่ละโหมด
         if mode_api_name == "/t2i":
             result = client.predict(prompt=prompt, api_name="/t2i")
         elif mode_api_name == "/multi_edit":
-            # สำหรับ Multi-edit ต้นทางอาจต้องการ list ของไฟล์
-            result = client.predict(images=[handle_file(image)], prompt=prompt, api_name="/multi_edit")
+            # ส่ง List ของไฟล์ภาพทั้งหมด (รองรับสูงสุด 3 ภาพตามสเปก 8B MM-DiT)
+            files = [handle_file(img.name if hasattr(img, 'name') else img) for img in images]
+            result = client.predict(images=files, prompt=prompt, api_name="/multi_edit")
         else:
-            # โหมดทั่วไป: Single Edit, Upscale, Watermark
-            result = client.predict(image=handle_file(image), prompt=prompt, api_name=mode_api_name)
+            # Single Edit, Upscale, Watermark
+            result = client.predict(image=handle_file(images), prompt=prompt, api_name=mode_api_name)
             
-        return result, f"Status: Processed via {mode_api_name}"
+        return result, f"Status: Success ({mode_api_name})"
     except Exception as e:
         return None, f"Remote Error: {str(e)}"
 
-# UI Layout แบบไม่มั่ว (Row/Column สัดส่วนชัดเจน)
 with gr.Blocks() as demo:
-    gr.HTML("<h1 style='text-align: center;'>🎨 Omni Editor 2.0 (Full Remote)</h1>")
+    gr.HTML("<h1 style='text-align: center;'>🎨 Omni Editor 2.0 (Remote)</h1>")
     
-    with gr.Tabs() as tabs:
+    with gr.Tabs():
         # 1. Single Image Edit
         with gr.TabItem("🖼️ Single Image Edit"):
             with gr.Row():
                 with gr.Column(scale=1):
                     s_in = gr.Image(type="filepath", label="Input Image")
-                    s_prompt = gr.Textbox(label="Prompt", placeholder="e.g. Change background to Mars")
+                    s_prompt = gr.Textbox(label="Prompt", placeholder="Describe your edit...")
                     s_run = gr.Button("Execute Edit", variant="primary")
                 with gr.Column(scale=1):
                     s_out = gr.Image(label="Result")
                     s_status = gr.Textbox(label="Status", interactive=False)
             s_run.click(lambda img, txt: call_remote_api(img, txt, "/predict"), [s_in, s_prompt], [s_out, s_status])
 
-        # 2. Multi-Image Edit
+        # 2. Multi-Image Edit (แก้ไขจุดที่มักง่ายแล้ว: รับได้หลายรูป)
         with gr.TabItem("🖼️🖼️ Multi-Image Edit"):
-            gr.Markdown("Select multiple images to edit (Handled by 8B MM-DiT)")
+            gr.Markdown("### 🚀 Multi-Modal Fusion (Select up to 3 images)")
             with gr.Row():
-                with gr.Column():
-                    m_in = gr.Image(type="filepath", label="Reference Image")
-                    m_prompt = gr.Textbox(label="Instruction")
+                with gr.Column(scale=1):
+                    # ใช้ File component แบบ multiple เพื่อให้เลือกได้หลายรูปพร้อมกัน
+                    m_in = gr.File(label="Upload Images (Base/Ref/Style)", file_count="multiple", file_types=["image"])
+                    m_prompt = gr.Textbox(label="Instruction", placeholder="How should these images interact?")
                     m_run = gr.Button("Execute Multi-Edit", variant="primary")
-                with gr.Column():
+                with gr.Column(scale=1):
                     m_out = gr.Image(label="Result")
-            m_run.click(lambda img, txt: call_remote_api(img, txt, "/multi_edit"), [m_in, m_prompt], [m_out, s_status])
+                    m_status = gr.Textbox(label="Status", interactive=False)
+            m_run.click(call_remote_api, [m_in, m_prompt, gr.State("/multi_edit")], [m_out, m_status])
 
         # 3. Text to Image
         with gr.TabItem("✨ Text to Image"):
             with gr.Row():
                 with gr.Column():
-                    t_prompt = gr.Textbox(label="What do you want to create?")
-                    t_run = gr.Button("Generate Image", variant="primary")
+                    t_prompt = gr.Textbox(label="Describe the image you want")
+                    t_run = gr.Button("Generate", variant="primary")
                 with gr.Column():
-                    t_out = gr.Image(label="Generated Result")
+                    t_out = gr.Image(label="Result")
             t_run.click(lambda txt: call_remote_api(None, txt, "/t2i"), [t_prompt], [t_out, s_status])
 
         # 4. Image Upscale & Watermark
-        with gr.TabItem("🔍 Tools (Upscale/Watermark)"):
+        with gr.TabItem("🔍 Tools"):
             tool_mode = gr.Radio(["Upscale", "Remove Watermark"], label="Select Tool", value="Upscale")
             with gr.Row():
                 with gr.Column():
                     tool_in = gr.Image(type="filepath", label="Input")
                     tool_run = gr.Button("Process", variant="primary")
                 with gr.Column():
-                    tool_out = gr.Image(label="Processed Result")
+                    tool_out = gr.Image(label="Result")
             
-            # เลือก API name ตาม Radio
             def tool_handler(mode, img):
                 api_name = "/upscale" if mode == "Upscale" else "/remove_watermark"
                 return call_remote_api(img, "", api_name)
             
             tool_run.click(tool_handler, [tool_mode, tool_in], [tool_out, s_status])
 
-    gr.Markdown("---")
-    gr.Markdown("🤖 **Omni Creator 2.0:** 8B Unified Multi-Modal Diffusion Transformer")
-
-demo.queue().launch(server_name="0.0.0.0")
+demo.queue().launch()