Update a.py

a.py

@@ -102,38 +102,7 @@ def generate_video(image_path, prompt_text, progress=gr.Progress()):
         
         progress(0.3, desc="Polling for results...")
         
-        # Poll for results
-        attempts = 0
-        max_attempts = 120
-        
-        while attempts < max_attempts:
-            try:
-                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:
-                print(f"Polling error: {e}")
-                time.sleep(2)
-                attempts += 1
-        
-        yield "⏰ Timeout after 2 minutes", None
+       
         
     except Exception as e:
         print(f"Error: {e}")