Update app.py

app.py

@@ -10,14 +10,17 @@ import sys
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Print versions for debugging
+logger.info(f"Gradio version: {gr.__version__}")
+logger.info(f"Python version: {sys.version}")
 
-# Suno API key
+# Get API key from environment variable "Key"
 api_key = os.environ.get("Key", "")
 if not api_key:
-    print("⚠️ SunoKey not set!")
-    
-# Print Gradio version for debugging
-logger.info(f"Gradio version: {gr.__version__}")
+    logger.warning("⚠️ API Key not set in environment variable 'Key'!")
+    logger.info("Will use input field instead")
+else:
+    logger.info("✅ API Key loaded from environment")
 
 # Import BytePlus SDK
 try:
@@ -29,27 +32,50 @@ except ImportError as e:
     SDK_AVAILABLE = False
     logger.warning(f"⚠️ BytePlus SDK not available: {e}")
 
-def generate_video(api_key, prompt_text, image_url, model_id, progress=gr.Progress()):
+def initialize_sdk_config():
+    """Initialize SDK configuration as per docs"""
+    try:
+        configuration = byteplussdkcore.Configuration()
+        configuration.client_side_validation = True
+        configuration.schema = "http"
+        configuration.debug = False
+        configuration.logger_file = "sdk.log"
+        byteplussdkcore.Configuration.set_default(configuration)
+        return True
+    except Exception as e:
+        logger.error(f"SDK config error: {e}")
+        return False
+
+def generate_video(input_api_key, prompt_text, image_url, model_id, progress=gr.Progress()):
     """Generate video using the BytePlus SDK"""
     
-    if not api_key or api_key == "key":
-        yield "⚠️ Please enter your actual BytePlus API key (the 'key' is just a placeholder)", None
+    # Use input API key if provided, 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:
+        yield "⚠️ Please enter your BytePlus API key or set it in the 'Key' environment variable", None
         return
     
     if not SDK_AVAILABLE:
-        yield "❌ BytePlus SDK not available. Please check installation of byteplus-python-sdk-v2", None
+        yield "❌ BytePlus SDK not available. Please check installation.", None
         return
     
     try:
         progress(0, desc="Initializing SDK...")
         
-        # Set environment variable
-        os.environ["Key"] = api_key
+        # Initialize SDK config
+        try:
+            initialize_sdk_config()
+        except:
+            pass  # Continue even if config fails
+        
+        # Set environment variable for SDK (it expects ARK_API_KEY)
+        os.environ["ARK_API_KEY"] = effective_api_key
         
         # Initialize client
         client = Ark(
             base_url="https://ark.ap-southeast.bytepluses.com/api/v3",
-            api_key=os.environ.get("Key"),
+            api_key=os.environ.get("ARK_API_KEY"),
         )
         
         progress(0.1, desc="Creating video generation request...")
