import gradio as gr
from serpapi import GoogleSearch # 수정된 부분이 없으나, 이는 문제의 원인을 정확히 파악하기 위한 예시입니다.

# SERP API 키 설정
API_KEY = "facc404e1b9394771101069c8ea6984719db4b437136cb33f7935e67dad7093f"

def search_and_show_image(query):
    params = {
        "engine": "google",
        "q": query,
        "api_key": API_KEY,
        "google_domain": "google.com",
        "gl": "us",
        "hl": "en"
    }
    
    search = GoogleSearch(params)
    results = search.get_dict()
    organic_results = results.get("organic_results", [])
    
    if organic_results:
        first_result = organic_results[0]
        screenshot_url = first_result.get("screenshot", {}).get("src", "No screenshot available for this result.")
        return screenshot_url
    else:
        return "No results found."

# Gradio 인터페이스 구성
with gr.Blocks() as demo:
    with gr.Row():
        query_input = gr.Textbox(label="검색어 입력")
        submit_button = gr.Button("검색")
    image_output = gr.Image()
    
    submit_button.click(search_and_show_image, inputs=query_input, outputs=image_output)

# Gradio 앱 실행
demo.launch()