Update app.py

app.py

@@ -1,7 +1,6 @@
 # Install necessary packages
 !pip install streamlit transformers diffusers gtts moviepy safetensors torch gradio
 
-# Import libraries
 import streamlit as st
 import torch
 from transformers import pipeline
@@ -65,11 +64,10 @@ prompt = st.text_input("Enter your prompt here:")
 if st.button("Generate Video"):
     if prompt:
         st.write("Generating video, please wait...")
-
-        # Generate the video
         video_path = generate_educational_video(prompt)
-
-        # Display the video in Streamlit
-        st.video(video_path)
+        if os.path.exists(video_path):
+            st.video(video_path)
+        else:
+            st.write("Video generation failed.")
     else:
         st.warning("Please enter a prompt.")