Update app.py

app.py

@@ -79,10 +79,10 @@ def get_real_news_summary(company, date):
     # OpenAI 클라이언트 초기화
     client = OpenAI(api_key=API_KEY, base_url="https://api.perplexity.ai")
     
-    # API 요청을 위한 메시지 구성
+    # API 요청을 위한 메시지 구성 - 한국어로만 응답을 받도록 지시
     messages = [
-        {"role": "system", "content": "You are a helpful assistant that summarizes stock news in Korean."},
-        {"role": "user", "content": f"Summarize the latest stock news for {company} on {date} in Korean. If there's no specific news for that date, provide the most recent relevant information."}
+        {"role": "system", "content": "You are a helpful assistant that summarizes stock news strictly in Korean."},
+        {"role": "user", "content": f"Summarize the latest stock news for {company} on {date} in Korean. If there's no specific news for that date, provide the most recent relevant information in Korean."}
     ]
     
     try:
@@ -98,7 +98,10 @@ def get_real_news_summary(company, date):
         # 한글, 숫자, 공백, 특수 기호만 남기는 정규 표현식
         korean_only_summary = re.sub(r'[^\w\s#.,!%()-]', '', summary)
 
-        return korean_only_summary
+        # ##로 시작하는 부분을 **으로 감싸서 Bold 처리
+        formatted_summary = re.sub(r'##\s*(.+)', r'**\1**', korean_only_summary)
+
+        return formatted_summary
     except Exception as e:
         return f"뉴스 요약 중 에러가 발생했습니다: {str(e)}"
 
@@ -106,19 +109,19 @@ def get_real_news_summary(company, date):
 def handle_click(input_value, date_clicked):
     return get_real_news_summary(input_value, date_clicked)
 
-# Gradio에서 사용할 함수 (뉴스 요약 포함)
-def update_news(input_value, selected_date):
-    if selected_date == "" or selected_date is None:
-        return "날짜를 선택해주세요."
-    else:
-        return handle_click(input_value, selected_date)
-
-# 종목명과 티커를 매핑하는 딕셔너리
+# 종목명과 티커를 매핑하는 딕셔너리 확장
 name_to_ticker = {
     "삼성전자": "005930.KS",
     "SK바이오팜": "326030.KS",
     "Apple": "AAPL",
-    # 필요한 종목들을 추가하세요
+    "Nvidia": "NVDA",
+    "Vertex": "VRTX",
+    "현대차": "005380.KS",
+    "카카오": "035720.KS",
+    "LG화학": "051910.KS",
+    "셀트리온": "068270.KS",
+    "네이버": "035420.KS",
+    # 추가로 필요한 종목들을 여기에 계속 추가하세요
 }
 
 # 주가 데이터를 가져오고 조건에 맞는 날짜와 그래프를 반환하는 함수
@@ -174,7 +177,7 @@ def display_stock_with_highlight(input_value, change_type, percent_change):
 
 # Gradio 인터페이스 생성 (3열 레이아웃)
 with gr.Blocks() as demo:
-    gr.Markdown("### **주가 그래프와 뉴스 요약**")
+    gr.Markdown("## 주가 그래프와 뉴스 요약")
 
     with gr.Row():
         with gr.Column():  # 입력값을 담을 첫 번째 열
@@ -198,7 +201,8 @@ with gr.Blocks() as demo:
             date_dropdown = gr.Dropdown(label="조건에 해당하는 날짜 선택", choices=[])
 
         with gr.Column():  # 뉴스 요약을 출력할 세 번째 열
-            news_output = gr.Textbox(label="뉴스 요약")
+            # news_output을 gr.Markdown으로 변경
+            news_output = gr.Markdown(label="뉴스 요약")
 
     # Submit 버튼 클릭 시 그래프 및 날짜 드롭다운 업데이트
     submit_btn.click(