From b4b6ed6ba2967583415ba2d83384c5a70ad81bae Mon Sep 17 00:00:00 2001
From: "codegen-sh[bot]" <131295404+codegen-sh[bot]@users.noreply.github.com>
Date: Mon, 14 Apr 2025 07:46:52 +0000
Subject: [PATCH] Fix syntax errors in main.py

---
 main.py | 34 ++++++++++++++++------------------
 1 file changed, 16 insertions(+), 18 deletions(-)

diff --git a/main.py b/main.py
index 525078a..f088c26 100644
--- a/main.py
+++ b/main.py
@@ -35,9 +35,9 @@ def generate_manim_code(refined_prompt):
         "Based on the "
         "The script should define a Manim class called 'AutoScene'. "
         "Ensure that text elements are well-aligned and old text disappears before new text appears. "
-        "Ensure proper alignment, remove previous text before adding new text, and include smooth transitions."#
-        "Take note of frame space and don't overflow out of frame"#
-        "try to utilize all the frame area withouth overlapping"#
+        "Ensure proper alignment, remove previous text before adding new text, and include smooth transitions. "
+        "Take note of frame space and don't overflow out of frame. "
+        "Try to utilize all the frame area without overlapping."
     )
 
     response = model.generate_content(final_prompt)
@@ -48,6 +48,17 @@ def generate_manim_code(refined_prompt):
         print("Error: Failed to generate Manim code.")
         return None
 
+def clean_text(text: str) -> str:
+    prefix = "```python"
+    suffix = "```"
+    
+    if text.startswith(prefix):
+        text = text[len(prefix):]
+    if text.endswith(suffix):
+        text = text[:-len(suffix)]
+    
+    return text.strip()
+
 def save_manim_script(code, filename="generated_manim.py"):
     """Save the AI-generated Manim script to a file."""
     with open(filename, "w", encoding="utf-8") as file:
@@ -66,36 +77,23 @@ def main():
 
     if refined_prompt:
         print("\nUsing refined prompt to generate Manim code...\n", refined_prompt)
-        manim_code =generate_manim_code(refined_prompt)
+        manim_code = generate_manim_code(refined_prompt)
 
         if manim_code:
             print("\nGenerated Manim Code:\n")
-            clean_manim_code=clean_text(manim_code)
+            clean_manim_code = clean_text(manim_code)
             print(clean_manim_code)  # Print the generated Manim script
 
             print("\nSaving Manim script...")
             
             save_manim_script(clean_manim_code)
             
-            
-
             print("\nRunning Manim script...")
             run_manim_script()
         else:
             print("Failed to generate Manim code.")
     else:
         print("Failed to refine prompt.")
-def clean_text(text: str) -> str:
-    prefix = "```python"
-    suffix = "```"
-    
-    if text.startswith(prefix):
-        text = text[len(prefix):]
-    if text.endswith(suffix):
-        text = text[:-len(suffix)]
-    
-    return text.strip()
-
 
 if __name__ == "__main__":
     main()