Update app.py

app.py

@@ -36,21 +36,14 @@ KOREAN_COMPANIES = [
 
 def convert_to_seoul_time(timestamp_str):
     try:
-        # 입력된 시간을 naive datetime 객체로 변환
         dt = datetime.strptime(timestamp_str, '%Y-%m-%d %H:%M:%S')
-        
-        # 서울 시간대 설정
         seoul_tz = pytz.timezone('Asia/Seoul')
-        
-        # 현재 시간을 서울 시간으로 인식하도록 수정
         seoul_time = seoul_tz.localize(dt)
-        
         return seoul_time.strftime('%Y-%m-%d %H:%M:%S KST')
     except Exception as e:
         print(f"시간 변환 오류: {str(e)}")
         return timestamp_str
 
-
 def analyze_sentiment_batch(articles, client):
     try:
         # 모든 기사의 제목과 내용을 하나의 텍스트로 결합
@@ -58,8 +51,9 @@ def analyze_sentiment_batch(articles, client):
             f"제목: {article.get('title', '')}\n내용: {article.get('snippet', '')}"
             for article in articles
         ])
-
-        prompt = f\"""다음 뉴스 모음에 대해 전반적인 감성 분석을 수행하세요:
+        
+        # f""" ... """ 형태로 여러 줄 문자열을 정확히 사용
+        prompt = f"""다음 뉴스 모음에 대해 전반적인 감성 분석을 수행하세요:
 
 뉴스 내용:
 {combined_text}
