Spaces:
Sleeping
Sleeping
File size: 1,641 Bytes
887f5f0 | 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 | import requests
import base64
class GoogleLensScanner:
"""
Utility class to interact with Google Lens upload API.
Provides a free method to perform reverse image searches.
"""
def __init__(self):
self.upload_url = "https://lens.google.com/v3/upload"
def get_lens_url_for_image(self, file_path: str = None, base64_str: str = None) -> str:
"""
Uploads an image (either from file path or base64 string) to Google Lens
and returns the resulting Google Search URL.
"""
if file_path:
with open(file_path, "rb") as f:
image_data = f.read()
filename = file_path.split("/")[-1].split("\\")[-1]
elif base64_str:
# Handle potential data URI scheme
if "," in base64_str:
base64_str = base64_str.split(",")[1]
image_data = base64.b64decode(base64_str)
filename = "upload.jpg"
else:
raise ValueError("Must provide either file_path or base64_str")
files = {
'encoded_image': (filename, image_data, 'image/jpeg')
}
try:
# We allow redirects to capture the final Google Search URL in response.url
response = requests.post(self.upload_url, files=files, timeout=10, allow_redirects=True)
if response.status_code == 200:
return response.url
else:
raise Exception(f"Google Lens returned status code {response.status_code}")
except Exception as e:
print(f"Error during Google Lens upload: {e}")
return None
|