Fix animation script formatting issues

app.py

@@ -11,43 +11,12 @@ import requests
 import time
 import random
 
-def inspect_mdm_repo():
-    """Check the structure of the MDM repository"""
-    if not Path("motion-diffusion-model").exists():
-        print("MDM repository not found.")
-        return
-    
-    print("Inspecting MDM repository structure:")
-    # List top-level directories and files
-    for item in sorted(os.listdir("motion-diffusion-model"))[:5]:  # Limit to first 5 for brevity
-        path = os.path.join("motion-diffusion-model", item)
-        if os.path.isdir(path):
-            print(f"Directory: {item}")
-            # List files in the directory (up to 3)
-            for subitem in sorted(os.listdir(path))[:3]:
-                print(f"  - {subitem}")
-        else:
-            print(f"File: {item}")
-    print("...")  # Indicate truncated output
-
 def create_motion_animation(text_prompt, motion_length=3.0, seed=0):
     """Create a motion animation based on input parameters"""
     print(f"Creating animation for: '{text_prompt}', length: {motion_length}s, seed: {seed}")
     
-    # Use the seed for reproducibility
-    np.random.seed(int(seed))
-    random.seed(int(seed))
-    
-    # Parse the text prompt to influence the animation
-    walking = "walk" in text_prompt.lower()
-    running = "run" in text_prompt.lower()
-    jumping = "jump" in text_prompt.lower()
-    dancing = "danc" in text_prompt.lower()
-    turning = "turn" in text_prompt.lower() or "spin" in text_prompt.lower()
-    waving = "wave" in text_prompt.lower()
-    
     # Create a unique filename based on parameters
-    output_filename = f"output_{hash(text_prompt)}_{motion_length}_{seed}.mp4"
+    output_filename = f"output_{abs(hash(text_prompt) % 10000)}_{int(motion_length)}_{seed}.mp4"
     
     # Create a simple visualization script
     with open("motion_animation.py", "w") as f:
@@ -63,6 +32,15 @@ import random
 np.random.seed({int(seed)})
 random.seed({int(seed)})
 
+# Animation parameters
+text_prompt = "{text_prompt.replace('"', '\\"')}"
+walking = "walk" in text_prompt.lower()
+running = "run" in text_prompt.lower()
+jumping = "jump" in text_prompt.lower()
+dancing = "danc" in text_prompt.lower()
+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)}):
     joints = 16  # Number of joints in a simplified skeleton
@@ -70,14 +48,6 @@ def generate_motion(frames={int(motion_length * 30)}):
     
     motion = np.zeros((frames, joints, dims))
     
-    # Parse animation parameters from text
-    walking = {walking}
-    running = {running}
-    jumping = {jumping}
-    dancing = {dancing}
-    turning = {turning}
-    waving = {waving}
-    
     # Set speed based on motion type
     if running:
         speed = 4.0
@@ -224,7 +194,24 @@ def visualize_motion(output_path):
                    [motion_data[frame, start, 1], motion_data[frame, end, 1]],
                    [motion_data[frame, start, 2], motion_data[frame, end, 2]], 'r-')
         
-        ax.set_title(f"{repr('{text_prompt}')} - Frame {{frame}}")
+        # Set title
+        action_type = ""
+        if running:
+            action_type = "Running"
+        elif walking:
+            action_type = "Walking"
+        elif jumping:
+            action_type = "Jumping"
+        elif dancing:
+            action_type = "Dancing"
+        elif turning:
+            action_type = "Turning"
+        elif waving:
+            action_type = "Waving"
+        else:
+            action_type = "Moving"
+            
+        ax.set_title(f"{action_type} Motion - Frame {{frame}}")
         return ax
     
     # Create animation
@@ -253,9 +240,6 @@ visualize_motion("{output_filename}")
 def text_to_motion(text_prompt, motion_length=3.0, seed=0):
     """Generate motion from text prompt using MDM"""
     try:
-        # Inspect the MDM repository structure (limited output)
-        inspect_mdm_repo()
-        
         # Create a motion animation based on the input parameters
         return create_motion_animation(text_prompt, motion_length, seed)
         
@@ -268,7 +252,7 @@ def text_to_motion(text_prompt, motion_length=3.0, seed=0):
 demo = gr.Interface(
     fn=text_to_motion,
     inputs=[
-        gr.Textbox(label="Text Prompt", placeholder="A person walks forward, then turns left", lines=3),
+        gr.Textbox(label="Text Prompt", placeholder="A person walks forward, then turns left", lines=3, value="A person walking"),
         gr.Slider(minimum=1.0, maximum=9.8, value=3.0, label="Motion Length (seconds)"),
         gr.Number(label="Random Seed", value=0)
     ],