@@ -74,7 +68,7 @@ def analyze_sentiment_batch(articles, client):
    - [항목2]
 4. 종합 평가: [상세 설명]
 """
-        
+
         response = client.chat.completions.create(
             model="CohereForAI/c4ai-command-r-plus-08-2024",
             messages=[{"role": "user", "content": prompt}],
@@ -101,27 +95,20 @@ def init_db():
     conn.commit()
     conn.close()
 
-
 def save_to_db(keyword, country, results):
     conn = sqlite3.connect("search_results.db")
     c = conn.cursor()
-    
-    # 현재 시간을 서울 시간으로 가져오기
     seoul_tz = pytz.timezone('Asia/Seoul')
     now = datetime.now(seoul_tz)
-    
-    # 시간대 정보를 제거하고 저장
     timestamp = now.strftime('%Y-%m-%d %H:%M:%S')
     
     c.execute("""INSERT INTO searches 
                  (keyword, country, results, timestamp) 
                  VALUES (?, ?, ?, ?)""",
               (keyword, country, json.dumps(results), timestamp))
-    
     conn.commit()
     conn.close()
 
-
 def load_from_db(keyword, country):
     conn = sqlite3.connect("search_results.db")
     c = conn.cursor()
@@ -133,7 +120,6 @@ def load_from_db(keyword, country):
         return json.loads(result[0]), convert_to_seoul_time(result[1])
     return None, None
 
-
 def display_results(articles):
     output = ""
     for idx, article in enumerate(articles, 1):
@@ -144,7 +130,6 @@ def display_results(articles):
         output += f"요약: {article['snippet']}\n\n"
     return output
 
-
 def search_company(company):
     error_message, articles = serphouse_search(company, "United States")
     if not error_message and articles:
@@ -152,14 +137,12 @@ def search_company(company):
         return display_results(articles)
     return f"{company}에 대한 검색 결과가 없습니다."
 
-
 def load_company(company):
     results, timestamp = load_from_db(company, "United States")
     if results:
         return f"### {company} 검색 결과\n저장 시간: {timestamp}\n\n" + display_results(results)
     return f"{company}에 대한 저장된 결과가 없습니다."
 
-
 def show_stats():
     conn = sqlite3.connect("search_results.db")
     c = conn.cursor()
@@ -186,7 +169,6 @@ def show_stats():
             output += f"- 저장된 기사 수: {len(articles)}건\n\n"
             
             if articles:
-                # 전체 기사에 대한 감성 분석
                 sentiment_analysis = analyze_sentiment_batch(articles, client)
                 output += "#### 뉴스 감성 분석\n"
                 output += f"{sentiment_analysis}\n\n"
@@ -196,11 +178,7 @@ def show_stats():
     conn.close()
     return output
 
-
 def search_all_companies():
-    """
-    등록된 모든 기업에 대해 순차적으로 검색 수행 후 결과를 합쳐서 반환
-    """
     overall_result = "# [전체 검색 결과]\n\n"
     for comp in KOREAN_COMPANIES:
         overall_result += f"## {comp}\n"
@@ -209,9 +187,6 @@ def search_all_companies():
     return overall_result
 
 def load_all_companies():
-    """
-    등록된 모든 기업에 대해 DB에 저장된 결과를 순차적으로 불러와서 합쳐서 반환
-    """
     overall_result = "# [전체 출력 결과]\n\n"
     for comp in KOREAN_COMPANIES:
         overall_result += f"## {comp}\n"
@@ -220,21 +195,13 @@ def load_all_companies():
     return overall_result
 
 def full_summary_report():
-    """
-    '전체 분석 보고 요약' 버튼 클릭 시,
-    1) 전체 검색 -> 2) 전체 출력 -> 3) 전체 통계(감성 분석)
-    순으로 실행하고 그 결과를 한 번에 리턴
-    """
     # 1) 전체 검색
     search_result_text = search_all_companies()
-    
     # 2) 전체 출력
     load_result_text = load_all_companies()
-    
     # 3) 전체 통계
     stats_text = show_stats()
     
-    # 최종 보고서 형태로 합쳐서 반환
     combined_report = (
         "# 전체 분석 보고 요약\n\n"
         "아래 순서로 실행되었습니다:\n"
@@ -246,7 +213,6 @@ def full_summary_report():
     )
     return combined_report
 
-
 ACCESS_TOKEN = os.getenv("HF_TOKEN")
 if not ACCESS_TOKEN:
     raise ValueError("HF_TOKEN environment variable is not set")
@@ -258,8 +224,6 @@ client = OpenAI(
 
 API_KEY = os.getenv("SERPHOUSE_API_KEY")
 
-
-# 국가별 언어 코드 매핑 (첫 번째 탭에서는 'United States'만 주로 사용)
 COUNTRY_LANGUAGES = {
     "United States": "en",
     "KOREA": "ko",
@@ -325,10 +289,10 @@ COUNTRY_LANGUAGES = {
     "Latvia": "lv",
     "Lithuania": "lt",
     "Slovenia": "sl",
-    "Luxembourg": "fr",
-    "Malta": "mt",
-    "Cyprus": "el",
-    "Iceland": "is"
+    "Luxembourg": "Luxembourg",
+    "Malta": "Malta",
+    "Cyprus": "Cyprus",
+    "Iceland": "Iceland"
 }
 
 COUNTRY_LOCATIONS = {
@@ -402,17 +366,15 @@ COUNTRY_LOCATIONS = {
     "Iceland": "Iceland"
 }
 
-
 @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"
@@ -438,11 +400,9 @@ def translate_query(query, country):
         print(f"번역 오류: {str(e)}")
         return query
 
-
 def is_english(text):
     return all(ord(char) < 128 for char in text.replace(' ', '').replace('-', '').replace('_', ''))
 
-
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
     
@@ -513,7 +473,6 @@ def search_serphouse(query, country, page=1, num_result=10):
             "translated_query": query
         }
 
-
 def format_results_from_raw(response_data):
     if "error" in response_data:
         return "Error: " + response_data["error"], []
@@ -525,8 +484,7 @@ def format_results_from_raw(response_data):
         news_results = results.get('results', {}).get('results', {}).get('news', [])
         if not news_results:
             return "검색 결과가 없습니다.", []
-
-        # 한국 도메인 및 한국 관련 키워드 필터링
+        
         korean_domains = [
             '.kr', 'korea', 'korean', 'yonhap', 'hankyung', 'chosun', 
             'donga', 'joins', 'hani', 'koreatimes', 'koreaherald'
@@ -542,7 +500,6 @@ def format_results_from_raw(response_data):
             title = result.get("title", "").lower()
             channel = result.get("channel", result.get("source", "")).lower()
             
-            # 한국 관련 컨텐츠 필터링
             is_korean_content = (
                 any(domain in url or domain in channel for domain in korean_domains) or
                 any(keyword in title.lower() for keyword in korean_keywords)
@@ -564,12 +521,10 @@ def format_results_from_raw(response_data):
     except Exception as e:
         return f"결과 처리 중 오류 발생: {str(e)}", []
 
-
 def serphouse_search(query, country):
     response_data = search_serphouse(query, country)
     return format_results_from_raw(response_data)
 
-
 css = """
 /* 전역 스타일 */
 footer {visibility: hidden;}
@@ -745,7 +700,6 @@ footer {visibility: hidden;}
 }
 """
 
-
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비스") as iface:
     init_db()
     
@@ -755,18 +709,15 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, title="NewsAI 서비
             gr.Markdown("## 한국 주요 기업 미국 뉴스 DB")
             gr.Markdown("각 기업의 미국 뉴스를 검색하여 DB에 저장하고 불러올 수 있습니다.")
             
-            # (수정) 상단에 '전체 분석 보고 요약' 버튼을 배치
             with gr.Row():
                 full_report_btn = gr.Button("전체 분석 보고 요약", variant="primary")
                 full_report_display = gr.Markdown()
             
-            # 버튼 클릭 시 full_summary_report() 실행
             full_report_btn.click(
                 fn=full_summary_report,
                 outputs=full_report_display
             )
             
-            # 이후 개별 기업 검색/출력 UI
             with gr.Column():
                 for i in range(0, len(KOREAN_COMPANIES), 2):
                     with gr.Row():