UP

utils.py

@@ -33,25 +33,35 @@ def extract_pdf_text(pdf_path: str) -> str:
 
 # ── Gemini LLM ────────────────────────────────────────────────────────────────
 
-def generate_manim_code(prompt_text: str, api_key: str) -> str:
-    """Stream Manim code from Gemini."""
+def generate_manim_code(prompt_text: str, api_key: str):
+    """Stream Manim code from Gemini 3 Flash Preview and return it as a string."""
     client = genai.Client(api_key=api_key)
-    model = "gemini-2.5-flash-preview-05-20"  # latest stable flash model
+    model = "gemini-3-flash-preview"
+
     contents = [
-        types.Content(role="user", parts=[types.Part.from_text(prompt_text)])
+        types.Content(
+            role="user",
+            parts=[types.Part.from_text(text=prompt_text)]
+        )
     ]
+
     config = types.GenerateContentConfig(
-        thinking_config=types.ThinkingConfig(thinking_budget=8192)
+        thinking_config=types.ThinkingConfig(
+            thinking_level="HIGH"
+        )
     )
 
     code = ""
     for chunk in client.models.generate_content_stream(
-        model=model, contents=contents, config=config
+        model=model,
+        contents=contents,
+        config=config
     ):
         if chunk.text:
             code += chunk.text
-    return code
+            print(chunk.text, end="")  # optional live streaming output
 
+    return code
 
 def sanitize_manim_code(raw: str, job_id: str) -> str:
     """