Update app.py

app.py

@@ -7,7 +7,6 @@ import shutil
 import time
 from threading import Timer
 import uuid
-import sys
 
 app = Flask(__name__)
 
@@ -28,19 +27,18 @@ def index():
         attempt = 0
         while attempt < max_retries:
             try:
-                print("Calling GenAI API...")
-                sys.stdout.flush()
+                # 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.
-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
@@ -49,8 +47,6 @@ You got this!! <3
                 if not match:
                     raise Exception("No python code block found in the AI response.")
                 code = match.group(1)
-                print("Extracted code (first 200 chars):", code[:200])
-                sys.stdout.flush()
                 
                 # Determine the scene class name from the generated code
                 scene_match = re.search(r"class\s+(\w+)\(.*Scene.*\):", code)
@@ -63,7 +59,7 @@ 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
+                # Save the generated code to a file (in the current working directory)
                 with open(code_filename, "w") as f:
                     f.write(code)
                 
@@ -75,13 +71,10 @@ You got this!! <3
                     code_filename,
                     scene_name
                 ]
-                print("Running Manim command:", " ".join(cmd))
-                sys.stdout.flush()
                 # Run Manim to generate the video, capturing its output
                 result = subprocess.run(cmd, check=True, capture_output=True, text=True)
                 print("Manim stdout:", result.stdout)
                 print("Manim stderr:", result.stderr)
-                sys.stdout.flush()
                 
                 # Construct the expected output path from Manim
                 video_path_in_media = os.path.join("media", "videos", code_filename.replace(".py", ""), "720p30", video_filename)
@@ -89,7 +82,7 @@ You got this!! <3
                 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
+                # Move both the video and the generated code file to /tmp (which is writable)
                 tmp_video_path = os.path.join("/tmp", video_filename)
                 shutil.move(video_path_in_media, tmp_video_path)
                 
@@ -104,29 +97,28 @@ You got this!! <3
                         if os.path.exists(tmp_code_path):
                             os.remove(tmp_code_path)
                         print("Removed files:", tmp_video_path, tmp_code_path)
-                        sys.stdout.flush()
                     except Exception as e:
                         app.logger.error("Error removing files: %s", e)
                 Timer(600, remove_files).start()
                 
-                # Generate a URL that points to our video-serving route
+                # Generate a URL that points to our video-serving route (the video file is in /tmp)
                 video_url = url_for('get_video', filename=video_filename)
                 return render_template("result.html", video_url=video_url)
             
             except Exception as e:
-                print(f"Attempt {attempt + 1} failed: {type(e).__name__}: {e}")
-                sys.stdout.flush()
+                print(f"Attempt {attempt + 1} failed: {e}")
                 attempt += 1
-                time.sleep(1)
+                time.sleep(1)  # Wait a bit before retrying
         
         # If we reach here, we've exceeded the maximum number of retries.
-        return render_template("result.html", error=f"Error: {e}")
+        return render_template("result.html", error="An error occurred. Please try again later.")
             
     return render_template("index.html")
 
 @app.route("/video/<filename>")
 def get_video(filename):
+    # Serve the video file from /tmp
     return send_from_directory("/tmp", filename)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", port=7860, debug=False)