Update app.py

app.py

@@ -5,22 +5,47 @@ import shutil
 import tempfile
 import re
 import ast
+import sys
+import time
+from pathlib import Path
 
 # --- Configuration ---
 BASE_DIR = "./media"  # Use relative path in user home directory
 
 def ensure_permissions():
     """Ensure the base directory exists and is writable."""
-    os.makedirs(BASE_DIR, exist_ok=True)
-    # Try to create a test file to check permissions
-    test_file = os.path.join(BASE_DIR, "test_write.txt")
     try:
+        # Create directories with proper permissions
+        os.makedirs(BASE_DIR, exist_ok=True, mode=0o755)
+        
+        # Try to create a test file to check permissions
+        test_file = os.path.join(BASE_DIR, "test_write.txt")
         with open(test_file, 'w') as f:
             f.write("test")
         os.remove(test_file)
+        print(f"✅ Directory {BASE_DIR} is writable")
+        
+    except PermissionError as e:
+        print(f"❌ Permission denied for {BASE_DIR}: {e}")
+        # Try alternative directories
+        for alt_dir in ["./tmp_media", "/tmp/manim_render", "./output"]:
+            try:
+                global BASE_DIR
+                BASE_DIR = alt_dir
+                os.makedirs(BASE_DIR, exist_ok=True, mode=0o755)
+                test_file = os.path.join(BASE_DIR, "test_write.txt")
+                with open(test_file, 'w') as f:
+                    f.write("test")
+                os.remove(test_file)
+                print(f"✅ Using alternative directory: {BASE_DIR}")
+                break
+            except:
+                continue
+        else:
+            raise gr.Error("Cannot create writable directory for video output")
     except Exception as e:
-        print(f"Warning: Permission issue with {BASE_DIR}: {e}")
-        # Fallback to current directory
+        print(f"⚠️ Warning: Issue with directory setup: {e}")
+        # Last resort: current directory
         global BASE_DIR
         BASE_DIR = "./media"
         os.makedirs(BASE_DIR, exist_ok=True)
@@ -43,6 +68,61 @@ def get_scene_class_name(code):
         match = re.search(pattern, code)
         return match.group(1) if match else None
 
+def validate_manim_code(code):
+    """Validate Manim code for common issues."""
+    if not code.strip():
+        return False, "Code is empty."
+    
+    # Check for Scene class
+    if not get_scene_class_name(code):
+        return False, "No Scene class found. Make sure your class inherits from Scene."
+    
+    # Check for basic syntax
+    try:
+        ast.parse(code)
+    except SyntaxError as e:
+        return False, f"Syntax error: {e}"
+    
+    # Check for required construct method
+    if 'def construct(' not in code:
+        return False, "Scene class must have a construct() method."
+    
+    return True, "Code looks good!"
+
+def check_manim_installation():
+    """Check if Manim is properly installed."""
+    try:
+        result = subprocess.run(
+            ["manim", "--version"],
+            capture_output=True,
+            text=True,
+            timeout=10
+        )
+        if result.returncode == 0:
+            return True, result.stdout.strip()
+        else:
+            return False, "Manim command failed"
+    except Exception as e:
+        return False, f"Manim not found: {e}"
+
+def setup_environment():
+    """Set up the environment for Manim rendering."""
+    # Ensure permissions
+    ensure_permissions()
+    
+    # Check Manim installation
+    manim_ok, manim_info = check_manim_installation()
+    if not manim_ok:
+        print(f"⚠️ Manim issue: {manim_info}")
+    else:
+        print(f"✅ {manim_info}")
+    
+    # Set environment variables for better rendering
+    os.environ['MANIM_CONFIG_DIR'] = BASE_DIR
+    os.environ['MANIM_PLUGINS_DIR'] = os.path.join(BASE_DIR, 'plugins')
+    
+    return manim_ok
+
 def render_manim_video(manim_code):
     """
     Runs the Manim command with dynamic code and returns the path to the rendered video.
@@ -266,4 +346,6 @@ with gr.Blocks(title="Manim Video Renderer", theme=gr.themes.Soft()) as demo:
 
 # --- Launch ---
 if __name__ == "__main__":
+    # Setup environment before launching
+    setup_environment()
     demo.launch(server_name="0.0.0.0", server_port=7860)