Update app.py

app.py

@@ -6,7 +6,6 @@ import time
 import logging
 from datetime import datetime
 
-
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -64,7 +63,6 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
     input_tokens = len(tokenizer.encode(prompt))
     total_tokens_used += input_tokens
     available_tokens = 32768 - total_tokens_used
-
     if available_tokens <= 0:
         yield f"Error: 입력이 최대 허용 토큰 수를 초과합니다. Total tokens used: {total_tokens_used}"
         return
@@ -78,11 +76,18 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
             output_part = response['generated_text'] if 'generated_text' in response else str(response)
             output_accumulated += output_part
             
-            # 데이터 다운로드 및 결과 출력
+            # 티커 추출 및 데이터 다운로드
             if "티커" in output_part:
-                ticker = extract_ticker(output_part)  # 티커 추출하는 더미 함수, 실제로는 티커를 추출하는 로직 필요
-                download_result = download_stock_data(ticker)
-                output_accumulated += download_result
+                ticker = extract_ticker(output_part)
+                if ticker:
+                    download_result = download_stock_data(ticker)
+                    output_accumulated += download_result
+                    
+                    # 티커 유효성 검사 및 재무 데이터 처리
+                    financial_data = process_financial_data(ticker)
+                    output_accumulated += financial_data
+                else:
+                    output_accumulated += "Error: 티커를 추출할 수 없습니다."
             
             yield output_accumulated + f"\n\n---\nTotal tokens used: {total_tokens_used}"
     except Exception as e:
@@ -99,12 +104,14 @@ def download_stock_data(ticker):
         else:
             return f"Success: {ticker} 데이터 다운로드 성공."
     except Exception as e:
+        logging.error(f"Error downloading data for {ticker}: {e}")
         return f"Error: 데이터 다운로드 중 오류가 발생했습니다. {str(e)}"
 
 def extract_ticker(output_part):
-    # 이 함수는 사용자 출력에서 티커를 추출하는 로직을 구현해야 함
-    # 여기서는 예시로 'AAPL'을 반환하도록 설정
-    return "AAPL"
+    # 사용자 출력에서 티커를 추출하는 로직 구현
+    # 예시: 'AAPL' 추출
+    ticker = "AAPL"  # 실제 티커 추출 로직으로 대체 필요
+    return ticker
 
 def validate_ticker(ticker):
     logging.debug(f"Validating ticker: {ticker}")
@@ -132,11 +139,6 @@ def process_financial_data(ticker):
         logging.error(f"Error processing financial data for {ticker}: {e}")
         return f"Error: {str(e)} - 주식 데이터를 가져올 수 없습니다."
 
-# 예시 사용
-ticker = "AAPL"  # 사용자 입력 티커
-result = process_financial_data(ticker)
-print(result)
-
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./botm.png"],
     bubble_full_width=False,
@@ -154,7 +156,7 @@ examples = [
 
 css = """
 h1 {
-    font-size: 14px; /* 제목 글꼴 크기를 작게 설정 */
+    font-size: 14px;
 }
 footer {
     visibility: hidden;