Add prompt input field

app.py

@@ -350,13 +350,16 @@ def download_video(url: str) -> str:
         return None
 
 
-def process(api_key: str, image_path: str):
+def process(api_key: str, image_path: str, prompt: str):
     if not api_key or not api_key.strip():
         gr.Warning("Please enter your API Key")
         return None
     if not image_path:
         gr.Warning("Please upload an image")
         return None
+    if not prompt or not prompt.strip():
+        gr.Warning("Please enter a prompt")
+        return None
 
     try:
         gr.Info("Uploading image...")
@@ -364,7 +367,10 @@ def process(api_key: str, image_path: str):
 
         gr.Info("Processing...")
         payload = {
-            "image": image_url,
+            "images": [image_url],
+            "prompt": prompt.strip(),
+            "enable_base64_output": False,
+            "enable_sync_mode": False,
         }
         result = call_api(api_key, MODEL_ENDPOINT, payload)
         request_id = result.get("id")
@@ -454,6 +460,11 @@ with gr.Blocks(css=CUSTOM_CSS, title="Seedream v4 Edit - WaveSpeed") as demo:
                     source="upload",
                     elem_classes=["upload-area"]
                 )
+                prompt_input = gr.Textbox(
+                    label="Prompt",
+                    placeholder="Describe how to edit the image...",
+                    lines=3
+                )
                 submit_btn = gr.Button("Process", variant="primary", elem_classes="primary-btn")
 
             # Right Column - Output
@@ -485,7 +496,7 @@ with gr.Blocks(css=CUSTOM_CSS, title="Seedream v4 Edit - WaveSpeed") as demo:
     # Event binding
     submit_btn.click(
         fn=process,
-        inputs=[api_key_input, image_input],
+        inputs=[api_key_input, image_input, prompt_input],
         outputs=output_image,
     )