Update app.py

app.py

@@ -1036,6 +1036,10 @@ class VideoGeneratorUI:
                 color: #666;
                 margin-top: 0.5rem;
             }
+            .submit-btn {
+                margin-top: 1rem;
+                padding: 0.5rem 1rem;
+            }
             </style>
         """, unsafe_allow_html=True)
 
@@ -1043,42 +1047,53 @@ class VideoGeneratorUI:
         st.markdown("Create professional videos for your digital legacy management platform")
         
         with st.container():
-            prompt = st.text_area("Enter your video script", height=200)
-            
-            if prompt:
-                with st.spinner("Analyzing prompt and fetching relevant images..."):
+            # Add form for prompt submission
+            with st.form(key='prompt_form'):
+                prompt = st.text_area("Enter your video script", height=200)
+                submit_button = st.form_submit_button(label='Analyze Script & Find Images')
+            
+            if submit_button and prompt:
+                # First show AI-selected images
+                with st.spinner("AI analyzing script and selecting relevant images..."):
                     try:
-                        # Get categorized images
+                        # Get AI-selected images first
+                        keywords = self.generator.image_scraper.extract_key_topics(prompt)
+                        st.write("🤖 AI-detected keywords:", ", ".join(keywords))
+                        
                         image_categories = self.generator.image_scraper.get_images(prompt)
-                            
-                        if image_categories and isinstance(image_categories, dict):  # Check if it's a dictionary
-                            # Display primary matches
+                        
+                        # Store selections in session state
+                        if 'selected_images' not in st.session_state:
+                            st.session_state.selected_images = []
+                        
+                        if image_categories and isinstance(image_categories, dict):
+                            # Display AI-selected primary matches first
                             if 'primary' in image_categories and image_categories['primary']:
-                                st.subheader("Most Relevant Images")
+                                st.subheader("🎯 AI-Selected Most Relevant Images")
                                 self.display_image_grid(image_categories['primary'])
                             
                             # Display secondary matches
                             if 'secondary' in image_categories and image_categories['secondary']:
-                                st.subheader("Related Images")
+                                st.subheader("🔄 AI-Selected Related Images")
                                 self.display_image_grid(image_categories['secondary'])
                             
-                            # Display general/fallback images
-                            if 'general' in image_categories and image_categories['general']:
-                                st.subheader("Additional Suggested Images")
-                                self.display_image_grid(image_categories['general'])
-                            
                             # Collect selected images
                             selected_images = []
                             for category in image_categories.values():
-                                if isinstance(category, list):  # Ensure category is a list
+                                if isinstance(category, list):
                                     for img in category:
                                         key = f"img_{img['url']}"
                                         if st.session_state.get(key, False):
                                             selected_images.append(img['url'])
                             
+                            st.session_state.selected_images = selected_images
+                            
                             # Video generation section
                             if selected_images:
                                 self.show_video_settings(prompt, selected_images)
+                            else:
+                                st.warning("Please select at least one image to generate the video.")
+                        
                         else:
                             st.warning("No images found. Please try a different prompt.")
                             
@@ -1087,16 +1102,13 @@ class VideoGeneratorUI:
                         print(f"Error in UI: {str(e)}")
     
     def display_image_grid(self, images: List[Dict[str, str]], cols: int = 3):
-        """Display images in a grid with metadata"""
-        # Ensure images is a list and not empty
+        """Display images in a grid with metadata and confidence scores"""
         if not images or not isinstance(images, list):
             return
         
-        # Calculate number of rows needed
         n_images = len(images)
         n_rows = (n_images + cols - 1) // cols
         
-        # Create grid
         for row in range(n_rows):
             with st.container():
                 columns = st.columns(cols)
@@ -1107,15 +1119,23 @@ class VideoGeneratorUI:
                         with columns[col]:
                             try:
                                 st.image(img['url'], use_container_width=True)
+                                
+                                # Add confidence score to checkbox label
+                                confidence = img.get('relevance_score', 0) * 100
+                                checkbox_label = f"Select (AI Confidence: {confidence:.1f}%)"
+                                
                                 st.checkbox(
-                                    "Select",
+                                    checkbox_label,
                                     key=f"img_{img['url']}",
                                     help=f"Keywords: {img['keyword']}\nTags: {img['tags']}"
                                 )
+                                
+                                # Show relevance metadata
                                 st.markdown(
                                     f"<div class='image-metadata'>"
-                                    f"Relevance: {img['relevance']}<br>"
-                                    f"Keywords: {img['keyword']}"
+                                    f"<b>AI Relevance:</b> {img['relevance']}<br>"
+                                    f"<b>Keywords:</b> {img['keyword']}<br>"
+                                    f"<b>Match Type:</b> {img.get('category', 'General')}"
                                     f"</div>",
                                     unsafe_allow_html=True
                                 )
@@ -1125,30 +1145,48 @@ class VideoGeneratorUI:
     def show_video_settings(self, prompt: str, selected_images: List[str]):
         """Show video generation settings and controls"""
         st.subheader("Video Settings")
-        col1, col2 = st.columns(2)
-        with col1:
-            style = st.selectbox(
-                "Choose style",
-                options=["Professional", "Creative", "Educational"],
-                index=0
-            )
-        with col2:
-            duration = st.slider(
-                "Video duration (seconds)",
-                min_value=30,
-                max_value=180,
-                value=60,
-                step=30
+        
+        with st.form(key='video_settings_form'):
+            col1, col2 = st.columns(2)
+            with col1:
+                style = st.selectbox(
+                    "Choose style",
+                    options=["Professional", "Creative", "Educational"],
+                    index=0
+                )
+            with col2:
+                duration = st.slider(
+                    "Video duration (seconds)",
+                    min_value=30,
+                    max_value=180,
+                    value=60,
+                    step=30
+                )
+                
+            generate_button = st.form_submit_button(
+                label="🎬 Generate Video",
+                type="primary"
             )
 
-        if st.button("Generate Video", type="primary"):
-            self.generate_video(prompt, style, duration, selected_images)
+            if generate_button:
+                self.generate_video(prompt, style, duration, selected_images)
 
     def generate_video(self, prompt: str, style: str, duration: int, selected_images: List[str]):
-        """Handle video generation"""
-        with st.spinner("Generating your video..."):
+        """Handle video generation with improved error handling"""
+        if not selected_images:
+            st.error("Please select at least one image before generating the video.")
+            return
+            
+        with st.spinner("🎥 Generating your video..."):
             try:
-                output_path = f"vaultgenix_video_{int(time.time())}.mp4"
+                # Create temp directory if it doesn't exist
+                temp_dir = Path("temp_videos")
+                temp_dir.mkdir(exist_ok=True)
+                
+                # Generate unique output path
+                output_path = str(temp_dir / f"vaultgenix_video_{int(time.time())}.mp4")
+                
+                # Generate video
                 video_path = self.generator.create_video(
                     prompt,
                     style,
@@ -1157,20 +1195,37 @@ class VideoGeneratorUI:
                     selected_images
                 )
                 
-                if os.path.exists(video_path):
+                if video_path and os.path.exists(video_path):
                     st.success("✨ Video generated successfully!")
-                    st.video(video_path)
                     
-                    with open(video_path, 'rb') as video_file:
-                        st.download_button(
-                            "⬇️ Download Video",
-                            video_file.read(),
-                            file_name=output_path,
-                            mime="video/mp4"
-                        )
+                    # Display video
+                    video_file = open(video_path, 'rb')
+                    video_bytes = video_file.read()
+                    st.video(video_bytes)
+                    
+                    # Download button
+                    st.download_button(
+                        label="⬇️ Download Video",
+                        data=video_bytes,
+                        file_name=os.path.basename(video_path),
+                        mime="video/mp4"
+                    )
+                    
+                    video_file.close()
+                else:
+                    st.error("Video generation failed. Please try again.")
+                    
             except Exception as e:
                 st.error(f"Failed to generate video: {str(e)}")
                 print(f"Video generation error: {str(e)}")
+                
+            finally:
+                # Cleanup temporary files
+                try:
+                    if 'video_file' in locals():
+                        video_file.close()
+                except Exception as e:
+                    print(f"Cleanup error: {e}")
 
 if __name__ == "__main__":
     ui = VideoGeneratorUI()