Update app.py

app.py

@@ -76,24 +76,36 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
     except Exception as e:
         yield f"Error: {str(e)}\nTotal tokens used: {total_tokens_used}"
 
+def validate_ticker(ticker):
+    stock = yf.Ticker(ticker)
+    # Yahoo Finance에서 기본 정보를 가져와 확인합니다.
+    try:
+        info = stock.info
+        if not info:
+            raise ValueError("유효하지 않은 티커입니다.")
+    except Exception as e:
+        return False, str(e)
+    return True, "티커 유효성 검증 성공."
+
 def process_financial_data(ticker):
+    valid, message = validate_ticker(ticker)
+    if not valid:
+        return f"Error: {message} - 주식 티커 '{ticker}'를 확인하세요."
+
     try:
         stock = yf.Ticker(ticker)
         real_time_price = stock.history(period="1d")  # 실시간 주식 가격
-        stock_info = stock.info  # 주식 정보
-        financial_data = stock.history(period="max")  # 주식 데이터 다운로드 및 분석
         financials = stock.financials  # 재무제표 다운로드 및 분석
-        news_links = stock.news  # 최신 뉴스 조회
-
         return f"""**실시간 주식 가격**: {real_time_price.tail(1)}
-**기본 주식 정보**: {stock_info.get('longName', '정보 없음')} ({stock_info.get('sector', '정보 없음')})
-**재무 데이터 (최근 5일)**: {financial_data.tail(5)}
 **재무제표**: {financials.head()}
-**최신 뉴스 링크**: {', '.join([news['title'] + ": " + news['link'] for news in news_links[:3]])}
 """
     except Exception as e:
-        return f"Error: {str(e)} - 주식 티커를 확인하세요. 요청하신 주식 티커 '{ticker}'는 찾을 수 없거나 정보를 불러올 수 없습니다."
+        return f"Error: {str(e)} - 주식 데이터를 가져올 수 없습니다."
 
+# 예시 사용
+ticker = "AAPL"  # 사용자 입력 티커
+result = process_financial_data(ticker)
+print(result)
 
 
 mychatbot = gr.Chatbot(