@@ -124,54 +150,213 @@ def generate_video(api_key, prompt_text, image_url, model_id, progress=gr.Progre
         logger.error(f"Error: {e}")
         yield f"❌ Error: {str(e)}", None
 
-# Simple, clean UI
-with gr.Blocks(title="BytePlus Video Generator", theme=gr.themes.Soft()) as demo:
+# Custom CSS
+custom_css = """
+.gradio-container {
+    max-width: 1200px !important;
+    margin: auto !important;
+}
+.video-container {
+    border-radius: 8px;
+    overflow: hidden;
+}
+.env-badge {
+    background: #e3f2fd;
+    color: #0d47a1;
+    padding: 5px 10px;
+    border-radius: 4px;
+    font-size: 0.9em;
+    margin-bottom: 10px;
+}
+"""
+
+# Create Gradio interface
+with gr.Blocks(title="BytePlus Video Generator") as demo:
+    
     gr.Markdown("""
     # 🎥 BytePlus Video Generator
-    Generate videos from images and text prompts
+    Generate stunning videos from images and text prompts using BytePlus AI
     """)
     
+    # Show API key source
+    if api_key:
+        gr.Markdown("""
+        <div class="env-badge">
+        ✅ API Key loaded from environment variable 'Key'
+        </div>
+        """)
+    else:
+        gr.Markdown("""
+        <div class="env-badge" style="background: #fff3e0; color: #bf360c;">
+        ⚠️ No API Key in environment variable 'Key'. Please enter below.
+        </div>
+        """)
+    
+    # SDK Status
     with gr.Row():
-        with gr.Column():
-            api_key = gr.Textbox(
-                label="🔑 API Key",
+        if SDK_AVAILABLE:
+            gr.Markdown("✅ **SDK Status:** Connected to BytePlus SDK")
+        else:
+            gr.Markdown("❌ **SDK Status:** SDK not available")
+    
+    gr.Markdown("---")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            # API Key input - will use env var as default if available
+            api_key_input = gr.Textbox(
+                label="🔑 BytePlus API Key",
+                placeholder="Enter your BytePlus API key here",
                 type="password",
-                placeholder="Enter your BytePlus API key",
-                value="key"
+                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*"
             )
             
+            # Model ID
             model_id = gr.Textbox(
                 label="🤖 Model ID",
-                value="seedance-1-5-pro-251215"
+                value="seedance-1-5-pro-251215",
+                info="Model ID for video generation"
             )
             
+            # Image URL input
             image_url = gr.Textbox(
-                label="🖼️ Image URL",
-                value="https://ark-doc.tos-ap-southeast-1.bytepluses.com/seepro_i2v%20.png"
+                label="🔗 Image URL",
+                placeholder="Enter public image URL",
+                value="https://ark-doc.tos-ap-southeast-1.bytepluses.com/seepro_i2v%20.png",
+                info="Image must be publicly accessible"
             )
             
+            # Image preview
+            image_preview = gr.Image(
+                label="Image Preview",
+                type="pil",
+                height=200
+            )
+            
+            # Text prompt
             prompt = gr.Textbox(
-                label="📝 Prompt",
-                lines=3,
-                value="At breakneck speed, drones thread through intricate obstacles --duration 5 --camerafixed false"
+                label="📝 Text Prompt",
+                placeholder="Describe your video...",
+                value="At breakneck speed, drones thread through intricate obstacles or stunning natural wonders, delivering an immersive, heart-pounding flying experience. --duration 5 --camerafixed false",
+                lines=4
             )
             
-            generate_btn = gr.Button("🚀 Generate", variant="primary")
+            # Generate button
+            generate_btn = gr.Button("🚀 Generate Video", variant="primary")
         
-        with gr.Column():
-            status = gr.Textbox(label="Status", lines=3)
-            video = gr.Video(label="Generated Video")
-            video_url = gr.Textbox(label="Video URL")
+        with gr.Column(scale=1):
+            # Status output
+            status = gr.Textbox(
+                label="📊 Generation Status",
+                lines=5
+            )
+            
+            # Video output
+            video = gr.Video(
+                label="🎬 Generated Video",
+                show_download_button=True,
+                show_share_button=True
+            )
+            
+            # Video URL
+            video_url = gr.Textbox(
+                label="🔗 Video URL",
+                interactive=False
+            )
+    
+    # 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")
+    
+    # Function to update image preview
+    def update_preview(url):
+        try:
+            if url and url.startswith(('http://', 'https://')):
+                import requests
+                from io import BytesIO
+                
+                response = requests.get(url, timeout=5)
+                img = Image.open(BytesIO(response.content))
+                return img
+            return None
+        except:
+            return None
+    
+    # Example prompt functions
+    def set_nature():
+        return "Aerial drone shot flying over majestic mountains at sunrise, cinematic lighting, smooth motion --duration 5 --camerafixed false"
+    
+    def set_city():
+        return "Fast-paced drone racing through futuristic city streets with neon lights, dynamic angles, high speed --duration 5 --camerafixed false"
+    
+    def set_ocean():
+        return "Drone following surfers riding massive waves, slow motion, dramatic ocean views, golden hour --duration 5 --camerafixed false"
     
-    generate_btn.click(
+    # Connect events
+    image_url.change(
+        fn=update_preview,
+        inputs=image_url,
+        outputs=image_preview
+    )
+    
+    example1.click(fn=set_nature, outputs=prompt)
+    example2.click(fn=set_city, outputs=prompt)
+    example3.click(fn=set_ocean, outputs=prompt)
+    
+    # Generate
+    generate_event = generate_btn.click(
         fn=generate_video,
-        inputs=[api_key, prompt, image_url, model_id],
+        inputs=[api_key_input, prompt, image_url, model_id],
         outputs=[status, video]
-    ).then(
+    )
+    
+    # Update video URL
+    generate_event.then(
         fn=lambda url: url if url else "",
         inputs=[video],
         outputs=[video_url]
     )
+    
+    # Clear button
+    clear_btn = gr.Button("🗑️ Clear", variant="secondary")
+    clear_btn.click(
+        fn=lambda: (
+            "key" if not api_key else "",  # Reset API key input
+            "https://ark-doc.tos-ap-southeast-1.bytepluses.com/seepro_i2v%20.png",
+            "Enter your prompt here...",
+            "",
+            None,
+            ""
+        ),
+        outputs=[api_key_input, image_url, prompt, status, video, video_url]
+    )
+    
+    # Footer
+    gr.Markdown("---")
+    gr.Markdown("""
+    <div style='text-align: center'>
+        <p>Powered by BytePlus SDK | Gradio 6.5.1</p>
+        <p style='font-size: 0.8em; color: gray;'>API key sourced from environment variable 'Key'</p>
+    </div>
+    """)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        server_name="0.0.0.0",
+        theme=gr.themes.Soft(
+            primary_hue="blue",
+            secondary_hue="purple",
+        ),
+        css=custom_css
+    )