Update a.py

a.py

@@ -2,18 +2,55 @@ import os
 import time
 import gradio as gr
 from byteplussdkarkruntime import Ark
+import httpx
+import socket
 
 # Get API key from Hugging Face secret "Key"
 API_KEY = os.environ.get("Key", "")
 
-# Initialize client exactly like your working code
+# Initialize client with better timeout and retry settings
 client = Ark(
     base_url="https://ark.ap-southeast.bytepluses.com/api/v3",
     api_key=API_KEY,
+    timeout=30.0,  # Shorter timeout to fail faster
+    max_retries=3,
 )
 
+# Fresh new prompts
+DEFAULT_PROMPTS = {
+    "Cinematic Nature": "Majestic drone shot soaring above misty mountains at golden hour, sunlight breaking through clouds, cinematic 4k quality, smooth motion --duration 5 --camerafixed false",
+    
+    "Urban Exploration": "Dynamic drone flight through narrow alleyways of a neon-lit Tokyo street at night, rain-slicked surfaces reflecting lights, cyberpunk aesthetic --duration 5 --camerafixed false",
+    
+    "Action Sequence": "High-speed drone chasing a sports car through a winding coastal road, dramatic cliffside views, action movie style --duration 5 --camerafixed false",
+    
+    "Abstract Art": "Surreal drone flight through floating geometric shapes in a dreamlike void, pastel colors, ethereal atmosphere --duration 5 --camerafixed false",
+    
+    "Wildlife Documentary": "Drone following a herd of elephants across the African savanna at sunset, National Geographic style, majestic wildlife cinematography --duration 5 --camerafixed false",
+    
+    "Sports Highlight": "Drone racing alongside professional skiers carving through fresh powder in the Alps, high-energy sports broadcast style --duration 5 --camerafixed false",
+    
+    "Beach Paradise": "Drone gliding over turquoise waters and white sand beaches, palm trees swaying, tropical paradise aerial view --duration 5 --camerafixed false",
+    
+    "Ancient Temple": "Drone circling around ancient Angkor Wat temple at dawn, mystical atmosphere, historical documentary style --duration 5 --camerafixed false"
+}
+
+def test_connection():
+    """Test if we can reach the API before trying"""
+    try:
+        socket.gethostbyname('ark.ap-southeast.bytepluses.com')
+        return True, "DNS resolved"
+    except Exception as e:
+        return False, f"DNS failed: {e}"
+
 def generate_video(image_path, prompt_text, progress=gr.Progress()):
-    """Generate video using HF temp URL trick"""
+    """Generate video with robust error handling"""
+    
+    # Test connection first
+    conn_ok, conn_msg = test_connection()
+    if not conn_ok:
+        yield f"❌ Connection Error: {conn_msg}. This might be a network restriction from Hugging Face.", None
+        return
     
     if not API_KEY:
         yield "❌ API Key not configured. Please add 'Key' secret.", None
@@ -26,8 +63,7 @@ def generate_video(image_path, prompt_text, progress=gr.Progress()):
     try:
         progress(0, desc="Preparing image...")
         
-        # THE TRICK: image_path is already the file path string
-        # HF automatically serves it as /file=path
+        # Get HF temp URL
         image_url = f"/file={image_path}"
         print(f"Using image URL: {image_url}")
         
@@ -35,23 +71,30 @@ def generate_video(image_path, prompt_text, progress=gr.Progress()):
         
         progress(0.2, desc="Creating request...")
         
