switch to claude instead of gemini for better outputs

app.py

@@ -24,17 +24,16 @@ from threading import Timer
 
 
 from flask import Flask, render_template, request, url_for, send_from_directory
-from google import genai
 
 app = Flask(__name__)
 
-# Load API key from environment variable
-API_KEY = os.environ.get("GOOGLE_API_KEY")
+API_KEY = os.environ.get("OPENROUTER")
 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 for Manim output
+    raise ValueError("Missing OPENROUTER environment variable.")
+client = OpenAI(
+  base_url="https://openrouter.ai/api/v1",
+  api_key=API_KEY,
+)
 media_dir = os.path.join("/tmp", "manim_media")
 os.makedirs(media_dir, exist_ok=True)
 
@@ -50,24 +49,33 @@ def index():
         last_error = None
         while attempt < max_retries:
             try:
-                # Call the GenAI API to get the Manim code and commentary script
-                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.
-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.
-Make sure to think through what you are going to do and think about the topic before you write the code.
-Use self.wait at the end to give the user enough time to process the frame.
-You got this!! <3
-"""
-                )
+                # Call the GenAI API to get the Manim code
+                completion = client.chat.completions.create(
+                    extra_body={},
+                    model="anthropic/claude-3.5-sonnet",
+                    messages=[
+                        {
+                        "role": "user",
+                        "content": [
+                            {
+                            "type": "text",
+                            "text": 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.
+                    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.
+                    Make sure to think through what you are going to do and think about the topic before you write the code.
+                    You got this!! <3
+                    """
+                    },
+                ]
+                }
+            ]
+            )
                 
-                # Extract the Python code block from the AI response
                 code_pattern = r"```python\s*(.*?)\s*```"
                 code_match = re.search(code_pattern, ai_response.text, re.DOTALL)
                 if not code_match:
@@ -75,22 +83,17 @@ You got this!! <3
                 code = 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)
                 
                 
-                # === Run Manim to generate the silent video ===
-                # Prepare the Manim command with the --media_dir flag
                 cmd = [
                     "manim",
                     "-qm",
@@ -105,17 +108,14 @@ You got this!! <3
                     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")
                 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)
 
-                # Schedule deletion of all temporary files after 10 minutes (600 seconds)
                 def remove_files():
                     for fpath in [tmp_video_path, code_filepath, commentary_audio_path, final_video_path]:
                         try:
@@ -126,7 +126,6 @@ You got this!! <3
                 
                 Timer(600, remove_files).start()
                 
-                # Use the final combined video for display
                 video_url = url_for('get_video', filename=video_filename)
                 return render_template("result.html", video_url=video_url)