Create different animations based on input parameters

app.py

@@ -18,9 +18,12 @@ def create_motion_animation(text_prompt, motion_length=3.0, seed=0):
     # Create a unique filename based on parameters
     output_filename = f"output_{abs(hash(text_prompt) % 10000)}_{int(motion_length)}_{seed}.mp4"
     
+    # Sanitize the text prompt for Python string
+    safe_prompt = text_prompt.replace('"', '')
+    
     # Create a simple visualization script
     with open("motion_animation.py", "w") as f:
-        f.write(f"""
+        f.write("""
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.animation import FuncAnimation
@@ -29,11 +32,11 @@ from mpl_toolkits.mplot3d import Axes3D
 import random
 
 # Set random seeds for reproducibility
-np.random.seed({int(seed)})
-random.seed({int(seed)})
+np.random.seed({})
+random.seed({})
 
 # Animation parameters
-text_prompt = "{text_prompt.replace('"', '\\"')}"
+text_prompt = "{}"
 walking = "walk" in text_prompt.lower()
 running = "run" in text_prompt.lower()
 jumping = "jump" in text_prompt.lower()
@@ -42,7 +45,7 @@ turning = "turn" in text_prompt.lower() or "spin" in text_prompt.lower()
 waving = "wave" in text_prompt.lower()
 
 # Create a motion based on text prompt
-def generate_motion(frames={int(motion_length * 30)}):
+def generate_motion(frames={}):
     joints = 16  # Number of joints in a simplified skeleton
     dims = 3  # x, y, z
     
@@ -226,8 +229,14 @@ def visualize_motion(output_path):
     return output_path
 
 # Create and visualize a motion
-visualize_motion("{output_filename}")
-""")
+visualize_motion("{}")
+""".format(
+            int(seed),  # First {}
+            int(seed),  # Second {}
+            safe_prompt,  # Third {}
+            int(motion_length * 30),  # Fourth {}
+            output_filename  # Fifth {}
+        ))
     
     # Run the script
     subprocess.run(["python", "motion_animation.py"])