Update app.py

app.py

@@ -1,34 +1,32 @@
 import streamlit as st
 from moviepy.editor import *
-from TTS.api import TTS
 import tempfile, os
+from transformers import pipeline
 
-st.title("Simple Text-to-Video App")
+st.title("Simplified Text-to-Video (no TTS temporarily)")
 
 @st.cache_resource()
-def load_tts():
-    return TTS("tts_models/en/ljspeech/tacotron2-DDC", progress_bar=False)
+def load_model():
+    return pipeline('text-to-video-generation', model='cerspense/zeroscope_v2_XS')
 
-tts_model = load_tts()
+video_gen = load_model()
 
-input_text = st.text_area("Enter very short text (max 100 chars):", max_chars=100)
+input_text = st.text_area("Enter short text (max 50 chars):", max_chars=50)
 
-if st.button("Generate Simple Video") and input_text:
-    with st.spinner("Generating..."):
-        # Audio Only (no cloning, fast)
-        audio_filename = tempfile.mktemp(".wav")
-        tts_model.tts_to_file(text=input_text, file_path=audio_filename)
+if st.button("Generate Video"):
+    if input_text:
+        with st.spinner("Generating video..."):
+            video_output = video_gen(input_text, num_frames=10)
+            video_tensor = video_output["video"]
+            video_np = (video_tensor * 255).astype('uint8')
 
-        # Single static image as video (for demonstration)
-        img_clip = ColorClip(size=(320, 240), color=(0, 0, 255)).set_duration(5)
-        audio_clip = AudioFileClip(audio_filename)
-        final_clip = img_clip.set_audio(audio_clip)
+            video_filename = tempfile.mktemp(".mp4")
+            clips = [ImageClip(frame).set_duration(0.3) for frame in video_np]
+            final_clip = concatenate_videoclips(clips, method="compose")
+            final_clip.write_videofile(video_filename, fps=5)
 
-        final_video_path = tempfile.mktemp(".mp4")
-        final_clip.write_videofile(final_video_path, fps=5)
+            st.video(video_filename)
 
-        st.video(final_video_path)
-
-        os.remove(audio_filename)
-        os.remove(final_video_path)
-        
+            os.remove(video_filename)
+    else:
+        st.warning("Please enter text.")