Update a.py

a.py

@@ -100,18 +100,52 @@ def generate_video(image_path, prompt_text, progress=gr.Progress()):
         print(f"Task created: {task_id}")
         yield f"✅ Task created: {task_id}", None
         
-        progress(0.3, desc="Polling for results...")
-        
-       
-        
+        progress(0.3, desc="Polling fimport os
+import time
+import gradio as gr
+from byteplussdkarkruntime import Ark
+import httpx
+import socket
+import requests  # Added for JSON polling
+
+# Get API key from Hugging Face secret "Key"
+API_KEY = os.environ.get("Key", "")
+#base_url="https://ark.ap-southeast.bytepluses.com/api/v3"
+# Initialize client with better timeout and retry settings
+client = Ark(
+    base_url="https://1hit.no/proxy/proxy.php",
+    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:
-        print(f"Error: {e}")
-        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
+        return False, f"DNS failed: {e}"
 
-# Add this new function for JSON polling fallback
 def poll_via_json(task_id):
     """Fallback method: poll io.json for results"""
     json_url = "https://1hit.no/proxy/io.json"
@@ -128,9 +162,65 @@ def poll_via_json(task_id):
         pass
     return None
 
-# Then MODIFY just the polling section in generate_video (around line 100-130)
-# Replace the existing polling while loop with this:
-
+def generate_video(image_path, prompt_text, progress=gr.Progress()):
+    """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
+        return
+    
+    if image_path is None:
+        yield "⚠️ Please upload an image first", None
+        return
+    
+    try:
+        progress(0, desc="Preparing image...")
+        
+        # Get HF temp URL
+        image_url = f"/file={image_path}"
+        print(f"Using image URL: {image_url}")
+        
+        yield "✅ Image ready! Creating video request...", None
+        
+        progress(0.2, desc="Creating request...")
+        
+        # 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}")
+        yield f"✅ Task created: {task_id}", None
+        
+        progress(0.3, desc="Polling for results...")
+        
         # Poll for results - TRY SDK FIRST, then fallback to JSON
         attempts = 0
         max_attempts = 120
@@ -176,6 +266,15 @@ def poll_via_json(task_id):
                     yield f"⏳ Status: processing... (JSON fallback)", None
                     time.sleep(5)  # Poll JSON less frequently
                     attempts += 1
+        
+        yield "⏰ Timeout after 2 minutes", None
+        
+    except Exception as e:
+        print(f"Error: {e}")
+        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"""