Spaces:
Sleeping
Sleeping
File size: 2,500 Bytes
e3d7308 |
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 requests
import os
from dotenv import load_dotenv
load_dotenv()
# Imgur and SERP API credentials
imgur_client_id = os.getenv('imgur_client_id')
serp_api_key = os.getenv('serp_api_key')
search_endpoint = 'https://serpapi.com/search'
def upload_image_to_imgur(image_path):
headers = {'Authorization': f'Client-ID {imgur_client_id}'}
data = {'image': open(image_path, 'rb').read()}
response = requests.post('https://api.imgur.com/3/image', headers=headers, files=data)
response_data = response.json()
if response.status_code == 200 and response_data['success']:
return response_data['data']['link']
else:
raise Exception(f"Error uploading image to Imgur: {response_data['data']['error']}")
def reverse_image_search(image_url):
params = {
'engine': 'google_reverse_image',
'image_url': image_url,
# "image_content": image_url,
'api_key': serp_api_key
}
response = requests.get(search_endpoint, params=params)
return response.json()
def extract_titles_and_descriptions(search_results, top_n=3):
titles_and_descriptions = []
for result in search_results.get('image_results', [])[:top_n]:
temp_dict = {}
title = result.get('title', '')
description = result.get('snippet', '')
temp_dict['title'] = title
temp_dict['description'] = description
titles_and_descriptions.append(temp_dict)
return titles_and_descriptions
def main(image_path):
# try:
# Upload image to Imgur and get the URL
image_url = upload_image_to_imgur(image_path)
print(f"Image uploaded to Imgur: {image_url}")
# Perform reverse image search
search_results = reverse_image_search(image_url)
if 'error' in search_results:
print("Error:", search_results['error'])
return
# Extract titles and descriptions
titles_and_descriptions = extract_titles_and_descriptions(search_results)
print(titles_and_descriptions)
# Print results
# for idx, (title, description) in enumerate(titles_and_descriptions):
# print(f"Result {idx+1}:")
# print("Title:", title)
# print("Description:", description)
# print("-" * 50)
# except Exception as e:
# print(f"An error occurred: {e}")
if __name__ == "__main__":
image_path = 'sampleImages/edited3.jpg' # Replace with the path to your local image
main(image_path)
|