Update app.py

app.py

@@ -504,88 +504,42 @@ class VideoGeneratorUI:
 
     def setup_ui(self):
         st.title("Enhanced Video Generator")
-        st.write("Create professional videos with AI-generated content")
 
         with st.form("video_generator_form"):
-            # Input fields
-            prompt = st.text_area(
-                "Enter your video topic/prompt",
-                height=100,
-                help="Describe what you want your video to be about"
-            )
-            
+            prompt = st.text_area("Enter your video topic/prompt", height=100)
             col1, col2 = st.columns(2)
-            with col1:
-                style = st.selectbox(
-                    "Choose style",
-                    options=list(self.generator.themes.keys())
-                )
             
+            with col1:
+                style = st.selectbox("Choose style", options=list(self.generator.themes.keys()))
             with col2:
-                duration = st.slider(
-                    "Video duration (seconds)",
-                    min_value=10,
-                    max_value=300,
-                    value=60,
-                    step=10
-                )
-
-            advanced_options = st.expander("Advanced Options")
-            with advanced_options:
-                use_premium_voice = st.checkbox(
-                    "Use premium voice-over",
-                    value=False,
-                    help="Requires ElevenLabs API key"
-                )
-                
-                include_music = st.checkbox(
-                    "Include background music",
-                    value=True
-                )
-                
-                fps = st.slider(
-                    "Frames per second",
-                    min_value=24,
-                    max_value=60,
-                    value=30
-                )
+                duration = st.slider("Video duration (seconds)", 10, 300, 60, 10)
 
             submit_button = st.form_submit_button("Generate Video")
 
-            if submit_button:
-                if not prompt:
-                    st.error("Please enter a prompt for your video.")
-                    return
-
+            if submit_button and prompt:
                 try:
-                    with st.spinner("Generating your video..."):
-                        output_path = f"generated_video_{int(time.time())}.mp4"
-                        
-                        # Update generator settings based on advanced options
-                        self.generator.use_premium_voice = use_premium_voice
-                        
-                        # Generate video
-                        video_path = self.generator.create_video(
-                            prompt,
-                            style,
-                            duration,
-                            output_path
-                        )
-                        
-                        # Show success message and download button
+                    output_path = os.path.join(os.getcwd(), f"generated_video_{int(time.time())}.mp4")
+                    
+                    video_path = self.generator.create_video(prompt, style, duration, output_path)
+                    
+                    if os.path.exists(video_path):
                         st.success("Video generated successfully!")
                         
-                        with open(video_path, 'rb') as f:
-                            st.download_button(
-                                label="Download Video",
-                                data=f.read(),
-                                file_name=output_path,
-                                mime="video/mp4"
-                            )
+                        with open(video_path, 'rb') as video_file:
+                            st.video(video_file)  # Display video
+                            video_bytes = video_file.read()
+                            
+                        st.download_button(
+                            "Download Video",
+                            video_bytes,
+                            file_name=os.path.basename(video_path),
+                            mime="video/mp4"
+                        )
+                    else:
+                        st.error("Video file not found")
                         
                 except Exception as e:
                     st.error(f"Failed to generate video: {str(e)}")
-                    st.error("Please try again with different settings or contact support.")
 
 if __name__ == "__main__":
     ui = VideoGeneratorUI()