Update app.py

app.py

@@ -7,6 +7,7 @@ import shutil
 import time
 from threading import Timer
 import uuid
+import sys
 
 app = Flask(__name__)
 
@@ -27,18 +28,19 @@ def index():
         attempt = 0
         while attempt < max_retries:
             try:
-                # Call the GenAI API to get the Manim code
+                print("Calling GenAI API...")
+                sys.stdout.flush()
                 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
@@ -47,6 +49,8 @@ def index():
                 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)
@@ -59,7 +63,7 @@ def index():
                 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)
+                # Save the generated code to a file
                 with open(code_filename, "w") as f:
                     f.write(code)
                 
@@ -71,10 +75,13 @@ def index():
                     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)
@@ -82,7 +89,7 @@ def index():
                 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
                 tmp_video_path = os.path.join("/tmp", video_filename)
                 shutil.move(video_path_in_media, tmp_video_path)
                 
@@ -97,28 +104,29 @@ def index():
                         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 (the video file is in /tmp)
+                # Generate a URL that points to our video-serving route
                 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: {e}")
+                print(f"Attempt {attempt + 1} failed: {type(e).__name__}: {e}")
+                sys.stdout.flush()
                 attempt += 1
-                time.sleep(1)  # Wait a bit before retrying
+                time.sleep(1)
         
         # If we reach here, we've exceeded the maximum number of retries.
-        return render_template("result.html", error="An error occurred. Please try again later.")
+        return render_template("result.html", error=f"Error: {e}")
             
     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=False)
+    app.run(host="0.0.0.0", port=7860, debug=True)