aiqtest2 / app.py
seawolf2357's picture
Update app.py
bb2db31 verified
raw
history blame
1.18 kB
import gradio as gr
import requests
from PIL import Image
from io import BytesIO
# ๊ตฌ๊ธ€ ์ด๋ฏธ์ง€ ๊ฒ€์ƒ‰ ํ•จ์ˆ˜
def google_image_search(query):
API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY'
CX = '์—ฌ๊ธฐ์—_๋‹น์‹ ์˜_์ปค์Šคํ…€_๊ฒ€์ƒ‰์—”์ง„_ID๋ฅผ_์ž…๋ ฅํ•˜์„ธ์š”'
URL = f"https://www.googleapis.com/customsearch/v1?q={query}&cx={CX}&searchType=image&key={API_KEY}"
response = requests.get(URL)
results = response.json()
images = []
for item in results.get('items', []):
image_url = item['link']
response = requests.get(image_url)
img = Image.open(BytesIO(response.content))
images.append(img)
return images
# Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ •์˜
iface = gr.Interface(fn=google_image_search,
inputs=gr.inputs.Textbox(lines=2, placeholder="๊ฒ€์ƒ‰ํ•  ์ด๋ฏธ์ง€์˜ ํ…์ŠคํŠธ๋ฅผ ์ž…๋ ฅํ•˜์„ธ์š”..."),
outputs=gr.outputs.Image(type="pil", label="๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ"),
title="๊ตฌ๊ธ€ ์ด๋ฏธ์ง€ ๊ฒ€์ƒ‰",
description="ํ…์ŠคํŠธ๋ฅผ ์ž…๋ ฅํ•˜๋ฉด ๊ตฌ๊ธ€ ์ด๋ฏธ์ง€ ๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ๋ฅผ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค.")
# ์ธํ„ฐํŽ˜์ด์Šค ์‹คํ–‰
iface.launch()