from serpapi import GoogleSearch
import gradio as gr

def search_images(query):
    params = {
        "engine": "google",
        "q": query,
        "tbm": "isch",  # 이미지 검색을 위한 파라미터
        "api_key": "56b76bc0db7f66e70958810f3486e99a7ad4fc9b4ad0719e34478b20d2f7ec4f"
    }
    
    search = GoogleSearch(params)
    results = search.get_dict()
    images_results = results.get("images_results", [])
    
    # 이미지 URL만 추출
    images_urls = [image["thumbnail"] for image in images_results]
    
    return images_urls

# Gradio 인터페이스 설정
iface = gr.Interface(
    fn=search_images,
    inputs=gr.Textbox(lines=2, placeholder="검색할 키워드를 입력하세요..."),
    outputs=gr.Gallery(label="검색 결과 이미지"),
    title="SerpAPI 이미지 검색",
    description="입력한 키워드에 대한 이미지 검색 결과를 보여줍니다."
)

if __name__ == "__main__":
    iface.launch()