Update app.py

app.py

@@ -34,8 +34,10 @@ class ImageScraper:
             url = f"https://www.pexels.com/search/{query.replace(' ', '%20')}/"
             response = requests.get(url, headers=self.headers)
             soup = BeautifulSoup(response.text, 'html.parser')
-            for img in soup.find_all('img', src=True):
-                if 'photos' in img['src'] and 'pexels.com' in img['src']:
+            
+            # Updated selector to target image sources
+            for img in soup.find_all('img', {'data-image-width': True}):
+                if img.get('src') and 'photos' in img['src']:
                     urls.append(img['src'])
         except Exception as e:
             print(f"Pexels scraping error: {e}")
@@ -47,9 +49,12 @@ class ImageScraper:
             url = f"https://unsplash.com/s/photos/{query.replace(' ', '-')}"
             response = requests.get(url, headers=self.headers)
             soup = BeautifulSoup(response.text, 'html.parser')
-            for img in soup.find_all('img', src=True):
-                if 'images.unsplash.com' in img['src']:
-                    urls.append(img['src'])
+            
+            # Updated selector for Unsplash
+            for img in soup.find_all('img', {'srcset': True}):
+                src = img.get('src')
+                if src and 'images.unsplash.com' in src:
+                    urls.append(src)
         except Exception as e:
             print(f"Unsplash scraping error: {e}")
         return urls