Update app.py

app.py

@@ -17,7 +17,7 @@ if not API_KEY:
     raise ValueError("Missing GOOGLE_API_KEY environment variable.")
 client = genai.Client(api_key=API_KEY)
 
-# Define a dedicated media directory in /tmp
+# Define a dedicated media directory in /tmp for Manim output
 media_dir = os.path.join("/tmp", "manim_media")
 os.makedirs(media_dir, exist_ok=True)
 
@@ -68,8 +68,9 @@ You got this!! <3
                 code_filename = f"generated_video_{uuid.uuid4().hex}.py"
                 video_filename = f"output_video_{uuid.uuid4().hex}.mp4"
                 
-                # Save the generated code to a file (in the current working directory)
-                with open(code_filename, "w") as f:
+                # Write the generated code file directly to /tmp
+                code_filepath = os.path.join("/tmp", code_filename)
+                with open(code_filepath, "w") as f:
                     f.write(code)
                 
                 # Prepare the Manim command with --media_dir flag
@@ -78,7 +79,7 @@ You got this!! <3
                     "-qm",
                     "--media_dir", media_dir,
                     "-o", video_filename,
-                    code_filename,
+                    code_filepath,
                     scene_name
                 ]
                 print("Running Manim command:", " ".join(cmd))
@@ -88,31 +89,32 @@ You got this!! <3
                 print("Manim stderr:", result.stderr)
                 sys.stdout.flush()
                 
-                # Debug: list the media directory structure
+                # Debug: List the media directory structure
                 for root, dirs, files in os.walk(media_dir):
                     print(root, dirs, files)
                 sys.stdout.flush()
                 
                 # Construct the expected output path from Manim.
+                # With --media_dir set, output should be in:
+                # {media_dir}/videos/<code_filename_without_.py>/720p30/<video_filename>
                 expected_dir = os.path.join(media_dir, "videos", code_filename.replace(".py", ""), "720p30")
                 video_path_in_media = os.path.join(expected_dir, video_filename)
                 if not os.path.exists(video_path_in_media):
                     raise Exception(f"Manim did not produce the expected output file at {video_path_in_media}")
                 
-                # Move both the video and the generated code file to /tmp
+                # Move the video file to /tmp (it will be served from here)
                 tmp_video_path = os.path.join("/tmp", video_filename)
                 shutil.move(video_path_in_media, tmp_video_path)
-                tmp_code_path = os.path.join("/tmp", code_filename)
-                shutil.move(code_filename, tmp_code_path)
+                # The code file is already in /tmp (at code_filepath)
                 
                 # Schedule deletion of both files after 10 minutes (600 seconds)
                 def remove_files():
                     try:
                         if os.path.exists(tmp_video_path):
                             os.remove(tmp_video_path)
-                        if os.path.exists(tmp_code_path):
-                            os.remove(tmp_code_path)
-                        print("Removed files:", tmp_video_path, tmp_code_path)
+                        if os.path.exists(code_filepath):
+                            os.remove(code_filepath)
+                        print("Removed files:", tmp_video_path, code_filepath)
                         sys.stdout.flush()
                     except Exception as e:
                         app.logger.error("Error removing files: %s", e)
@@ -134,6 +136,7 @@ You got this!! <3
 
 @app.route("/video/<filename>")
 def get_video(filename):
+    # Serve the video file from /tmp
     return send_from_directory("/tmp", filename)
 
 if __name__ == "__main__":