-        # Exactly your working code structure
-        print("----- create request -----")
-        create_result = client.content_generation.tasks.create(
-            model="seedance-1-5-pro-251215",
-            content=[
-                {
-                    "type": "text",
-                    "text": prompt_text
-                },
-                {
-                    "type": "image_url",
-                    "image_url": {
-                        "url": image_url  # Using HF temp URL!
+        # Create task with timeout
+        try:
+            create_result = client.content_generation.tasks.create(
+                model="seedance-1-5-pro-251215",
+                content=[
+                    {
+                        "type": "text",
+                        "text": prompt_text
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": image_url
+                        }
                     }
-                }
-            ]
-        )
+                ]
+            )
+        except Exception as e:
+            error_str = str(e)
+            if "Connection error" in error_str or "202602" in error_str:
+                yield f"❌ Network Error: Cannot reach BytePlus API. This is likely a Hugging Face network restriction.", None
+            else:
+                yield f"❌ API Error: {error_str}", None
+            return
         
         task_id = create_result.id
         print(f"Task created: {task_id}")
@@ -59,68 +102,107 @@ def generate_video(image_path, prompt_text, progress=gr.Progress()):
         
         progress(0.3, desc="Polling for results...")
         
-        # Polling exactly like your working code
+        # Poll for results
         attempts = 0
-        max_attempts = 120  # 2 minutes max
+        max_attempts = 120
         
         while attempts < max_attempts:
-            get_result = client.content_generation.tasks.get(task_id=task_id)
-            status = get_result.status
-            
-            if status == "succeeded":
-                progress(1.0, desc="Complete!")
-                # Extract video URL from the response structure we saw
-                video_url = get_result.content.video_url if hasattr(get_result, 'content') else None
-                print(f"Video URL: {video_url}")
-                yield "✅ Video generated successfully!", video_url
-                return
+            try:
+                get_result = client.content_generation.tasks.get(task_id=task_id)
+                status = get_result.status
                 
-            elif status == "failed":
-                error_msg = get_result.error if hasattr(get_result, 'error') else "Unknown error"
-                yield f"❌ Failed: {error_msg}", None
-                return
-            else:
-                progress(0.3 + (attempts/max_attempts)*0.7, desc=f"Status: {status}")
-                yield f"⏳ Status: {status}... (attempt {attempts + 1})", None
-                time.sleep(1)
+                if status == "succeeded":
+                    progress(1.0, desc="Complete!")
+                    video_url = get_result.content.video_url if hasattr(get_result, 'content') else None
+                    print(f"Video URL: {video_url}")
+                    yield "✅ Video generated successfully!", video_url
+                    return
+                    
+                elif status == "failed":
+                    error_msg = get_result.error if hasattr(get_result, 'error') else "Unknown error"
+                    yield f"❌ Failed: {error_msg}", None
+                    return
+                else:
+                    progress(0.3 + (attempts/max_attempts)*0.7, desc=f"Status: {status}")
+                    yield f"⏳ Status: {status}... (attempt {attempts + 1})", None
+                    time.sleep(1)
+                    attempts += 1
+                    
+            except Exception as e:
+                print(f"Polling error: {e}")
+                time.sleep(2)
                 attempts += 1
         
         yield "⏰ Timeout after 2 minutes", None
         
     except Exception as e:
         print(f"Error: {e}")
-        yield f"❌ Error: {str(e)}", None
+        if "Connection error" in str(e):
+            yield "❌ Connection Error: Cannot reach BytePlus API. This is likely a Hugging Face network restriction. Try running locally instead.", None
+        else:
+            yield f"❌ Error: {str(e)}", None
+
+def update_prompt(choice):
+    """Update prompt when shot type changes"""
+    return DEFAULT_PROMPTS.get(choice, "")
 
-# Simple, clean interface
+# Interface
 with gr.Blocks(title="BytePlus Video Generator", theme=gr.themes.Soft()) as demo:
     
     gr.Markdown("""
     # 🎥 BytePlus Video Generator
-    
-    Upload an image and describe the video you want to generate.
+    Upload an image and choose a shot type to generate a stunning video.
     """)
     
