File size: 2,276 Bytes
73b98c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
import os, uuid, shutil
from flask import Flask, request, jsonify
from PIL import Image
import face_recognition
import requests
from io import BytesIO
from bs4 import BeautifulSoup
TMP_DIR = "temp"
os.makedirs(TMP_DIR, exist_ok=True)
app = Flask(__name__)
def scrape_yandex_reverse(image_bytes):
session = requests.Session()
headers = {"User-Agent": "Mozilla/5.0"}
temp_path = os.path.join(TMP_DIR, f"{uuid.uuid4().hex}.jpg")
with open(temp_path, "wb") as f:
f.write(image_bytes)
files = {"upfile": open(temp_path, "rb")}
upload_url = "https://yandex.com/images/search"
r = session.post(upload_url, files=files, headers=headers, allow_redirects=True)
search_url = r.url
r2 = session.get(search_url, headers=headers)
soup = BeautifulSoup(r2.text, "html.parser")
image_elements = soup.select("img.serp-item__thumb")
image_links = [img.get("src") for img in image_elements if img.get("src")]
return image_links[:10]
@app.route('/upload', methods=['POST'])
def upload():
if 'image' not in request.files:
return jsonify({"error": "No image uploaded"}), 400
uploaded = request.files['image']
image = face_recognition.load_image_file(uploaded)
try:
original_enc = face_recognition.face_encodings(image)[0]
except:
return jsonify({"error": "No face found"}), 400
matches = []
image_bytes = uploaded.read()
results = scrape_yandex_reverse(image_bytes)
for link in results:
try:
response = requests.get(link, timeout=6)
img = Image.open(BytesIO(response.content)).convert("RGB")
path = os.path.join(TMP_DIR, f"{uuid.uuid4().hex}.jpg")
img.save(path)
comp = face_recognition.load_image_file(path)
encs = face_recognition.face_encodings(comp)
for enc in encs:
if face_recognition.compare_faces([original_enc], enc, tolerance=0.5)[0]:
matches.append({"image_url": link, "source_url": link})
break
except:
continue
shutil.rmtree(TMP_DIR)
os.makedirs(TMP_DIR, exist_ok=True)
return jsonify({"matches": matches})
if __name__ == '__main__':
app.run(host="0.0.0.0", port=7860) |