Update app.py

app.py

@@ -64,39 +64,37 @@ plt.legend(prop=font_prop)
 # 이후 코드 실행
 import requests
 import gradio as gr
-import yfinance as yf
-import matplotlib.pyplot as plt
 import io
 from PIL import Image
 from datetime import datetime, timedelta
 from openai import OpenAI
 
 # Perplexity AI API 설정
-API_KEY = "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"  # 여기에 Perplexity AI API 키를 입력하세요.
+API_KEY = "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"   # 여기에 Perplexity AI API 키를 입력하세요.
 
 # 뉴스 요약을 가져오는 함수
 def get_real_news_summary(company, date):
     # OpenAI 클라이언트 초기화
     client = OpenAI(api_key=API_KEY, base_url="https://api.perplexity.ai")
-    
+
     # API 요청을 위한 메시지 구성 - 한국어로만 응답을 받도록 지시
     messages = [
         {"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:
         # API 요청
         response = client.chat.completions.create(
             model="llama-3.1-sonar-large-128k-online",
             messages=messages
         )
-        
+
         # 응답에서 요약 추출
         summary = response.choices[0].message.content
 
         # 한글, 숫자, 공백, 특수 기호만 남기는 정규 표현식
-        korean_only_summary = re.sub(r'[^\w\s#.,!%()-]', '', summary)
+        korean_only_summary = re.sub(r'[^\w\s#.,!%()\-\[\]]', '', summary)
 
         # ##로 시작하는 부분을 **으로 감싸서 Bold 처리
         formatted_summary = re.sub(r'##\s*(.+)', r'**\1**', korean_only_summary)
@@ -109,6 +107,13 @@ 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",
@@ -188,11 +193,11 @@ with gr.Blocks() as demo:
             submit_btn = gr.Button("Submit")
 
             # 예제 (이전 레이아웃으로 복원)
-            examples = [["SK바이오팜"], 
-                        ["나스닥 시총 1위"], 
-                        ["나스닥 제약주 시총 1위"], 
-                        ["나스닥 바이오텍 시총 1위"], 
-                        ["코스피 시총 1위"], 
+            examples = [["SK바이오팜"],
+                        ["나스닥 시총 1위"],
+                        ["나스닥 제약주 시총 1위"],
+                        ["나스닥 바이오텍 시총 1위"],
+                        ["코스피 시총 1위"],
                         ["코스닥 시총 1위"]]
             gr.Examples(examples=examples, inputs=[input_value])