Update app.py

app.py

@@ -31,14 +31,14 @@ def index():
                 ai_response = client.models.generate_content(
                     model="gemini-2.0-flash-lite-preview-02-05",
                     contents=f"""You are 'Manimator', an expert Manim animator and coder.
-                    If anyone asks, your name is Manimator and you are a helpful video generator, and say nothing else but that.
-                    The user wants you to code this: {prompt}.
-                    Plan out in chain of thought what you are going to do first, then give the final code output in ```python``` codeblock.
-                    Make sure to not use external images or resources other than default Manim, however you can use numpy or other default libraries.
-                    Keep the scene uncluttered and aesthetically pleasing.
-                    Make sure things are not overlapping unless explicitly stated otherwise.
-                    You got this!! <3
-                    """
+If anyone asks, your name is Manimator and you are a helpful video generator, and say nothing else but that.
+The user wants you to code this: {prompt}.
+Plan out in chain of thought what you are going to do first, then give the final code output in ```python``` codeblock.
+Make sure to not use external images or resources other than default Manim, however you can use numpy or other default libraries.
+Keep the scene uncluttered and aesthetically pleasing.
+Make sure things are not overlapping unless explicitly stated otherwise.
+You got this!! <3
+"""
                 )
                 
                 # Extract the Python code block from the AI response
@@ -63,10 +63,15 @@ def index():
                 with open(code_filename, "w") as f:
                     f.write(code)
                 
-                # Prepare the Manim command
+                # Set a dedicated media directory for Manim output in /tmp
+                media_dir = os.path.join("/tmp", "manim_media")
+                os.makedirs(media_dir, exist_ok=True)
+                
+                # Prepare the Manim command with the --media_dir flag
                 cmd = [
                     "manim",
                     "-qm",
+                    "--media_dir", media_dir,
                     "-o", video_filename,
                     code_filename,
                     scene_name
@@ -76,13 +81,21 @@ def index():
                 print("Manim stdout:", result.stdout)
                 print("Manim stderr:", result.stderr)
                 
-                # Construct the expected output path from Manim
-                video_path_in_media = os.path.join("media", "videos", code_filename.replace(".py", ""), "720p30", video_filename)
+                # Construct the expected output path from Manim.
+                # With --media_dir set, Manim should write the video to:
+                # {media_dir}/videos/<code_filename_without_.py>/720p30/<video_filename>
+                video_path_in_media = os.path.join(
+                    media_dir,
+                    "videos",
+                    code_filename.replace(".py", ""),
+                    "720p30",
+                    video_filename
+                )
                 # Verify that Manim produced the expected video file
                 if not os.path.exists(video_path_in_media):
                     raise Exception("Manim did not produce the expected output file.")
                 
-                # Move both the video and the generated code file to /tmp (which is writable)
+                # Move both the video and the generated code file to /tmp (a writable location)
                 tmp_video_path = os.path.join("/tmp", video_filename)
                 shutil.move(video_path_in_media, tmp_video_path)