Update app.py

app.py

@@ -1,45 +1,58 @@
-import os
-os.system("pip install opencv-python-headless")  # Ensure OpenCV is installed
-import cv2
+# 🔹 app.py (Optimized with Path Handling & Real-time Previews)
+
 import streamlit as st
 import tempfile
 import os
 import numpy as np
 import cv2
 import moviepy.editor as mp
-from model import generate_stick_animation, add_camera_effects, add_background, add_sound
+from model import AnimationGenerator
 
-# Streamlit UI
+# Initialize core system
 st.set_page_config(page_title="AI Stick Animation", layout="wide")
-st.title("🎭 AI Stick Animation Generator")
-
-# User Inputs
-story = st.text_area("📝 Enter Your Story:", "John is walking in a dark forest...")
-character_name = st.text_input("🎭 Enter Main Character Name:", "John")
-
-# Extra Features Selection
-background_choice = st.selectbox("🌆 Choose Background", ["None", "Dark Forest", "Haunted House", "Battlefield"])
-camera_effect = st.selectbox("📽️ Choose Camera Effect", ["None", "Shake", "Zoom", "Slow Motion"])
-sound_effect = st.selectbox("🎶 Choose Background Music", ["None", "Horror", "Action", "Comedy"])
-
-# Generate Animation Button
-if st.button("🚀 Generate Animation"):
-    with st.spinner("AI is generating animation..."):
-        output_frames = generate_stick_animation(story, character_name)
-
-        # Apply Extra Features
-        output_frames = add_camera_effects(output_frames, camera_effect)
-        output_frames = add_background(output_frames, background_choice)
-
-        # Save animation as MP4
-        temp_dir = tempfile.mkdtemp()
-        output_path = os.path.join(temp_dir, "animation.mp4")
-        clip = mp.ImageSequenceClip(output_frames, fps=24)  # High FPS for smooth motion
-        clip.write_videofile(output_path, codec="libx264", fps=24)
-
-        # Add Sound Effect
-        final_video = add_sound(output_path, sound_effect)
-
-        # Display Video
-        st.video(final_video)
-        st.download_button("📥 Download Animation", open(final_video, "rb"), file_name="stick_animation.mp4")
+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:
+                # Generate core animation
+                video_path = self.animator.create_animation(
+                    self.story,
+                    self.character,
+                    self.background,
+                    self.camera_effect,
+                    self.soundtrack,
+                    tmp_dir
+                )
+                
+                # Display results
+                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()