|
|
|
|
|
|
|
|
import streamlit as st |
|
|
import tempfile |
|
|
import os |
|
|
import numpy as np |
|
|
import cv2 |
|
|
import moviepy.editor as mp |
|
|
from model import AnimationGenerator |
|
|
|
|
|
|
|
|
st.set_page_config(page_title="AI Stick Animation", layout="wide") |
|
|
st.title("🎭 Ultra-Advanced AI Stick Animation") |
|
|
|
|
|
class AnimationApp: |
|
|
def __init__(self): |
|
|
self.animator = AnimationGenerator() |
|
|
self.setup_ui() |
|
|
|
|
|
def setup_ui(self): |
|
|
col1, col2 = st.columns([2, 1]) |
|
|
with col1: |
|
|
self.story = st.text_area("📝 Story Script:", |
|
|
"John walks cautiously through the dark forest...", height=200) |
|
|
self.character = st.text_input("🎭 Main Character:", "John") |
|
|
|
|
|
with col2: |
|
|
self.background = st.selectbox("🌆 Background", |
|
|
["None", "Dark Forest", "Haunted House", "City"]) |
|
|
self.camera_effect = st.selectbox("📽️ Camera Effect", |
|
|
["None", "Dynamic Shake", "Cinematic Zoom", "Slow Motion"]) |
|
|
self.soundtrack = st.selectbox("🎶 Soundtrack", |
|
|
["None", "Epic", "Suspense", "Upbeat"]) |
|
|
|
|
|
if st.button("🚀 Generate Animation"): |
|
|
self.process_animation() |
|
|
|
|
|
def process_animation(self): |
|
|
with st.spinner("🧠 AI is crafting your masterpiece..."): |
|
|
with tempfile.TemporaryDirectory() as tmp_dir: |
|
|
|
|
|
video_path = self.animator.create_animation( |
|
|
self.story, |
|
|
self.character, |
|
|
self.background, |
|
|
self.camera_effect, |
|
|
self.soundtrack, |
|
|
tmp_dir |
|
|
) |
|
|
|
|
|
|
|
|
st.success("✅ Animation Complete!") |
|
|
st.video(video_path) |
|
|
st.download_button("💾 Download", open(video_path, "rb"), |
|
|
file_name="ai_animation.mp4") |
|
|
|
|
|
if __name__ == "__main__": |
|
|
AnimationApp() |
