Update app.py

app.py

@@ -27,7 +27,7 @@ def index():
         if not prompt:
             return render_template("index.html")
         
-        max_retries = 2
+        max_retries = 3
         attempt = 0
         last_error = None
         while attempt < max_retries:
@@ -42,6 +42,7 @@ 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.
+It is crucial that the script works correctly on the first try, so make sure to think about the layout and storyboard and stuff of the scene.
 You got this!! <3
 """
                 )
@@ -66,7 +67,7 @@ You got this!! <3
                 with open(code_filepath, "w") as f:
                     f.write(code)
                 
-                # Prepare the Manim command with --media_dir flag
+                # Prepare the Manim command with the --media_dir flag
                 cmd = [
                     "manim",
                     "-qm",
@@ -75,7 +76,12 @@ You got this!! <3
                     code_filepath,
                     scene_name
                 ]
-                subprocess.run(cmd, check=True, capture_output=True, text=True)
+                try:
+                    subprocess.run(cmd, check=True, capture_output=True, text=True)
+                except subprocess.CalledProcessError as cpe:
+                    # Log only if an error occurs from Manim
+                    app.logger.error("Manim error output: %s", cpe.stderr)
+                    raise Exception(f"Manim failed: {cpe.stderr}")
                 
                 # Construct the expected output path from Manim.
                 expected_dir = os.path.join(media_dir, "videos", code_filename.replace(".py", ""), "720p30")
@@ -86,7 +92,7 @@ You got this!! <3
                 # 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.
                 
                 # Schedule deletion of both files after 10 minutes (600 seconds)
                 def remove_files():