Update app.py

app.py

@@ -1119,10 +1119,7 @@ class EnhancedVideoGenerator:
 # Streamlit UI Class
 class VideoGeneratorUI:
     def __init__(self):
-        self.generator = EnhancedVideoGenerator()
-        # Create a base directory for video outputs
-        self.output_base_dir = "generated_videos"
-        os.makedirs(self.output_base_dir, exist_ok=True)
+        self.generator = VideoGenerator()
         self.setup_ui()
 
     def setup_ui(self):
@@ -1283,21 +1280,25 @@ class VideoGeneratorUI:
                 timestamp = int(time.time())
                 output_filename = f"vaultgenix_video_{timestamp}.mp4"
                 
-                # Create the full output path
-                output_path = os.path.join(self.output_base_dir, output_filename)
+                # Create absolute output path
+                output_path = os.path.abspath(os.path.join(
+                    self.generator.base_dir,
+                    output_filename
+                ))
                 
-                # Ensure the output directory exists
-                os.makedirs(os.path.dirname(output_path), exist_ok=True)
+                # Show progress message
+                progress_message = st.empty()
+                progress_message.text("🎥 Generating your video... This may take a few minutes.")
                 
-                video_path = self.generator.create_video(
+                video_path = self.generator.generate_video(
                     prompt,
-                    style,
+                    selected_images,
                     duration,
-                    output_path,
-                    selected_images
+                    output_path
                 )
                 
-                if video_path and os.path.exists(video_path):
+                if os.path.exists(video_path):
+                    progress_message.empty()
                     st.success("✨ Video generated successfully!")
                     
                     # Display video
@@ -1557,6 +1558,7 @@ class VideoGenerator:
             print(f"Cleanup error: {e}")
 
 def create_ui():
+    ui = VideoGeneratorUI()
     st.title("VaultGenix Video Generator")
     st.markdown("Create professional videos for your digital legacy management platform")