Update app.py

app.py

@@ -909,30 +909,31 @@ class VideoGeneratorUI:
                     try:
                         # Get categorized images
                         image_categories = self.generator.image_scraper.get_images(prompt)
-                        
-                        if any(image_categories.values()):
+                            
+                        if image_categories and isinstance(image_categories, dict):  # Check if it's a dictionary
                             # Display primary matches
-                            if image_categories['primary']:
+                            if 'primary' in image_categories and image_categories['primary']:
                                 st.subheader("Most Relevant Images")
                                 self.display_image_grid(image_categories['primary'])
                             
                             # Display secondary matches
-                            if image_categories['secondary']:
+                            if 'secondary' in image_categories and image_categories['secondary']:
                                 st.subheader("Related Images")
                                 self.display_image_grid(image_categories['secondary'])
                             
                             # Display general/fallback images
-                            if image_categories['general']:
+                            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():
-                                for img in category:
-                                    key = f"img_{img['url']}"
-                                    if st.session_state.get(key, False):
-                                        selected_images.append(img['url'])
+                                if isinstance(category, list):  # Ensure category is a list
+                                    for img in category:
+                                        key = f"img_{img['url']}"
+                                        if st.session_state.get(key, False):
+                                            selected_images.append(img['url'])
                             
                             # Video generation section
                             if selected_images:
@@ -943,7 +944,7 @@ class VideoGeneratorUI:
                     except Exception as e:
                         st.error(f"An error occurred: {str(e)}")
                         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"""
         for i in range(0, len(images), cols):