Update app.py

app.py

@@ -49,7 +49,7 @@ def initialize_sdk_config():
 def generate_video(input_api_key, prompt_text, image_url, model_id, progress=gr.Progress()):
     """Generate video using the BytePlus SDK"""
     
-    # Use input API key if provided, otherwise fall back to environment variable
+    # Use input API key if provided and not "key", otherwise fall back to environment variable
     effective_api_key = input_api_key if input_api_key and input_api_key != "key" else api_key
     
     if not effective_api_key:
@@ -171,7 +171,7 @@ custom_css = """
 """
 
 # Create Gradio interface
-with gr.Blocks(title="BytePlus Video Generator") as demo:
+with gr.Blocks(title="BytePlus Video Generator", fill_width=True) as demo:
     
     gr.Markdown("""
     # 🎥 BytePlus Video Generator
@@ -180,7 +180,7 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
     
     # Show API key source
     if api_key:
-        gr.Markdown("""
+        gr.Markdown(f"""
         <div class="env-badge">
         ✅ API Key loaded from environment variable 'Key'
         </div>
@@ -208,13 +208,8 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
                 label="🔑 BytePlus API Key",
                 placeholder="Enter your BytePlus API key here",
                 type="password",
-                value="key" if not api_key else "",  # Empty if env var exists
-                info=f"Using env var 'Key' by default. Enter only to override."
-            )
-            
-            # Show if env var is being used
-            env_status = gr.Markdown(
-                f"*Using environment variable 'Key'*" if api_key else "*Enter API key above*"
+                value="" if api_key else "key",  # Empty if env var exists, otherwise "key"
+                info="Using env var 'Key' by default. Enter only to override."
             )
             
             # Model ID
@@ -248,7 +243,7 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
             )
             
             # Generate button
-            generate_btn = gr.Button("🚀 Generate Video", variant="primary")
+            generate_btn = gr.Button("🚀 Generate Video", variant="primary", size="lg")
         
         with gr.Column(scale=1):
             # Status output
@@ -257,11 +252,10 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
                 lines=5
             )
             
-            # Video output
+            # Video output - REMOVED the unsupported parameters
             video = gr.Video(
-                label="🎬 Generated Video",
-                show_download_button=True,
-                show_share_button=True
+                label="🎬 Generated Video"
+                # show_download_button and show_share_button removed as they're not in Gradio 6.5.1
             )
             
             # Video URL
@@ -269,15 +263,23 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
                 label="🔗 Video URL",
                 interactive=False
             )
+            
+            # Add download button separately if needed
+            with gr.Row():
+                download_btn = gr.DownloadButton(
+                    label="📥 Download Video",
+                    variant="secondary",
+                    visible=False  # Hidden until video is generated
+                )
     
     # Example prompts
     gr.Markdown("---")
     gr.Markdown("## 📋 Example Prompts")
     
     with gr.Row():
-        example1 = gr.Button("🌄 Nature Drone")
-        example2 = gr.Button("🏙️ City Racing")
-        example3 = gr.Button("🌊 Ocean Waves")
+        example1 = gr.Button("🌄 Nature Drone", size="sm")
+        example2 = gr.Button("🏙️ City Racing", size="sm")
+        example3 = gr.Button("🌊 Ocean Waves", size="sm")
     
     # Function to update image preview
     def update_preview(url):
@@ -290,7 +292,8 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
                 img = Image.open(BytesIO(response.content))
                 return img
             return None
-        except:
+        except Exception as e:
+            logger.error(f"Preview error: {e}")
             return None
     
     # Example prompt functions
@@ -303,6 +306,12 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
     def set_ocean():
         return "Drone following surfers riding massive waves, slow motion, dramatic ocean views, golden hour --duration 5 --camerafixed false"
     
+    # Function to update download button when video is generated
+    def update_download_button(video_path):
+        if video_path and video_path != "None" and video_path != "":
+            return gr.update(visible=True, value=video_path)
+        return gr.update(visible=False)
+    
     # Connect events
     image_url.change(
         fn=update_preview,
@@ -321,25 +330,30 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
         outputs=[status, video]
     )
     
-    # Update video URL
+    # Update video URL and download button
     generate_event.then(
         fn=lambda url: url if url else "",
         inputs=[video],
         outputs=[video_url]
+    ).then(
+        fn=update_download_button,
+        inputs=[video],
+        outputs=[download_btn]
     )
     
     # Clear button
     clear_btn = gr.Button("🗑️ Clear", variant="secondary")
     clear_btn.click(
         fn=lambda: (
-            "key" if not api_key else "",  # Reset API key input
+            "" if api_key else "key",  # Reset API key input
             "https://ark-doc.tos-ap-southeast-1.bytepluses.com/seepro_i2v%20.png",
             "Enter your prompt here...",
             "",
             None,
-            ""
+            "",
+            gr.update(visible=False)
         ),
-        outputs=[api_key_input, image_url, prompt, status, video, video_url]
+        outputs=[api_key_input, image_url, prompt, status, video, video_url, download_btn]
     )
     
     # Footer
@@ -354,6 +368,7 @@ with gr.Blocks(title="BytePlus Video Generator") as demo:
 if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
+        server_port=7860,
         theme=gr.themes.Soft(
             primary_hue="blue",
             secondary_hue="purple",