Update app.py

app.py

@@ -12,6 +12,7 @@ from bs4 import BeautifulSoup
 import re
 import pathlib
 import sqlite3
+import pytz
 
 # 한국 기업 리스트
 KOREAN_COMPANIES = [
@@ -33,7 +34,52 @@ KOREAN_COMPANIES = [
     "Celltrion" 
 ]
 
-# DB 관련 함수들
+def convert_to_seoul_time(timestamp_str):
+    try:
+        utc_time = datetime.strptime(timestamp_str, '%Y-%m-%d %H:%M:%S')
+        utc_time = pytz.utc.localize(utc_time)
+        seoul_tz = pytz.timezone('Asia/Seoul')
+        seoul_time = utc_time.astimezone(seoul_tz)
+        return seoul_time.strftime('%Y-%m-%d %H:%M:%S KST')
+    except:
+        return timestamp_str
+
+def analyze_sentiment_batch(articles, client):
+    try:
+        # 모든 기사의 제목과 내용을 하나의 텍스트로 결합
+        combined_text = "\n\n".join([
+            f"제목: {article.get('title', '')}\n내용: {article.get('snippet', '')}"
+            for article in articles
+        ])
+
+        prompt = f"""다음 뉴스 모음에 대해 전반적인 감성 분석을 수행하세요:
+
+뉴스 내용:
+{combined_text}
+
+다음 형식으로 분석해주세요:
+1. 전반적 감성: [긍정/부정/중립]
+2. 주요 긍정적 요소:
+   - [항목1]
+   - [항목2]
+3. 주요 부정적 요소:
+   - [항목1]
+   - [항목2]
+4. 종합 평가: [상세 설명]
+"""
+        
+        response = client.chat.completions.create(
+            model="CohereForAI/c4ai-command-r-plus-08-2024",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.3,
+            max_tokens=1000
+        )
+        
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"감성 분석 실패: {str(e)}"
+
+# DB 초기화 함수
 def init_db():
     db_path = pathlib.Path("search_results.db")
     conn = sqlite3.connect(db_path)
@@ -47,14 +93,24 @@ def init_db():
     conn.commit()
     conn.close()
 
+# 검색 결과 저장 함수
 def save_to_db(keyword, country, results):
     conn = sqlite3.connect("search_results.db")
     c = conn.cursor()
-    c.execute("INSERT INTO searches (keyword, country, results) VALUES (?, ?, ?)",
-              (keyword, country, json.dumps(results)))
+    
+    # 서울 시간으로 변환
+    seoul_tz = pytz.timezone('Asia/Seoul')
+    seoul_time = datetime.now(seoul_tz)
+    
+    c.execute("""INSERT INTO searches 
+                 (keyword, country, results, timestamp) 
+                 VALUES (?, ?, ?, ?)""",
+              (keyword, country, json.dumps(results), 
+               seoul_time.strftime('%Y-%m-%d %H:%M:%S')))
     conn.commit()
     conn.close()
 
+# DB에서 검색 결과 불러오기 함수
 def load_from_db(keyword, country):
     conn = sqlite3.connect("search_results.db")
     c = conn.cursor()
@@ -63,9 +119,10 @@ def load_from_db(keyword, country):
     result = c.fetchone()
     conn.close()
     if result:
-        return json.loads(result[0]), result[1]
+        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):
@@ -76,6 +133,7 @@ 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:
@@ -83,38 +141,54 @@ 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()
-    c.execute("""
-        SELECT keyword, COUNT(*) as count, 
-               MAX(timestamp) as last_search
-        FROM searches 
-        WHERE keyword IN ({})
-        GROUP BY keyword 
-        ORDER BY count DESC, last_search DESC
-    """.format(','.join(['?']*len(KOREAN_COMPANIES))), KOREAN_COMPANIES)
-    stats = c.fetchall()
-    conn.close()
     
-    output = "## 한국 기업 검색 통계\n\n"
-    for keyword, count, last_search in stats:
-        output += f"### {keyword}\n"
-        output += f"- 검색 횟수: {count}회\n"
-        output += f"- 마지막 검색: {last_search}\n\n"
+    output = "## 한국 기업 뉴스 분석 리포트\n\n"
+    
+    for company in KOREAN_COMPANIES:
+        c.execute("""
+            SELECT results, timestamp 
+            FROM searches 
+            WHERE keyword = ? 
+            ORDER BY timestamp DESC 
+            LIMIT 1
+        """, (company,))
+        
+        result = c.fetchone()
+        if result:
+            results_json, timestamp = result
+            articles = json.loads(results_json)
+            seoul_time = convert_to_seoul_time(timestamp)
+            
+            output += f"### {company}\n"
+            output += f"- 마지막 업데이트: {seoul_time}\n"
+            output += f"- 저장된 기사 수: {len(articles)}건\n\n"
+            
+            if articles:
+                # 전체 기사에 대한 감성 분석
+                sentiment_analysis = analyze_sentiment_batch(articles, client)
+                output += "#### 뉴스 감성 분석\n"
+                output += f"{sentiment_analysis}\n\n"
+            
+            output += "---\n\n"
+    
+    conn.close()
     return output
 
 
 
 
 
-
     
 ACCESS_TOKEN = os.getenv("HF_TOKEN")
 if not ACCESS_TOKEN: