Update app.py

app.py

@@ -1,38 +1,31 @@
+from serpapi import GoogleSearch
 import gradio as gr
-from serpapi import GoogleSearch # 수정된 부분이 없으나, 이는 문제의 원인을 정확히 파악하기 위한 예시입니다.
 
-# SERP API 키 설정
-API_KEY = "facc404e1b9394771101069c8ea6984719db4b437136cb33f7935e67dad7093f"
-
-def search_and_show_image(query):
+def search_images(query):
     params = {
         "engine": "google",
         "q": query,
-        "api_key": API_KEY,
-        "google_domain": "google.com",
-        "gl": "us",
-        "hl": "en"
+        "tbm": "isch",  # 이미지 검색을 위한 파라미터
+        "api_key": "56b76bc0db7f66e70958810f3486e99a7ad4fc9b4ad0719e34478b20d2f7ec4f"
     }
     
     search = GoogleSearch(params)
     results = search.get_dict()
-    organic_results = results.get("organic_results", [])
+    images_results = results.get("images_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()
+    # 이미지 URL만 추출
+    images_urls = [image["thumbnail"] for image in images_results]
     
-    submit_button.click(search_and_show_image, inputs=query_input, outputs=image_output)
+    return images_urls
+
+# Gradio 인터페이스 설정
+iface = gr.Interface(
+    fn=search_images,
+    inputs=gr.Textbox(lines=2, placeholder="검색할 키워드를 입력하세요..."),
+    outputs=gr.Gallery(label="검색 결과 이미지"),
+    title="SerpAPI 이미지 검색",
+    description="입력한 키워드에 대한 이미지 검색 결과를 보여줍니다."
+)
 
-# Gradio 앱 실행
-demo.launch()
+if __name__ == "__main__":
+    iface.launch()