Update app.py

app.py

@@ -201,81 +201,81 @@ class ImageScraper:
         ]
 
     def get_images(self, query: str, num_images: int = 15) -> Dict[str, List[Dict[str, str]]]:
-    try:
-        # Ensure query is not empty
-        if not query.strip():
-            query = "digital security"
-
-        base_url = "https://pixabay.com/api/"
-        params = {
-            'key': self.PIXABAY_API_KEY,
-            'q': query,
-            'image_type': 'photo',
-            'per_page': num_images,
-            'safesearch': True,
-            'lang': 'en'
-        }
-        
-        print(f"Querying Pixabay API with params: {params}")  # Debug log
-        response = requests.get(base_url, params=params)
-        print(f"API Response status: {response.status_code}")  # Debug log
-        
-        if response.status_code == 200:
-            data = response.json()
-            hits = data.get('hits', [])
-            print(f"Found {len(hits)} images")  # Debug log
-            
-            if not hits:
-                # If no results, try backup keywords
-                backup_keywords = ["digital security", "technology", "cyber security", "business technology"]
-                for keyword in backup_keywords:
-                    params['q'] = keyword
-                    backup_response = requests.get(base_url, params=params)
-                    if backup_response.status_code == 200:
-                        backup_data = backup_response.json()
-                        hits.extend(backup_data.get('hits', []))
-                        if hits:
-                            break
-
-            # Structure response
-            result = {
-                'primary': [],
-                'secondary': [],
-                'general': []
+        try:
+            # Ensure query is not empty
+            if not query.strip():
+                query = "digital security"
+    
+            base_url = "https://pixabay.com/api/"
+            params = {
+                'key': self.PIXABAY_API_KEY,
+                'q': query,
+                'image_type': 'photo',
+                'per_page': num_images,
+                'safesearch': True,
+                'lang': 'en'
             }
-
-            # Ensure we have at least some images
-            if not hits:
-                result['general'] = [{'url': url, 'keyword': 'technology', 'relevance': 'Fallback', 'tags': 'technology'} 
-                                   for url in self.get_stock_images()]
-                return result
-
-            # Categorize found images
-            for hit in hits[:num_images]:
-                image_data = {
-                    'url': hit['largeImageURL'],
-                    'keyword': query,
-                    'relevance': 'Primary match',
-                    'tags': hit.get('tags', '')
-                }
+            
+            print(f"Querying Pixabay API with params: {params}")  # Debug log
+            response = requests.get(base_url, params=params)
+            print(f"API Response status: {response.status_code}")  # Debug log
+            
+            if response.status_code == 200:
+                data = response.json()
+                hits = data.get('hits', [])
+                print(f"Found {len(hits)} images")  # Debug log
                 
-                if len(result['primary']) < num_images // 3:
-                    result['primary'].append(image_data)
-                elif len(result['secondary']) < num_images // 3:
-                    result['secondary'].append(image_data)
-                else:
-                    result['general'].append(image_data)
-
-            return result
-
-    except Exception as e:
-        print(f"Error in get_images: {str(e)}")  # Debug log
-        # Return stock images as fallback
-        return {
-            'general': [{'url': url, 'keyword': 'technology', 'relevance': 'Fallback', 'tags': 'technology'} 
-                       for url in self.get_stock_images()]
-        }
+                if not hits:
+                    # If no results, try backup keywords
+                    backup_keywords = ["digital security", "technology", "cyber security", "business technology"]
+                    for keyword in backup_keywords:
+                        params['q'] = keyword
+                        backup_response = requests.get(base_url, params=params)
+                        if backup_response.status_code == 200:
+                            backup_data = backup_response.json()
+                            hits.extend(backup_data.get('hits', []))
+                            if hits:
+                                break
+    
+                # Structure response
+                result = {
+                    'primary': [],
+                    'secondary': [],
+                    'general': []
+                }
+    
+                # Ensure we have at least some images
+                if not hits:
+                    result['general'] = [{'url': url, 'keyword': 'technology', 'relevance': 'Fallback', 'tags': 'technology'} 
+                                       for url in self.get_stock_images()]
+                    return result
     
+                # Categorize found images
+                for hit in hits[:num_images]:
+                    image_data = {
+                        'url': hit['largeImageURL'],
+                        'keyword': query,
+                        'relevance': 'Primary match',
+                        'tags': hit.get('tags', '')
+                    }
+                    
+                    if len(result['primary']) < num_images // 3:
+                        result['primary'].append(image_data)
+                    elif len(result['secondary']) < num_images // 3:
+                        result['secondary'].append(image_data)
+                    else:
+                        result['general'].append(image_data)
+    
+                return result
+    
+        except Exception as e:
+            print(f"Error in get_images: {str(e)}")  # Debug log
+            # Return stock images as fallback
+            return {
+                'general': [{'url': url, 'keyword': 'technology', 'relevance': 'Fallback', 'tags': 'technology'} 
+                           for url in self.get_stock_images()]
+            }
+        
         
     def get_fallback_keywords(self) -> List[Dict[str, str]]:
         """Return fallback keywords if AI extraction fails"""