Create app.py

app.py

@@ -0,0 +1,54 @@
+import gradio as gr
+import requests
+from bs4 import BeautifulSoup
+from PIL import Image
+import io
+
+def search_image(image):
+    # Функция для поиска изображения в TinEye
+    def tineye_search(image):
+        url = "https://tineye.com/search"
+        files = {"image": ("image.jpg", image)}
+        response = requests.post(url, files=files)
+        soup = BeautifulSoup(response.text, "html.parser")
+        results = soup.find_all("div", class_="result-image")
+        return [result.find("img")["src"] for result in results[:5]]
+
+    # Функция для поиска изображения в Google Images
+    def google_search(image):
+        url = "https://images.google.com/searchbyimage/upload"
+        files = {"encoded_image": ("image.jpg", image)}
+        response = requests.post(url, files=files, allow_redirects=False)
+        search_url = response.headers["Location"]
+        response = requests.get(search_url)
+        soup = BeautifulSoup(response.text, "html.parser")
+        results = soup.find_all("img", class_="rg_i")
+        return [result["src"] for result in results[:5] if "src" in result.attrs]
+
+    # Преобразование изображения в байты
+    img_byte_arr = io.BytesIO()
+    image.save(img_byte_arr, format='JPEG')
+    img_byte_arr = img_byte_arr.getvalue()
+
+    # Поиск изображения
+    tineye_results = tineye_search(img_byte_arr)
+    google_results = google_search(img_byte_arr)
+
+    return tineye_results, google_results
+
+# Создание интерфейса Gradio
+iface = gr.Interface(
+    fn=search_image,
+    inputs=gr.Image(type="pil"),
+    outputs=[
+        gr.Gallery(label="TinEye Results"),
+        gr.Gallery(label="Google Images Results")
+    ],
+    title="Image Search Engine",
+    description="Upload an image to search for similar images using TinEye and Google Images.",
+    theme="huggingface",
+    layout="vertical"
+)
+
+# Запуск интерфейса
+iface.launch()