Update app.py

app.py

@@ -34,7 +34,6 @@ KOREAN_COMPANIES = [
     "investing" 
 ]
 
-
 #########################################################
 # 공통 함수
 #########################################################
@@ -129,6 +128,7 @@ def load_from_db(keyword, country):
         return json.loads(result[0]), convert_to_seoul_time(result[1])
     return None, None
 
+
 #########################################################
 # "id"로 직접 로딩하기 (히스토리에서 사용)
 #########################################################
@@ -168,6 +168,10 @@ def display_results(articles):
 # (1) 검색 시 => 기사 + 분석 동시 출력, DB 저장
 #########################################################
 def search_company(company):
+    """
+    지정된 KOREAN_COMPANIES 기업(또는 키워드)에 대해 
+    미국 뉴스 검색 -> 감성 분석 -> DB 저장 -> 기사+분석 보고 출력
+    """
     error_message, articles = serphouse_search(company, "United States")
     if not error_message and articles:
         analysis = analyze_sentiment_batch(articles, client)
@@ -185,6 +189,9 @@ def search_company(company):
 # (2) 출력 시 => DB에 저장된 기사 + 분석 함께 출력
 #########################################################
 def load_company(company):
+    """
+    DB에 이미 저장된 (기업명) 에 대한 기사+분석을 함께 출력
+    """
     data, timestamp = load_from_db(company, "United States")
     if data:
         articles = data.get("articles", [])
@@ -252,7 +259,6 @@ def show_stats():
     
     return output
 
-
 #########################################################
 # 전체 검색 (병렬) / 전체 출력 / 전체 리포트
 #########################################################
@@ -262,6 +268,7 @@ def search_all_companies():
     def do_search(comp):
         return comp, search_company(comp)
     
+    # 병렬 처리
     with ThreadPoolExecutor(max_workers=5) as executor:
         futures = [executor.submit(do_search, c) for c in KOREAN_COMPANIES]
         for future in as_completed(futures):
