Update app.py

app.py

@@ -27,21 +27,7 @@ COUNTRY_LANGUAGES = {
     "Saudi Arabia": "ar",
     "United Arab Emirates": "ar",
     "Egypt": "ar",
-    "Morocco": "ar",
-    "Greece": "el",
-    "Poland": "pl",
-    "Czech Republic": "cs",
-    "Hungary": "hu",
-    "Turkey": "tr",
-    "Romania": "ro",
-    "Bulgaria": "bg",
-    "Croatia": "hr",
-    "Serbia": "sr",
-    "Slovakia": "sk",
-    "Slovenia": "sl",
-    "Estonia": "et",
-    "Latvia": "lv",
-    "Lithuania": "lt"
+    "Morocco": "ar"
 }
 
 COUNTRY_LOCATIONS = {
@@ -124,10 +110,9 @@ def translate_query(query, country):
             
             translated = hf_client.text_generation(
                 prompt,
-                max_new_tokens=50,  # 번역된 텍스트만 필요하므로 토큰 수 감소
-                temperature=0.1     # 정확한 번역을 위해 temperature 낮춤
+                max_new_tokens=50,
+                temperature=0.1
             )
-            # 번역된 텍스트만 추출
             translated = translated.strip()
             print(f"Original query: {query}")
             print(f"Translated query: {translated}")
@@ -140,12 +125,11 @@ 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)
     
     payload = {
         "data": {
-            "q": translated_query,  # 번역된 검색어 사용
+            "q": translated_query,
             "domain": "google.com",
             "loc": COUNTRY_LOCATIONS.get(country, "United States"),
             "lang": "en",
@@ -208,22 +192,18 @@ css = """
 footer {visibility: hidden;}
 """
 
-# Gradio 인터페이스 수정
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as iface:
-    gr.Markdown("검색어를 입력하고 원하는 국가를 선택하면, 검색어와 일치하는 24시간 이내 뉴스를 최대 100개 출력합니다.")
-
+    gr.Markdown("검색어를 입력하고 원하는 국가를 선택하면, 해당 국가의 언어로 번역된 검색어로 뉴스를 검색합니다.")
 
     with gr.Column():
         with gr.Row():
             query = gr.Textbox(label="검색어")
             country = gr.Dropdown(MAJOR_COUNTRIES, label="국가", value="South Korea")
             
-        # 번역된 검색어 표시
-        translated_display = gr.Markdown(visible=True, label="번역된 검색어")
+        # 번역된 검색어 표시 컴포넌트
+        translated_display = gr.Markdown(visible=True)
             
         search_button = gr.Button("검색", variant="primary")
-        
-
 
         progress = gr.Progress()
         status_message = gr.Markdown(visible=False)
@@ -248,21 +228,18 @@ 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="검색어 번역 중...")
-    
-    # 검색어 번역
+        
         translated_query = translate_query(query, country)
         if translated_query != query:
             translated_display_text = f"원본 검색어: {query}\n번역된 검색어: {translated_query}"
         else:
             translated_display_text = f"검색어: {query}"
-    
+        
         progress(0.2, desc="검색 시작...")
         response_data = search_serphouse(query, country)
         error_message, articles = format_results_from_raw(response_data)
         
-        outputs = [gr.update(value=translated_display_text, visible=True)]  # 번역 결과 표시
-                
-
+        outputs = [gr.update(value=translated_display_text, visible=True)]
         
         if error_message:
             outputs.append(gr.update(value=error_message, visible=True))
@@ -301,14 +278,21 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         outputs.append(gr.update(visible=False))
         return outputs
 
-    search_outputs = [translated_query_display, gr.Markdown(visible=False)]  # 번역 결과 컴포넌트 추가
+    # 검색 버튼 클릭 시 업데이트될 출력 컴포넌트 목록
+    search_outputs = [translated_display, gr.Markdown(visible=False)]  # 번역 결과 표시 컴포넌트
     for comp in article_components:
-        search_outputs.extend([comp['group'], comp['title'], comp['image'],
-                             comp['snippet'], comp['info']])
+        search_outputs.extend([
+            comp['group'],
+            comp['title'],
+            comp['image'],
+            comp['snippet'],
+            comp['info']
+        ])
     search_outputs.extend([articles_state, status_message])
 
+    # 검색 버튼 클릭 이벤트 설정
     search_button.click(
-        search_and_display,
+        fn=search_and_display,
         inputs=[query, country, articles_state],
         outputs=search_outputs,
         show_progress=True