75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
|
from animo import Animo
|
||
|
|
import os
|
||
|
|
import time
|
||
|
|
from pprint import pprint
|
||
|
|
|
||
|
|
# Get API key from environment variable (recommended)
|
||
|
|
api_key = os.getenv("ANIMO_API_KEY")
|
||
|
|
|
||
|
|
# Initialize the Animo client
|
||
|
|
client = Animo(api_key=api_key)
|
||
|
|
|
||
|
|
# Define the prompt for generating a video
|
||
|
|
prompt = "Create a simple animation of a circle transforming into a square and use voiceover"
|
||
|
|
|
||
|
|
print("🤖 Generating video from prompt...")
|
||
|
|
|
||
|
|
try:
|
||
|
|
# Generate the video
|
||
|
|
response = client.videos.generate(
|
||
|
|
prompt=prompt,
|
||
|
|
engine="anthropic",
|
||
|
|
model="claude-3-7-sonnet-20250219"
|
||
|
|
)
|
||
|
|
|
||
|
|
# Print the initial response
|
||
|
|
print("\n📋 Initial Response:")
|
||
|
|
pprint(response)
|
||
|
|
|
||
|
|
# Get the request ID
|
||
|
|
request_id = response.get("requestId")
|
||
|
|
|
||
|
|
if not request_id:
|
||
|
|
print("\n❌ Error: No request ID in the response")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
print(f"\n⏳ Processing request ID: {request_id}")
|
||
|
|
print("Checking status every 5 seconds...")
|
||
|
|
|
||
|
|
# Poll for status until complete
|
||
|
|
while True:
|
||
|
|
# Wait 5 seconds between status checks
|
||
|
|
time.sleep(5)
|
||
|
|
|
||
|
|
# Check the status
|
||
|
|
status_response = client.videos.retrieve(request_id=request_id)
|
||
|
|
status = status_response.get("status")
|
||
|
|
|
||
|
|
print(f"Status: {status}")
|
||
|
|
|
||
|
|
# If there's an error, exit
|
||
|
|
if status_response.get("error"):
|
||
|
|
print(f"\n❌ Error: {status_response.get('error')}")
|
||
|
|
break
|
||
|
|
|
||
|
|
# If complete, show the video URL
|
||
|
|
if status == "COMPLETED":
|
||
|
|
video_url = status_response.get("videoUrl")
|
||
|
|
generated_code = status_response.get("generatedCode")
|
||
|
|
|
||
|
|
print("\n✅ Success! Your video is ready.")
|
||
|
|
print(f"Video URL: {video_url}")
|
||
|
|
|
||
|
|
if generated_code:
|
||
|
|
print("\n📝 Generated Manim Code:")
|
||
|
|
print(generated_code)
|
||
|
|
|
||
|
|
break
|
||
|
|
|
||
|
|
# If failed, exit
|
||
|
|
if status == "FAILED":
|
||
|
|
print("\n❌ Generation failed")
|
||
|
|
break
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"\n❌ Error: {str(e)}")
|