@@ -345,7 +352,6 @@ def get_custom_search_history():
     (id, timestamp, keyword, country) 형태로 반환
     최신순 정렬
     """
-    # set으로 만들어서 빠르게 검색
     company_set = set(k.lower() for k in KOREAN_COMPANIES)
     
     conn = sqlite3.connect("search_results.db")
@@ -360,9 +366,6 @@ def get_custom_search_history():
     for (sid, kw, cty, ts) in rows:
         # KOREAN_COMPANIES 에 없는 경우 -> 사용자 임의 검색
         if kw.lower() not in company_set:
-            # "id, 시간, 키워드(국가)" 로 표시할 텍스트 구성
-            # 예: "12 | 2025-01-22 10:23:00 KST | Apple (United States)"
-            # 실제 dropdown value 는 sid (id) 만 저장
             display_time = convert_to_seoul_time(ts)
             label = f"{sid} | {display_time} | {kw} ({cty})"
             history_list.append((str(sid), label))
@@ -384,7 +387,7 @@ def view_history_record(record_id):
     keyword = data["keyword"]
     country = data["country"]
     timestamp = data["timestamp"]
-    stored = data["data"]  # { articles, analysis }
+    stored = data["data"]
     
     articles = stored.get("articles", [])
     analysis = stored.get("analysis", "")
@@ -401,46 +404,12 @@ def view_history_record(record_id):
 
 
 #########################################################
-# SerpHouse API
+# SerpHouse API (함수명 수정) => serphouse_search
 #########################################################
-def is_english(text):
-    return all(ord(char) < 128 for char in text.replace(' ', '').replace('-', '').replace('_', ''))
-
-@lru_cache(maxsize=100)
-def translate_query(query, country):
-    try:
-        if is_english(query):
-            return query
-        
-        if country in COUNTRY_LANGUAGES:
-            if country == "South Korea":
-                return query
-            target_lang = COUNTRY_LANGUAGES[country]
-            
-            url = "https://translate.googleapis.com/translate_a/single"
-            params = {
-                "client": "gtx",
-                "sl": "auto",
-                "tl": target_lang,
-                "dt": "t",
-                "q": query
-            }
-            
-            session = requests.Session()
-            retries = Retry(total=3, backoff_factor=0.5)
-            session.mount('https://', HTTPAdapter(max_retries=retries))
-            
-            response = session.get(url, params=params, timeout=(5, 10))
-            translated_text = response.json()[0][0][0]
-            return translated_text
-            
-        return query
-        
-    except Exception as e:
-        print(f"번역 오류: {str(e)}")
-        return query
-
-def search_serphouse(query, country, page=1, num_result=10):
+def serphouse_search(query, country, page=1, num_result=10):
+    """
+    SerpHouse API 호출 + 결과 포매팅
+    """
     url = "https://api.serphouse.com/serp/live"
     
     now = datetime.utcnow()
@@ -492,23 +461,20 @@ def search_serphouse(query, country, page=1, num_result=10):
         )
         
         response.raise_for_status()
-        return {"results": response.json(), "translated_query": translated_query}
+        response_data = response.json()
+        
+        return format_results_from_raw({
+            "results": response_data,
+            "translated_query": translated_query
+        })
         
     except requests.exceptions.Timeout:
-        return {
-            "error": "검색 시간이 초과되었습니다. 잠시 후 다시 시도해주세요.",
-            "translated_query": query
-        }
+        return ("검색 시간이 초과되었습니다. 잠시 후 다시 시도해주세요.", [])
     except requests.exceptions.RequestException as e:
-        return {
-            "error": f"검색 중 오류가 발생했습니다: {str(e)}",
-            "translated_query": query
-        }
+        return (f"검색 중 오류가 발생했습니다: {str(e)}", [])
     except Exception as e:
-        return {
-            "error": f"예기치 않은 오류가 발생했습니다: {str(e)}",
-            "translated_query": query
-        }
+        return (f"예기치 않은 오류가 발생했습니다: {str(e)}", [])
+
 
 def format_results_from_raw(response_data):
     if "error" in response_data:
@@ -697,14 +663,14 @@ COUNTRY_LOCATIONS = {
     "Bangladesh": "Bangladesh",
     "Pakistan": "Pakistan",
     "Egypt": "Egypt",
-    "Morocco": "Morocco",
-    "Nigeria": "Nigeria",
-    "Kenya": "Kenya",
-    "Ukraine": "Ukraine",
-    "Croatia": "Croatia",
-    "Slovakia": "Slovakia",
-    "Bulgaria": "Bulgaria",
-    "Serbia": "Serbia",
+    "Morocco": "ar",
+    "Nigeria": "en",
+    "Kenya": "sw",
+    "Ukraine": "uk",
+    "Croatia": "hr",
+    "Slovakia": "sk",
+    "Bulgaria": "bg",
+    "Serbia": "sr",
     "Estonia": "et",
     "Latvia": "lv",
     "Lithuania": "lt",
@@ -897,7 +863,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
     
     with gr.Tabs():
         # 첫 번째 탭
-        with gr.Tab("Earnbot"):
+        with gr.Tab("지정 자동 검색/분석"):
             gr.Markdown("## EarnBot: 글로벌 빅테크 기업 및 투자 전망 AI 자동 분석")
             gr.Markdown(" - '전체 분석 보고 요약' 클릭 시 전체 자동 보고 생성.\n - 아래 개별 종목의 '검색(DB 자동 저장)'과 '출력(DB 자동 호출)'도 가능.\n - 하단 '수동 검색 히스토리'에서 이전에 수동 입력한 검색어 기록 확인 가능.")
             
@@ -958,7 +924,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                                         outputs=result_display
                                     )
             
-            # (추가) 수동 검색 히스토리
+            # (추가) 수동 검색 히스토리 (첫 번째 탭)
             gr.Markdown("---")
             gr.Markdown("### 수동 검색 히스토리")
             with gr.Row():
@@ -966,27 +932,14 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
             
             history_dropdown = gr.Dropdown(
                 label="검색 기록 목록",
-                choices=[],  # 이 부분은 click 이벤트에서 동적으로 업데이트
+                choices=[],
                 value=None
             )
             hist_view_btn = gr.Button("보기", variant="primary")
             hist_result_display = gr.Markdown()
             
-            # 1) 히스토리 갱신 함수
             def update_history_dropdown():
-                history_list = get_custom_search_history()  # [(id, label), ...]
-                labels = [lbl for (id_val, lbl) in history_list]
-                values = [id_val for (id_val, lbl) in history_list]
-                
-                # gr.Dropdown 에 choices=[..., ...] 형태로 전달해야 하므로
-                # zip 하여 (value, label) 쌍을 만듦
-                # Gradio 3.x에서는 choices를 list로 주고, label/value 분리 불가.
-                # -> workaround: "value|label" 식 or just store label in "choices" and store value separate
-                # 여기서는 choices 에 label만 넣고, id는 label 파싱 등 해야함.
-                # 좀 더 편하게: return a dictionary for "choices"
-                
-                # 방법1) label만 전달 -> user picks label -> then parse. 
-                # 간단히 "value=label"이 되도록:
+                history_list = get_custom_search_history()
                 choice_list = []
                 for (id_val, label) in history_list:
                     choice_list.append(label)
@@ -998,19 +951,13 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                 outputs=history_dropdown
             )
             
-            # 2) 히스토리 보기 함수
             def show_history_record(selected_label):
-                # selected_label 이 "id | time | keyword (country)" 형식
                 if not selected_label:
                     return "히스토리가 선택되지 않았습니다."
-                
-                # id는 가장 앞부분의 숫자
-                # 예: "12 | 2025-01-22 10:23:00 KST | Apple (United States)"
-                # -> id=12
                 splitted = selected_label.split("|")
                 if len(splitted) < 2:
                     return "형식 오류"
-                record_id = splitted[0].strip()  # "12"
+                record_id = splitted[0].strip()
                 return view_history_record(record_id)
             
             hist_view_btn.click(
@@ -1019,11 +966,10 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                 outputs=hist_result_display
             )
 
-
-        # 두 번째 탭: "지정 자동 검색/분석"
-        with gr.Tab("지정 자동 검색/분석"):
+        # 두 번째 탭: "수동 검색/분석" + 히스토리
+        with gr.Tab("수동 검색/분석"):
             gr.Markdown("## 사용자 임의 키워드 + 국가 검색/분석")
-            gr.Markdown("검색 결과가 DB에 저장되며, 첫 번째 탭의 '수동 검색 히스토리'에서 확인 가능합니다.")
+            gr.Markdown("검색 결과가 DB에 저장되며, 아래 '수동 검색 히스토리'에서도 확인 가능합니다.")
             
             with gr.Row():
                 with gr.Column():
@@ -1047,6 +993,33 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
                 inputs=[user_input, country_selection],
                 outputs=custom_search_output
             )
+            
+            # 동일하게, 두 번째 탭에도 히스토리 적용
+            gr.Markdown("---")
+            gr.Markdown("### 수동 검색 히스토리 (두 번째 탭)")
+            with gr.Row():
+                refresh_hist_btn2 = gr.Button("히스토리 갱신", variant="secondary")
+            
+            history_dropdown2 = gr.Dropdown(
+                label="검색 기록 목록",
+                choices=[],
+                value=None
+            )
+            hist_view_btn2 = gr.Button("보기", variant="primary")
+            hist_result_display2 = gr.Markdown()
+            
+            # 동일 로직 재사용
+            refresh_hist_btn2.click(
+                fn=update_history_dropdown,
+                inputs=[],
+                outputs=history_dropdown2
+            )
+            
+            hist_view_btn2.click(
+                fn=show_history_record,
+                inputs=[history_dropdown2],
+                outputs=hist_result_display2
+            )
 
 
 iface.launch(