Update app.py

app.py

@@ -172,20 +172,17 @@ def translate_query(query, country):
             target_lang = COUNTRY_LANGUAGES[country]
             print(f"번역 시도: {query} -> {country}({target_lang})")
             
-            # Google Translate API URL
             url = f"https://translate.googleapis.com/translate_a/single"
-            
             params = {
                 "client": "gtx",
-                "sl": "auto",  # 자동 언어 감지
-                "tl": target_lang,  # 대상 언어
+                "sl": "auto",
+                "tl": target_lang,
                 "dt": "t",
                 "q": query
             }
             
             response = requests.get(url, params=params)
             translated_text = response.json()[0][0][0]
-            
             print(f"번역 완료: {query} -> {translated_text} ({country})")
             return translated_text
             
@@ -195,16 +192,26 @@ def translate_query(query, country):
         print(f"번역 오류: {str(e)}")
         return query
 
-def is_english(text):
-    # 영어와 일반적인 기호만 포함된 경우 True 반환
-    return all(ord(char) < 128 for char in text.replace(' ', '').replace('-', '').replace('_', ''))
-
-    
-def is_korean(text):
-    return any('\uAC00' <= char <= '\uD7A3' for char in text)
+def translate_to_korean(text):
+    try:
+        url = "https://translate.googleapis.com/translate_a/single"
+        params = {
+            "client": "gtx",
+            "sl": "auto",
+            "tl": "ko",
+            "dt": "t",
+            "q": text
+        }
+        
+        response = requests.get(url, params=params)
+        translated_text = response.json()[0][0][0]
+        return translated_text
+    except Exception as e:
+        print(f"한글 번역 오류: {str(e)}")
+        return text
 
 def is_english(text):
-    return all(ord(char) < 128 for char in text.replace(' ', ''))
+    return all(ord(char) < 128 for char in text.replace(' ', '').replace('-', '').replace('_', ''))
     
 def is_korean(text):
     return any('\uAC00' <= char <= '\uD7A3' for char in text)
@@ -212,22 +219,20 @@ def is_korean(text):
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
     
-    # 24시간 이내 날짜 범위 설정
     now = datetime.utcnow()
     yesterday = now - timedelta(days=1)
     date_range = f"{yesterday.strftime('%Y-%m-%d')},{now.strftime('%Y-%m-%d')}"
     
-    # 검색어 번역
     translated_query = translate_query(query, country)
     print(f"Original query: {query}")
     print(f"Translated query: {translated_query}")
     
     payload = {
         "data": {
-            "q": translated_query,  # 여기를 번역된 검색어로 변경
+            "q": translated_query,
             "domain": "google.com",
             "loc": COUNTRY_LOCATIONS.get(country, "United States"),
-            "lang": COUNTRY_LANGUAGES.get(country, "en"),  # 언어 코드도 해당 국가 언어로 변경
+            "lang": COUNTRY_LANGUAGES.get(country, "en"),
             "device": "desktop",
             "serp_type": "news",
             "page": "1",
@@ -245,7 +250,7 @@ def search_serphouse(query, country, page=1, num_result=10):
 
     try:
         response = requests.post(url, json=payload, headers=headers)
-        print("Request payload:", json.dumps(payload, indent=2, ensure_ascii=False))  # ensure_ascii=False 추가
+        print("Request payload:", json.dumps(payload, indent=2, ensure_ascii=False))
         print("Response status:", response.status_code)
         
         response.raise_for_status()
@@ -291,23 +296,22 @@ footer {visibility: hidden;}
 
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as iface:
     gr.Markdown("검색어를 입력하고 원하는 국가(67개국)를 선택하면, 검색어와 일치하는 24시간 이내 뉴스를 최대 100개 출력합니다.")
-    gr.Markdown("국가 선택후 검색어에 '한글'을 입력하면 현지 언어로 번역되어 검색합니다. 예: 'Taiwan' 국가 선택후 '삼성' 입력시 '三星'으로 자동 검색 ")
+    gr.Markdown("국가 선택후 검색어에 '한글'을 입력하면 현지 언어로 번역되어 검색합니다. 예: 'Taiwan' 국가 선택후 '삼성' 입력시 '三星'으로 자동 검색")
     
     with gr.Column():
         with gr.Row():
             query = gr.Textbox(label="검색어")
             country = gr.Dropdown(MAJOR_COUNTRIES, label="국가", value="South Korea")
-            
-        # 검색 상태 메시지 컴포넌트 추가
-        search_status = gr.Markdown(visible=False)
+        
+        # 검색 상태 메시지    
+        status_message = gr.Markdown("", visible=True)
         
         # 번역 결과를 보여주는 컴포넌트
         translated_query_display = gr.Markdown(visible=False)
-            
+        
         search_button = gr.Button("검색", variant="primary")
-
+        
         progress = gr.Progress()
-        status_message = gr.Markdown(visible=False)
         articles_state = gr.State([])
 
         article_components = []
@@ -317,7 +321,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                 image = gr.Image(width=200, height=150)
                 snippet = gr.Markdown()
                 info = gr.Markdown()
-
+                
                 article_components.append({
                     'group': article_group,
                     'title': title,
@@ -328,8 +332,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                 })
 
     def search_and_display(query, country, articles_state, progress=gr.Progress()):
-        # 검색 시작 메시지 표시
-        search_status_output = gr.update(value="검색을 진행중입니다. 잠시만 기다리세요...", visible=True)
+        # 검색 상태 메시지 업데이트
+        status_msg = "검색을 진행중입니다. 잠시만 기다리세요..."
         
         progress(0, desc="검색어 번역 중...")
         
@@ -341,10 +345,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         error_message, articles = serphouse_search(query, country)
         progress(0.5, desc="결과 처리 중...")
         
-        outputs = [
-            search_status_output,  # 검색 상태 메시지
-            gr.update(value=translated_display, visible=True)  # 번역 결과 표시
-        ]
+        outputs = []
+        outputs.append(gr.update(value=status_msg, visible=True))  # 상태 메시지
+        outputs.append(gr.update(value=translated_display, visible=True))  # 번역 결과
         
         if error_message:
             outputs.append(gr.update(value=error_message, visible=True))
@@ -364,11 +367,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                     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)
                     
+                    # 요약 내용 한글 번역
+                    korean_summary = translate_to_korean(article['snippet'])
+                    
                     outputs.extend([
                         gr.update(visible=True),
                         gr.update(value=f"### [{article['title']}]({article['link']})"),
                         image_update,
-                        gr.update(value=f"**요약:** {article['snippet']}"),
+                        gr.update(value=f"**요약:** {article['snippet']}\n\n**한글 요약:** {korean_summary}"),
                         gr.update(value=f"**출처:** {article['channel']} | **시간:** {article['time']}")
                     ])
                 else:
@@ -380,23 +386,24 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
 
         progress(1.0, desc="완료!")
         outputs.append(articles_state)
-        outputs.append(gr.update(visible=False))
         
         # 검색 완료 후 상태 메시지 숨김
-        search_status_output = gr.update(visible=False)
-        outputs[0] = search_status_output
+        outputs[0] = gr.update(value="", visible=False)
         
         return outputs
 
     search_outputs = [
-        search_status,  # 검색 상태 메시지 출력 추가
-        translated_query_display, 
+        status_message,
+        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']])
-    search_outputs.extend([articles_state, status_message])
+        search_outputs.extend([
+            comp['group'], comp['title'], comp['image'],
+            comp['snippet'], comp['info']
+        ])
+    search_outputs.append(articles_state)
 
     search_button.click(
         search_and_display,