Update app.py

app.py

@@ -99,6 +99,7 @@ def image_to_base64(file_path):
 def generate_video(
     context_scale,
     enable_safety_checker,
+    enable_fast_mode,
     flow_shift,
     guidance_scale,
     images,
@@ -120,7 +121,7 @@ def generate_video(
 
     if not rate_limiter.check(session_id):
         error_img = create_error_image("每小时限制20次请求")
-        yield "❌ 请求过于频繁，请稍后再试", error_img
+        yield "❌ rate limit exceeded", error_img
         return
 
     session = session_manager.get_session(session_id)
@@ -129,13 +130,13 @@ def generate_video(
 
     API_KEY = os.getenv("WAVESPEED_API_KEY")
     if not API_KEY:
-        error_img = create_error_image("API密钥缺失")
+        error_img = create_error_image("API key not found")
         yield "❌ Error: Missing API Key", error_img
         return
 
     try:
         if not images or len(images) < 2:
-            raise ValueError("需要上传至少两张图片")
+            raise ValueError("must provide at least 2 images")
             
         base64_images = []
         for img_path in images[:2]:
@@ -144,7 +145,7 @@ def generate_video(
             
     except Exception as e:
         error_img = create_error_image(str(e))
-        yield f"❌ 文件处理失败: {str(e)}", error_img
+        yield f"❌failed to upload images: {str(e)}", error_img
         return
 
     video_payload = ""
@@ -156,7 +157,7 @@ def generate_video(
 
     payload = {
     "context_scale": context_scale,
-    "enable_fast_mode": False,
+    "enable_fast_mode": enable_fast_mode,
     "enable_safety_checker": enable_safety_checker,
     "flow_shift": flow_shift,
     "guidance_scale": guidance_scale,
@@ -170,7 +171,7 @@ def generate_video(
     "video": str(video_payload) if video_payload else "",
     }
 
-    logging.debug(f"API请求payload: {json.dumps(payload, indent=2)}")
+    logging.debug(f"API request payload: {json.dumps(payload, indent=2)}")
 
     headers = {
         "Content-Type": "application/json",
@@ -187,14 +188,14 @@ def generate_video(
         
         if response.status_code != 200:
             error_img = create_error_image(response.text)
-            yield f"❌ API错误 ({response.status_code}): {response.text}", error_img
+            yield f"❌ API Error ({response.status_code}): {response.text}", error_img
             return
             
         request_id = response.json()["data"]["id"]
-        yield f"✅ 任务已提交 (ID: {request_id})", None
+        yield f"✅ Task ID (ID: {request_id})", None
     except Exception as e:
         error_img = create_error_image(str(e))
-        yield f"❌ 连接错误: {str(e)}", error_img
+        yield f"❌ Connection Error: {str(e)}", error_img
         return
 
     result_url = f"https://api.wavespeed.ai/api/v2/predictions/{request_id}/result"
@@ -246,6 +247,11 @@ with gr.Blocks(
     .safe { background: #e8f5e9; border: 1px solid #a5d6a7; }
     .warning { background: #fff3e0; border: 1px solid #ffcc80; }
     .error { background: #ffebee; border: 1px solid #ef9a9a; }
+    #centered_button {
+    align-self: center !important;
+    height: fit-content !important;
+    margin-top: 22px !important;  # 根据输入框高度微调
+}
     """
 ) as app:
     
@@ -256,19 +262,28 @@ with gr.Blocks(
 
     with gr.Row():
         with gr.Column(scale=1):
-            images = gr.File(label="upload image", file_count="multiple",  file_types=["image"], type="filepath", elem_id="image-uploader")
-            video = gr.Video(label="Input Video", format="mp4", sources=["upload"])
-            prompt = gr.Textbox(label="Prompt", lines=5, placeholder="Prompt...")
+            with gr.Row():
+                images = gr.File(label="upload image", file_count="multiple",  file_types=["image"], type="filepath", elem_id="image-uploader",
+            scale=1)
+                video = gr.Video(label="Input Video", format="mp4", sources=["upload"],
+            scale=1)
+            prompt = gr.Textbox(label="Prompt", lines=5, placeholder="Prompt")
             negative_prompt = gr.Textbox(label="Negative Prompt", lines=2)
-            size = gr.Dropdown(["832*480", "480*832"], value="832*480", label="Size")
-            context_scale = gr.Slider(0, 2, value=1, step=0.1, label="Context Scale")
-            num_inference_steps = gr.Slider(1, 100, value=20, step=1, label="Inference Steps")
-            task = gr.Dropdown(["depth", "pose"], value="depth", label="Task") 
-            seed = gr.Number(-1, label="Seed")
-            random_seed_btn = gr.Button("Random🎲Seed", variant="secondary")
-            guidance = gr.Slider(1, 20, value=7.5, step=0.1, label="Guidance_Scale")
-            flow_shift = gr.Slider(1, 20, value=16, step=1, label="Shift")
-            enable_safety_checker = gr.Checkbox(True, label="Enable Safety Checker", interactive=True)
+            with gr.Row():
+                size = gr.Dropdown(["832*480", "480*832"], value="832*480", label="Size")
+                task = gr.Dropdown(["depth", "pose"], value="depth", label="Task")
+            with gr.Row():
+                num_inference_steps = gr.Slider(1, 100, value=20, step=1, label="Inference Steps")
+                context_scale = gr.Slider(0, 2, value=1, step=0.1, label="Context Scale")
+            with gr.Row():
+                guidance = gr.Slider(1, 20, value=7.5, step=0.1, label="Guidance_Scale")
+                flow_shift = gr.Slider(1, 20, value=16, step=1, label="Shift")
+            with gr.Row(): 
+                seed = gr.Number(-1, label="Seed")
+                random_seed_btn = gr.Button("Random🎲Seed", variant="secondary", elem_id="centered_button")
+            with gr.Row():
+                enable_safety_checker = gr.Checkbox(True, label="Enable Safety Checker", interactive=True)
+                enable_fast_mode = gr.Checkbox(False, label="To enable the fast mode, please visit Wave Speed AI", interactive=False)
         with gr.Column(scale=1):
             video_output = gr.Video(label="Video Output", format="mp4", interactive=False,  elem_classes=["video-preview"])
             generate_btn = gr.Button("Generate", variant="primary")
@@ -296,6 +311,7 @@ with gr.Blocks(
     inputs=[
         context_scale,
         enable_safety_checker, 
+        enable_fast_mode,
         flow_shift,
         guidance,
         images,