Update app.py

app.py

@@ -1107,8 +1107,6 @@ class VideoGeneratorUI:
         st.set_page_config(layout="wide")
     
         # Initialize session state
-        if 'selected_images' not in st.session_state:
-            st.session_state.selected_images = []
         if 'processing_complete' not in st.session_state:
             st.session_state.processing_complete = False
         if 'current_step' not in st.session_state:
@@ -1128,8 +1126,6 @@ class VideoGeneratorUI:
         with st.container():
             if st.session_state.current_step == 'input':
                 self.show_input_form()
-            elif st.session_state.current_step == 'image_selection':
-                self.show_image_selection()
             elif st.session_state.current_step == 'video_generation':
                 self.show_video_generation()
 
@@ -1148,13 +1144,52 @@ class VideoGeneratorUI:
                     ["Professional Male", "Professional Female", "Casual Male", "Casual Female"]
                 )
             
-            submit_button = st.form_submit_button(label='Find Images')
+            submit_button = st.form_submit_button(label='Generate Video')
             
             if submit_button and prompt:
                 st.session_state.prompt = prompt
-                st.session_state.current_step = 'image_selection'
+                # Automatically select images based on AI analysis
+                with st.spinner("AI analyzing script and selecting relevant images..."):
+                    self.auto_select_images()
+                st.session_state.current_step = 'video_generation'
                 st.rerun()
 
+    def auto_select_images(self):
+        """Automatically select the most relevant images based on AI analysis"""
+        try:
+            keywords = self.generator.image_scraper.extract_key_topics(st.session_state.prompt)
+            st.write("🤖 AI-detected keywords:", ", ".join(keywords))
+            
+            image_categories = self.generator.image_scraper.get_images(st.session_state.prompt)
+            selected_images = []
+            
+            if image_categories and isinstance(image_categories, dict):
+                # Select top images from each category based on relevance score
+                for category, images in image_categories.items():
+                    # Sort images by relevance score
+                    sorted_images = sorted(images, key=lambda x: x.get('relevance_score', 0), reverse=True)
+                    
+                    # Select top images from each category
+                    num_to_select = {
+                        'primary': 3,
+                        'secondary': 2,
+                        'general': 1
+                    }.get(category.lower(), 1)
+                    
+                    selected = [img['url'] for img in sorted_images[:num_to_select]]
+                    selected_images.extend(selected)
+            
+            if not selected_images:
+                # Fallback to stock images if no images were selected
+                selected_images = self.generator.image_scraper.get_stock_images()[:6]
+            
+            st.session_state.selected_images = selected_images
+            
+        except Exception as e:
+            st.error(f"Error in image selection: {str(e)}")
+            # Fallback to stock images
+            st.session_state.selected_images = self.generator.image_scraper.get_stock_images()[:6]
+
     def show_image_selection(self):
         st.subheader("AI-Selected Images for Your Video")