Update app.py

app.py

@@ -3,27 +3,36 @@ import requests
 from bs4 import BeautifulSoup
 
 def search_image(query):
-    # Use a search engine to find images
+    # Format the search URL for images
     url = f"https://www.google.com/search?hl=en&tbm=isch&q={query}"
-    response = requests.get(url)
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+    }
+    response = requests.get(url, headers=headers)
     
-    # Parse the HTML
+    # Parse the HTML response
     soup = BeautifulSoup(response.text, 'html.parser')
     
-    # Find the image URLs
+    # Find all image tags and their data attributes for high-quality images
     image_tags = soup.find_all("img")
     
-    # Get the URLs of the images (skip the first one, as it's a placeholder)
-    image_urls = [img['src'] for img in image_tags[1:6]]  # Get the first 5 images
-    return image_urls
+    # Extract URLs; for Google Images, we need to look for larger images
+    image_urls = []
+    for img in image_tags[1:]:
+        img_url = img.get("src")
+        if img_url and "http" in img_url:
+            image_urls.append(img_url)
+
+    # Return the first 5 high-quality images
+    return image_urls[:5]
 
 # Create the Gradio interface
 iface = gr.Interface(
     fn=search_image,
     inputs=gr.Textbox(placeholder="Type something to search for..."),
     outputs=gr.Gallery(label="Search Results"),
-    title="Image Search",
-    description="Type in a search term to find images."
+    title="HD Image Search",
+    description="Type in a search term to find HD images."
 )
 
 # Launch the app