Update app.py

app.py

@@ -136,23 +136,16 @@ def translate_query(query, country):
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
     
-    # 검색어 번역
-    translated_query = translate_query(query, country)
-    print(f"Original query: {query}")
-    print(f"Translated query: {translated_query}")
-    
     payload = {
         "data": {
-            "q": translated_query,
+            "q": query,  # 원본 검색어 사용
             "domain": "google.com",
             "loc": COUNTRY_LOCATIONS.get(country, "United States"),
-            "lang": COUNTRY_LANGUAGES.get(country, "en"),
+            "lang": "en",  # 기본 영어로 설정
             "device": "desktop",
             "serp_type": "news",
             "page": "1",
-            "num": "10",
-            "verbatim": "0",
-            "gfilter": "0"
+            "num": "10"
         }
     }
 
@@ -168,9 +161,9 @@ def search_serphouse(query, country, page=1, num_result=10):
         print("Response status:", response.status_code)
         
         response.raise_for_status()
-        return response.json(), translated_query
+        return response.json()
     except requests.RequestException as e:
-        return {"error": f"Error: {str(e)}"}, query
+        return {"error": f"Error: {str(e)}"}
 
 def format_results_from_raw(results, translated_query):
     if isinstance(results, dict) and "error" in results:
@@ -205,6 +198,7 @@ css = """
 footer {visibility: hidden;}
 """
 
+# Gradio 인터페이스 수정
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as iface:
     gr.Markdown("검색어를 입력하고 원하는 국가를 선택하면, 검색어와 일치하는 24시간 이내 뉴스를 최대 100개 출력합니다.")
 
@@ -212,7 +206,11 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         with gr.Row():
             query = gr.Textbox(label="검색어")
             country = gr.Dropdown(MAJOR_COUNTRIES, label="국가", value="South Korea")
-            search_button = gr.Button("검색", variant="primary")
+            
+        # 번역 결과를 보여주는 컴포넌트 추가
+        translated_query_display = gr.Markdown(visible=False)
+            
+        search_button = gr.Button("검색", variant="primary")
 
         progress = gr.Progress()
         status_message = gr.Markdown(visible=False)
@@ -236,10 +234,17 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                 })
 
     def search_and_display(query, country, articles_state, progress=gr.Progress()):
-        progress(0, desc="검색 시작...")
+        progress(0, desc="검색어 번역 중...")
+        
+        # 검색어 번역
+        translated_query = translate_query(query, country)
+        translated_display = f"**원본 검색어:** {query}\n**번역된 검색어:** {translated_query}" if translated_query != query else f"**검색어:** {query}"
+        
+        progress(0.2, desc="검색 시작...")
         error_message, articles = serphouse_search(query, country)
         progress(0.5, desc="결과 처리 중...")
-        outputs = []
+        
+        outputs = [gr.update(value=translated_display, visible=True)]  # 번역 결과 표시
         
         if error_message:
             outputs.append(gr.update(value=error_message, visible=True))
@@ -258,12 +263,10 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                     article = articles[idx]
                     image_url = article['image_url']
                     image_update = gr.update(value=image_url, visible=True) if image_url and not image_url.startswith('data:image') else gr.update(value=None, visible=False)
-
-                    translated_info = f"\n**번역된 검색어:** {article['translated_query']}" if article['translated_query'] != query else ""
                     
                     outputs.extend([
                         gr.update(visible=True),
-                        gr.update(value=f"### [{article['title']}]({article['link']}){translated_info}"),
+                        gr.update(value=f"### [{article['title']}]({article['link']})"),
                         image_update,
                         gr.update(value=f"**요약:** {article['snippet']}"),
                         gr.update(value=f"**출처:** {article['channel']} | **시간:** {article['time']}")
@@ -280,7 +283,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         outputs.append(gr.update(visible=False))
         return outputs
 
-    search_outputs = [gr.Markdown(visible=False)]
+    search_outputs = [translated_query_display, gr.Markdown(visible=False)]  # 번역 결과 컴포넌트 추가
     for comp in article_components:
         search_outputs.extend([comp['group'], comp['title'], comp['image'],
                              comp['snippet'], comp['info']])