Fix video input handling

app.py

@@ -350,15 +350,36 @@ def download_video(url: str) -> str:
         return None
 
 
-def process(api_key: str):
+def upload_video(api_key: str, file_path: str) -> str:
+    """Upload video and return URL"""
+    headers = {"Authorization": f"Bearer {api_key.strip()}"}
+    with open(file_path, "rb") as f:
+        resp = requests.post(UPLOAD_ENDPOINT, headers=headers, files={"file": f}, timeout=120)
+    if resp.status_code == 401:
+        raise Exception("Invalid API Key")
+    elif resp.status_code >= 400:
+        raise Exception(f"Upload failed: {resp.status_code}")
+    data = resp.json()
+    if data.get("code") != 200:
+        raise Exception(data.get("message", "Upload failed"))
+    return data.get("data", {}).get("download_url")
+
+
+def process(api_key: str, video_path: str):
     if not api_key or not api_key.strip():
         gr.Warning("Please enter your API Key")
         return None
+    if not video_path:
+        gr.Warning("Please upload a video")
+        return None
 
     try:
-        gr.Info("Generating video...")
+        gr.Info("Uploading video...")
+        video_url = upload_video(api_key, video_path)
+
+        gr.Info("Removing watermark...")
         payload = {
-            ,
+            "video": video_url,
         }
         result = call_api(api_key, MODEL_ENDPOINT, payload)
         request_id = result.get("id")
@@ -442,11 +463,14 @@ with gr.Blocks(css=CUSTOM_CSS, title="Video Watermark Remover - WaveSpeed") as d
                 gr.HTML("""
                     <div class="section-title">
                         <svg width="20" height="20" fill="#8b5cf6" viewBox="0 0 20 20"><path fill-rule="evenodd" d="M4 3a2 2 0 00-2 2v10a2 2 0 002 2h12a2 2 0 002-2V5a2 2 0 00-2-2H4zm12 12H4l4-8 3 6 2-4 3 6z" clip-rule="evenodd"/></svg>
-                        Enter Prompt
+                        Upload Video
                     </div>
                 """)
-                
-                submit_btn = gr.Button("Process", variant="primary", elem_classes="primary-btn")
+                input_video = gr.Video(
+                    label="",
+                    sources=["upload"]
+                )
+                submit_btn = gr.Button("Remove Watermark", variant="primary", elem_classes="primary-btn")
 
             # Right Column - Output
             with gr.Column(scale=1):
@@ -475,7 +499,7 @@ with gr.Blocks(css=CUSTOM_CSS, title="Video Watermark Remover - WaveSpeed") as d
     # Event binding
     submit_btn.click(
         fn=process,
-        inputs=[api_key_input],
+        inputs=[api_key_input, input_video],
         outputs=output_video,
     )