-    # Show API key status
-    if API_KEY:
-        gr.Markdown("✅ **API Key:** Configured")
-    else:
-        gr.Markdown("❌ **API Key:** Not configured - please add 'Key' secret")
+    # Status row
+    with gr.Row():
+        if API_KEY:
+            gr.Markdown("✅ **API Key:** Configured")
+        else:
+            gr.Markdown("❌ **API Key:** Not configured - please add 'Key' secret")
+    
+    # Connection test button
+    with gr.Row():
+        test_btn = gr.Button("🔌 Test Connection", size="sm")
+        test_result = gr.Textbox(label="Connection Test", lines=1)
+    
+    def run_connection_test():
+        ok, msg = test_connection()
+        return f"{'✅' if ok else '❌'} {msg}"
+    
+    test_btn.click(fn=run_connection_test, outputs=test_result)
+    
+    gr.Markdown("---")
     
     with gr.Row():
         with gr.Column():
-            # Image upload - type="filepath" returns a string path
+            # Image upload
             image_input = gr.Image(
                 label="Upload Starting Image",
-                type="filepath",  # This returns a string path
+                type="filepath",
                 height=300
             )
             
-            # Text prompt
+            # Shot type dropdown
+            shot_type = gr.Dropdown(
+                choices=list(DEFAULT_PROMPTS.keys()),
+                label="🎬 Shot Type",
+                value="Cinematic Nature"
+            )
+            
+            # Custom prompt
             prompt = gr.Textbox(
-                label="Video Description",
-                lines=4,
-                placeholder="Describe what you want to see...",
-                value="At breakneck speed, drones thread through intricate obstacles or stunning natural wonders, delivering an immersive, heart-pounding flying experience. --duration 5 --camerafixed false"
+                label="📝 Custom Prompt (override)",
+                lines=3,
+                placeholder="Or write your own prompt here...",
+                value=DEFAULT_PROMPTS["Cinematic Nature"]
+            )
+            
+            # Update prompt when shot type changes
+            shot_type.change(
+                fn=update_prompt,
+                inputs=shot_type,
+                outputs=prompt
             )
             
             # Generate button
@@ -130,7 +212,7 @@ with gr.Blocks(title="BytePlus Video Generator", theme=gr.themes.Soft()) as demo
             # Status
             status = gr.Textbox(
                 label="Status",
-                lines=4,
+                lines=5,
                 interactive=False
             )
             
@@ -140,24 +222,36 @@ with gr.Blocks(title="BytePlus Video Generator", theme=gr.themes.Soft()) as demo
                 interactive=False
             )
     
-    # Example prompts
+    # Example shots as quick buttons
     gr.Markdown("---")
-    gr.Markdown("### Example Prompts")
+    gr.Markdown("### Quick Shot Select")
     with gr.Row():
-        gr.Button("Nature").click(
-            fn=lambda: "Aerial drone shot over mountains at sunrise, cinematic --duration 5 --camerafixed false",
-            outputs=prompt
+        gr.Button("🏔️ Nature").click(
+            fn=lambda: "Cinematic Nature",
+            outputs=shot_type
+        )
+        gr.Button("🌃 City").click(
+            fn=lambda: "Urban Exploration",
+            outputs=shot_type
+        )
+        gr.Button("🏎️ Action").click(
+            fn=lambda: "Action Sequence",
+            outputs=shot_type
+        )
+        gr.Button("🎨 Abstract").click(
+            fn=lambda: "Abstract Art",
+            outputs=shot_type
         )
-        gr.Button("City").click(
-            fn=lambda: "Fast drone racing through futuristic city streets --duration 5 --camerafixed false",
-            outputs=prompt
+        gr.Button("🦁 Wildlife").click(
+            fn=lambda: "Wildlife Documentary",
+            outputs=shot_type
         )
-        gr.Button("Ocean").click(
-            fn=lambda: "Drone following waves at golden hour --duration 5 --camerafixed false",
-            outputs=prompt
+        gr.Button("⛷️ Sports").click(
+            fn=lambda: "Sports Highlight",
+            outputs=shot_type
         )
     
-    # Connect the generate button
+    # Connect generate button
     generate_event = generate_btn.click(
         fn=generate_video,
         inputs=[image_input, prompt],