Update app.py

app.py

@@ -27,11 +27,12 @@ def index():
         if not prompt:
             return render_template("index.html")
         
-        max_retries = 3
+        max_retries = 2
         attempt = 0
         last_error = None
         while attempt < max_retries:
             try:
+                # Call the GenAI API to get the Manim code
                 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.
@@ -41,28 +42,31 @@ Plan out in chain of thought what you are going to do first, then give the final
 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.
-Make sure to keep things consistant with colors and also make sure to really plan out the scene so you get it right first try.
-It is crucial that the code works first try.
 You got this!! <3
 """
                 )
                 
+                # Extract the Python code block from the AI response
                 pattern = r"```python\s*(.*?)\s*```"
                 match = re.search(pattern, ai_response.text, re.DOTALL)
                 if not match:
                     raise Exception("No python code block found in the AI response.")
                 code = match.group(1)
                 
+                # Determine the scene class name from the generated code
                 scene_match = re.search(r"class\s+(\w+)\(.*Scene.*\):", code)
                 scene_name = scene_match.group(1) if scene_match else "MyScene"
                 
+                # Generate randomized filenames for the generated code and video
                 code_filename = f"generated_video_{uuid.uuid4().hex}.py"
                 video_filename = f"output_video_{uuid.uuid4().hex}.mp4"
                 
+                # 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
                 cmd = [
                     "manim",
                     "-qm",
@@ -73,23 +77,24 @@ You got this!! <3
                 ]
                 subprocess.run(cmd, check=True, capture_output=True, text=True)
                 
+                # Construct the expected output path from Manim.
                 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 the video file to /tmp (to serve it from there)
                 tmp_video_path = os.path.join("/tmp", video_filename)
                 shutil.move(video_path_in_media, tmp_video_path)
+                # The code file is already in /tmp (code_filepath)
                 
-                # The code file is already at code_filepath in /tmp.
-                tmp_code_path = 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)
+                        if os.path.exists(code_filepath):
+                            os.remove(code_filepath)
                     except Exception as e:
                         app.logger.error("Error removing files: %s", e)
                 Timer(600, remove_files).start()
@@ -98,6 +103,7 @@ You got this!! <3
                 return render_template("result.html", video_url=video_url)
             
             except Exception as e:
+                app.logger.error("Attempt %d failed: %s", attempt + 1, e)
                 last_error = e
                 attempt += 1
                 time.sleep(1)