Update app.py

app.py

@@ -80,7 +80,7 @@ def create_error_image(message):
 
 def classify_prompt(prompt):
     inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
-    with torch.no_grad():
+   with torch.no_grad():
         outputs = model(**inputs)
     return torch.argmax(outputs.logits).item()
 
@@ -96,6 +96,21 @@ def image_to_base64(file_path):
     except Exception as e:
         raise ValueError(f"Base64编码失败: {str(e)}")
 
+def video_to_base64(file_path):
+    """
+    将视频文件转换为Base64格式
+    """
+    try:
+        with open(file_path, "rb") as video_file:
+            raw_data = video_file.read()
+            encoded = base64.b64encode(raw_data)
+            missing_padding = len(encoded) % 4
+            if missing_padding:
+                encoded += b'=' * (4 - missing_padding)
+            return encoded.decode('utf-8')
+    except Exception as e:
+        raise ValueError(f"Base64编码失败: {str(e)}")
+
 def generate_video(
     context_scale,
     enable_safety_checker,
@@ -128,21 +143,18 @@ def generate_video(
     session['last_active'] = time.time()
     session['count'] += 1
 
-    API_KEY = os.getenv("WAVESPEED_API_KEY")
+    API_KEY = "30a09de38569400bcdab9cec1c9a660b1924a2b5f54aa386eeb87f96a112fb93"
     if not API_KEY:
         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("must provide at least 2 images")
-            
         base64_images = []
-        for img_path in images[:2]:
-            base64_img = image_to_base64(img_path)
-            base64_images.append(base64_img)
-            
+        if images is not None:  # 检查 images 是否为 None
+            for img_path in images:
+                base64_img = image_to_base64(img_path)
+                base64_images.append(base64_img)
     except Exception as e:
         error_img = create_error_image(str(e))
         yield f"❌failed to upload images: {str(e)}", error_img
@@ -155,6 +167,15 @@ def generate_video(
         else:
             video_payload = video
 
+        # 将视频文件转换为Base64格式
+        try:
+            base64_video = video_to_base64(video_payload)
+            video_payload = base64_video
+        except Exception as e:
+            error_img = create_error_image(str(e))
+            yield f"❌ Failed to encode video: {str(e)}", error_img
+            return
+
     payload = {
     "context_scale": context_scale,
     "enable_fast_mode": enable_fast_mode,
@@ -283,23 +304,23 @@ with gr.Blocks(
                 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)
+                enable_fast_mode = gr.Checkbox(True, 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")
             status_output = gr.Textbox(label="status", interactive=False, lines=4)
             gr.Examples(
-                examples=[
+               examples=[
+                  [
+                     "The elegant lady carefully selects bags in the boutique, and she shows the charm of a mature woman in a black slim dress with a pearl necklace, as well as her pretty face. Holding a vintage-inspired blue leather half-moon handbag, she is carefully observing its craftsmanship and texture. The interior of the store is a haven of sophistication and luxury. Soft, ambient lighting casts a warm glow over the polished wooden floors",
                     [
-                        "The elegant lady carefully selects bags in the boutique, and she shows the charm of a mature woman in a black slim dress with a pearl necklace, as well as her pretty face. Holding a vintage-inspired blue leather half-moon handbag, she is carefully observing its craftsmanship and texture. The interior of the store is a haven of sophistication and luxury. Soft, ambient lighting casts a warm glow over the polished wooden floors",
-                        [
-                            "https://d2g64w682n9w0w.cloudfront.net/media/ec44bbf6abac4c25998dd2c4af1a46a7/images/1747413751234102420_md9ywspl.png",
-                            "https://d2g64w682n9w0w.cloudfront.net/media/ec44bbf6abac4c25998dd2c4af1a46a7/images/1747413586520964413_7bkgc9ol.png"
-                        ]
+                       "https://d2g64w682n9w0w.cloudfront.net/media/ec44bbf6abac4c25998dd2c4af1a46a7/images/1747413751234102420_md9ywspl.png",
+                      "https://d2g64w682n9w0w.cloudfront.net/media/ec44bbf6abac4c25998dd2c4af1a46a7/images/1747413586520964413_7bkgc9ol.png"
                     ]
+                  ]
                 ],
-                inputs=[prompt, images],
-                )
+                 inputs=[prompt, images],
+                 )
 
     random_seed_btn.click(
         fn=lambda: random.randint(0, 999999),
@@ -344,5 +365,6 @@ if __name__ == "__main__":
     app.queue(max_size=4).launch(
         server_name="0.0.0.0",
         max_threads=16,
+        server_port=8009,
         share=False
     )