Update app.py

app.py

@@ -256,58 +256,99 @@ def render_video_from_code(code, quality):
         return f"❌ {e}", None
 
 def render_video(prompt, template, duration, quality):
-    """Generate and render video"""
+    """Generate and render video with auto-retry"""
     if not prompt:
         yield "❌ Enter a prompt!", None, None
         return
     
-    try:
-        temp_dir = Path(tempfile.mkdtemp(prefix="manim_"))
-        
-        yield f"🤖 Generating {template} video ({duration})...", None, None
-        
-        code, error = generate_code_with_gemini(prompt, template, duration, quality)
-        
-        if error:
-            yield error, None, None
-            return
-        
-        if "```python" in code:
-            code = code.split("```python")[1].split("```")[0]
-        elif "```" in code:
-            code = code.split("```")[1].split("```")[0]
-        code = code.strip()
-        
-        code = fix_all_known_errors(code)
-        
-        match = re.search(r'class\s+(\w+)\s*\(Scene\)', code)
-        if not match:
-            yield "❌ No Scene class!", code, None
-            return
-        
-        class_name = match.group(1)
-        code_file = temp_dir / "animation.py"
-        code_file.write_text(code)
-        
-        yield f"🎬 Rendering {class_name}...", code, None
-        
-        quality_map = {'Low (fast)': '-ql', 'Medium': '-qm', 'High (1080p)': '-qh'}
-        quality_flag = quality_map.get(quality, '-qm')
-        
-        cmd = f"manim {code_file} {class_name} {quality_flag} --disable_caching --media_dir {temp_dir}/media"
-        result = subprocess.run(cmd, shell=True, capture_output=True, text=True, timeout=300)
-        
-        if result.returncode == 0:
-            videos = list((temp_dir / "media").rglob("*.mp4"))
-            if videos:
-                yield "✅ YouTube-ready video rendered! 🎉", code, str(videos[0])
+    MAX_ATTEMPTS = 3
+    
+    for attempt in range(1, MAX_ATTEMPTS + 1):
+        try:
+            temp_dir = Path(tempfile.mkdtemp(prefix="manim_"))
+            
+            if attempt == 1:
+                yield f"🤖 Generating {template} video ({duration})...", None, None
             else:
-                yield "❌ Video not found", code, None
-        else:
-            yield f"❌ Failed:\n{result.stderr[-500:]}", code, None
+                yield f"🔄 Attempt {attempt}/{MAX_ATTEMPTS} - Simplifying code...", None, None
             
-    except Exception as e:
-        yield f"❌ {e}", None, None
+            # Adjust prompt complexity based on attempt
+            if attempt == 1:
+                current_prompt = prompt
+            elif attempt == 2:
+                current_prompt = f"KEEP IT VERY SIMPLE. {prompt}. Use only basic shapes and text, no complex arrangements."
+            else:
+                current_prompt = f"EXTREMELY SIMPLE VERSION: {prompt}. Only 2-3 simple elements, no arrays, no loops, no complex positioning."
+            
+            code, error = generate_code_with_gemini(current_prompt, template, duration, quality)
+            
+            if error:
+                if attempt < MAX_ATTEMPTS:
+                    continue
+                yield error, None, None
+                return
+            
+            if "```python" in code:
+                code = code.split("```python")[1].split("```")[0]
+            elif "```" in code:
+                code = code.split("```")[1].split("```")[0]
+            code = code.strip()
+            
+            code = fix_all_known_errors(code)
+            
+            # Extra safety fixes for common index errors
+            # Remove array/list indexing that might fail
+            code = re.sub(r'\[(\d+)\]\.', '[0].', code)  # Replace [N] with [0]
+            code = re.sub(r'for\s+\w+\s+in\s+range\([^)]+\):', '# Loop removed', code)  # Comment out loops
+            
+            match = re.search(r'class\s+(\w+)\s*\(Scene\)', code)
+            if not match:
+                if attempt < MAX_ATTEMPTS:
+                    continue
+                yield "❌ No Scene class!", code, None
+                return
+            
+            class_name = match.group(1)
+            code_file = temp_dir / "animation.py"
+            code_file.write_text(code)
+            
+            yield f"🎬 Rendering (attempt {attempt})...", code, None
+            
+            quality_map = {'Low (fast)': '-ql', 'Medium': '-qm', 'High (1080p)': '-qh'}
+            quality_flag = quality_map.get(quality, '-qm')
+            
+            cmd = f"manim {code_file} {class_name} {quality_flag} --disable_caching --media_dir {temp_dir}/media"
+            result = subprocess.run(cmd, shell=True, capture_output=True, text=True, timeout=300)
+            
+            if result.returncode == 0:
+                videos = list((temp_dir / "media").rglob("*.mp4"))
+                if videos:
+                    if attempt > 1:
+                        yield f"✅ Success on attempt {attempt}! (Simplified version) 🎉", code, str(videos[0])
+                    else:
+                        yield "✅ YouTube-ready video rendered! 🎉", code, str(videos[0])
+                    return
+                else:
+                    if attempt < MAX_ATTEMPTS:
+                        continue
+                    yield "❌ Video not found", code, None
+                    return
+            else:
+                # Rendering failed, try again with simpler code
+                if attempt < MAX_ATTEMPTS:
+                    error_snippet = result.stderr[-200:] if result.stderr else "Unknown error"
+                    print(f"Attempt {attempt} failed: {error_snippet}")
+                    continue
+                else:
+                    yield f"❌ All {MAX_ATTEMPTS} attempts failed. Last error:\n{result.stderr[-500:]}", code, None
+                    return
+                
+        except Exception as e:
+            if attempt < MAX_ATTEMPTS:
+                print(f"Attempt {attempt} crashed: {e}")
+                continue
+            yield f"❌ Error after {MAX_ATTEMPTS} attempts: {e}", None, None
+            return
 
 # Gradio Interface
 with gr.Blocks(title="YouTube Math Explainer") as demo: