Update app.py

app.py

@@ -12,7 +12,7 @@ try:
     GENAI_AVAILABLE = True
 except ImportError:
     GENAI_AVAILABLE = False
-    print("")
+    print("⚠️ google-generativeai not installed")
 
 # Get API key from environment
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
@@ -70,6 +70,26 @@ def fix_all_known_errors(code):
     # 8. Replace Tex with MathTex
     code = re.sub(r'\bTex\(', 'MathTex(', code)
     
+    # 9. CRITICAL FIX: Add color=BLACK to Text objects that don't have color specified
+    # This prevents invisible white text on white background
+    # Match Text(...) that doesn't have color= parameter
+    def add_black_color(match):
+        text_content = match.group(0)
+        # Check if color parameter already exists
+        if 'color=' not in text_content:
+            # Find the closing parenthesis of Text()
+            # Add color=BLACK before the last closing paren
+            if ', font_size=' in text_content:
+                # Insert after font_size
+                text_content = text_content.replace(')', ', color=BLACK)', 1)
+            else:
+                # Insert before closing paren
+                text_content = text_content[:-1] + ', color=BLACK)'
+        return text_content
+    
+    # Apply to all Text() objects
+    code = re.sub(r'Text\([^)]+\)', add_black_color, code)
+    
     return code
 
 def generate_code_with_gemini(prompt):
@@ -88,67 +108,75 @@ def generate_code_with_gemini(prompt):
         
         model = genai.GenerativeModel(model_name)
         
-        full_prompt = f"""Generate Manim code. CRITICAL RULES:
+        full_prompt = f"""You are a Manim code generator. Generate WORKING Manim animation code.
+
+USER REQUEST: {prompt}
+
+CRITICAL RULES YOU MUST FOLLOW:
 
-1. COLORS - ONLY these 10: BLUE, RED, GREEN, YELLOW, ORANGE, PURPLE, PINK, TEAL, WHITE, BLACK
-   NO variants (_A, _B, _C, _D, _E)
+1. ALWAYS start with: from manim import *
 
-2. TEXT OVERLAP PREVENTION (MOST IMPORTANT):
-   - ALWAYS FadeOut text before showing new text
-   - NEVER reuse the same position without FadeOut first
-   - Pattern for EVERY text:
+2. Class structure MUST be:
+```python
+from manim import *
+
+class MyScene(Scene):
+    def construct(self):
+        # Your animation code here
+        # MUST have at least one self.play() or self.add()
+```
+
+3. COLORS - ONLY use these 10: BLUE, RED, GREEN, YELLOW, ORANGE, PURPLE, PINK, TEAL, WHITE, BLACK
+   - NO color variants like BLUE_A, RED_B, etc.
+
+4. TEXT RULES:
+   - Use: Text("message", font_size=36, color=BLACK)
+   - ALWAYS FadeOut text before creating new text in same position
+   - Example:
    ```python
-   title = Text("Title", font_size=36).to_edge(UP, buff=1)
-   self.play(Write(title))
-   self.wait(1.5)
-   self.play(FadeOut(title))  # ← MANDATORY!
+   text1 = Text("First", font_size=36, color=BLACK)
+   self.play(Write(text1))
+   self.wait(1)
+   self.play(FadeOut(text1))  # Must FadeOut!
    
-   # Now safe to reuse UP position
-   subtitle = Text("Next", font_size=32).to_edge(UP, buff=1)
-   self.play(Write(subtitle))
-   self.wait(1.5)
-   self.play(FadeOut(subtitle))  # ← MANDATORY!
+   text2 = Text("Second", font_size=36, color=BLACK)
+   self.play(Write(text2))
+   self.wait(1)
+   self.play(FadeOut(text2))
    ```
 
-3. SAFE BOUNDARIES (CRITICAL):
-   - Font size: MAX 36 (never larger!)
-   - Shifts: UP*1.5, DOWN*1.5, LEFT*3, RIGHT*3 (MAX!)
-   - Always use buff=1 with to_edge()
-   - Safe zone: X[-4, 4], Y[-2, 2]
+5. SHAPES:
+   - Circle: Circle(radius=1, color=BLUE)
+   - Square: Square(side_length=2, color=RED)
+   - Rectangle: Rectangle(width=3, height=2, color=GREEN)
+   - Line: Line(start=LEFT, end=RIGHT, color=YELLOW)
+
+6. POSITIONING:
+   - Use .shift(UP), .shift(DOWN), .shift(LEFT), .shift(RIGHT)
+   - Use .to_edge(UP), .to_edge(DOWN), .to_edge(LEFT), .to_edge(RIGHT)
+   - Use .move_to(ORIGIN)
+
+7. ANIMATIONS:
+   - Show objects: Create(), Write(), FadeIn()
+   - Hide objects: FadeOut(), Uncreate()
+   - Transform: Transform(obj1, obj2)
+   - Move: obj.animate.shift(UP)
 
-4. NEVER use:
+8. NEVER USE:
    - get_x_axis_label(), get_y_axis_label()
-   - Checkmark, SVGMobject, ImageMobject
+   - SVGMobject, ImageMobject
+   - Checkmark, XMark
    - .add_prefix(), .add_suffix()
 
-5. For axis labels:
-   x_label = Text("Time", font_size=24).next_to(axes.x_axis, DOWN, buff=0.5)
-   y_label = Text("Value", font_size=24).next_to(axes.y_axis, LEFT, buff=0.5)
+9. SIMPLE AND WORKING:
+   - Keep it simple
+   - Test each animation
+   - Make sure there's visible content
+   - Animation should be 5-10 seconds long
 
-6. STRUCTURE - Follow this pattern:
-   ```python
-   from manim import *
-   
-   class MyScene(Scene):
-       def construct(self):
-           self.camera.background_color = WHITE
-           
-           # Section 1
-           text1 = Text("First", font_size=36).to_edge(UP, buff=1)
-           self.play(Write(text1))
-           self.wait(1.5)
-           self.play(FadeOut(text1))  # ← Clean up!
-           
-           # Section 2  
-           text2 = Text("Second", font_size=32).to_edge(UP, buff=1)
-           self.play(Write(text2))
-           self.wait(1.5)
-           self.play(FadeOut(text2))  # ← Clean up!
-   ```
+Generate COMPLETE, WORKING code that creates a visible animation for: {prompt}
 
-User wants: {prompt}
-
-Generate complete code following ALL rules above. EVERY text MUST have FadeOut before next text!"""
+The code must actually show something on screen!"""
         
         response = model.generate_content(full_prompt)
         code = response.text
@@ -223,11 +251,18 @@ def render_video(prompt, quality="low"):
         )
         
         output_lines = []
+        has_error = False
         for line in process.stdout:
             output_lines.append(line)
+            # Check for common errors in output
+            if any(err in line.lower() for err in ['error', 'exception', 'traceback', 'failed']):
+                has_error = True
         
         process.wait()
         
+        # Get full output for debugging
+        full_output = ''.join(output_lines)
+        
         if process.returncode == 0:
             # Find the generated video
             media_path = temp_dir / "media"
@@ -235,11 +270,19 @@ def render_video(prompt, quality="low"):
             
             if video_files:
                 video_path = str(video_files[0])
-                yield f"✅ Video rendered successfully! 🎉", code, video_path
+                # Check if video file is too small (blank video indicator)
+                video_size = os.path.getsize(video_path)
+                
+                if video_size < 5000:  # Less than 5KB means blank/error
+                    yield f"⚠️ Video rendered but seems blank (only {video_size} bytes)!\n\nMaybe there's an error in the code. Check the generated code below.\n\nRender output (last 40 lines):\n{chr(10).join(output_lines[-40:])}", code, None
+                elif has_error:
+                    yield f"⚠️ Video rendered but errors detected!\n\nRender output (last 40 lines):\n{chr(10).join(output_lines[-40:])}", code, video_path
+                else:
+                    yield f"✅ Video rendered successfully! 🎉", code, video_path
             else:
                 yield "❌ Video file not found after rendering", code, None
         else:
-            error_msg = ''.join(output_lines[-30:])  # Last 30 lines
+            error_msg = ''.join(output_lines[-40:])  # Last 40 lines
             yield f"❌ Rendering failed:\n\n{error_msg}", code, None
         
         # Cleanup
@@ -259,7 +302,11 @@ def create_interface():
         gr.Markdown("""
         # 🎬 Manim Video Generator
         Create mathematical animations using AI! Describe what you want to see animated.
-                
+        
+        **⚠️ IMPORTANT: You need to set GEMINI_API_KEY in Space Settings → Variables and secrets**
+        
+        Get your free Gemini API key from: https://aistudio.google.com/app/apikey
+        
         **Examples:**
         - "Explain Pythagorean theorem with animation"
         - "Show a sine wave transforming into a cosine wave"
@@ -295,6 +342,11 @@ def create_interface():
         - Medium/High quality may take 2-3 minutes
         - Complex animations may timeout on free tier
         
+        ### 🔧 Setup Required:
+        1. Get a free Gemini API key from: https://aistudio.google.com/app/apikey
+        2. In your Space, go to Settings → Variables and secrets
+        3. Add: Name: `GEMINI_API_KEY`, Value: `your-api-key-here`
+        4. Restart the Space
         """)
         
         generate_btn.click(
@@ -319,4 +371,4 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch() 
+    demo.launch()