Update app.py

app.py

@@ -1,26 +1,28 @@
 import gradio as gr
 import requests
 from bs4 import BeautifulSoup
+import re
 
 def search_image(query):
-    # Format the search URL for images
+    # Create the URL for the image search
     url = f"https://www.google.com/search?hl=en&tbm=isch&q={query}"
     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 response
     soup = BeautifulSoup(response.text, 'html.parser')
-    
-    # Find all image tags and their data attributes for high-quality images
+
+    # Find image URLs from the <img> tags
     image_tags = soup.find_all("img")
-    
-    # Extract URLs; for Google Images, we need to look for larger images
+
+    # Extract the src URLs of the images
     image_urls = []
-    for img in image_tags[1:]:
+    for img in image_tags:
         img_url = img.get("src")
-        if img_url and "http" in img_url:
+        if img_url and img_url.startswith("http"):
+            # Append image URL to the list
             image_urls.append(img_url)
 
     # Return the first 5 high-quality images
@@ -30,7 +32,7 @@ def search_image(query):
 iface = gr.Interface(
     fn=search_image,
     inputs=gr.Textbox(placeholder="Type something to search for..."),
-    outputs=gr.Gallery(label="Search Results"),
+    outputs=gr.Gallery(label="Search Results", show_label=True),
     title="HD Image Search",
     description="Type in a search term to find HD images."
 )