Update a.py

a.py

@@ -142,6 +142,72 @@ def generate_video(image_path, prompt_text, progress=gr.Progress()):
         else:
             yield f"❌ Error: {str(e)}", None
 
+# 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"
+    try:
+        response = requests.get(json_url)
+        data = response.json()
+        if task_id in data:
+            task_data = data[task_id]
+            if task_data.get('status') == 'succeeded':
+                return task_data.get('video_url')
+            elif task_data.get('status') == 'failed':
+                return None
+    except:
+        pass
+    return None
+
+# Then MODIFY just the polling section in generate_video (around line 100-130)
+# Replace the existing polling while loop with this:
+
+        # Poll for results - TRY SDK FIRST, then fallback to JSON
+        attempts = 0
+        max_attempts = 120
+        used_fallback = False
+        
+        while attempts < max_attempts:
+            try:
+                # Try SDK method first
+                get_result = client.content_generation.tasks.get(task_id=task_id)
+                status = get_result.status
+                
+                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:
+                # If SDK fails, try JSON fallback
+                if not used_fallback:
+                    print("SDK polling failed, trying JSON fallback...")
+                    used_fallback = True
+                
+                video_url = poll_via_json(task_id)
+                if video_url:
+                    yield "✅ Video generated successfully! (via JSON)", video_url
+                    return
+                elif video_url is None:  # Failed
+                    yield "❌ Task failed (via JSON)", None
+                    return
+                else:  # Still processing
+                    yield f"⏳ Status: processing... (JSON fallback)", None
+                    time.sleep(5)  # Poll JSON less frequently
+                    attempts += 1
+
 def update_prompt(choice):
     """Update prompt when shot type changes"""
     return DEFAULT_PROMPTS.get(choice, "")