Update app.py

app.py

@@ -61,16 +61,8 @@ plt.legend(prop=font_prop)
 
 # B 코드
 
-# 이후 코드 실행
-import requests
-import gradio as gr
-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):
@@ -104,15 +96,18 @@ def get_real_news_summary(company, date):
         return f"뉴스 요약 중 에러가 발생했습니다: {str(e)}"
 
 # 뉴스 요약을 가져오는 함수
-def handle_click(input_value, date_clicked):
-    return get_real_news_summary(input_value, date_clicked)
+def handle_click(company_name, date_clicked):
+    return get_real_news_summary(company_name, 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)
+        # 종목명을 가져와서 Perplexity로 검색
+        ticker = name_to_ticker.get(input_value, input_value)  
+        company_name = input_value if ticker == input_value else list(name_to_ticker.keys())[list(name_to_ticker.values()).index(ticker)]
+        return handle_click(company_name, selected_date)
 
 # 종목명과 티커를 매핑하는 딕셔너리 확장 및 조건별 종목 매핑
 name_to_ticker = {
@@ -168,9 +163,10 @@ def display_stock_with_highlight(input_value, change_type, percent_change):
             plt.text(index, row['Close'], index.strftime('%Y-%m-%d'), fontsize=10, fontweight='bold', color=color, ha='right')
             plt.axvline(x=index, color=color, linestyle='--', linewidth=1)  # x축과의 연결선 점선으로 표시
 
-        plt.title(f'{input_value} 주가 추이 ({change_type} 표시)')
-        plt.xlabel('날짜')
-        plt.ylabel('종가')
+        # 종목명 + '주가 추이'로 제목 설정
+        plt.title(f'{input_value} 주가 추이', fontproperties=font_prop)
+        plt.xlabel('날짜', fontproperties=font_prop)
+        plt.ylabel('종가', fontproperties=font_prop)
         plt.legend()
 
         buf = io.BytesIO()