Update app.py

app.py

@@ -72,10 +72,30 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
         for response in stream:
             output_part = response['generated_text'] if 'generated_text' in response else str(response)
             output_accumulated += output_part
-            yield output_accumulated + f"\n\n---\nTotal tokens used: {total_tokens_used}"
+            yield process_financial_data(output_part) + f"\n\n---\nTotal tokens used: {total_tokens_used}"
     except Exception as e:
         yield f"Error: {str(e)}\nTotal tokens used: {total_tokens_used}"
 
+def process_financial_data(ticker):
+    stock = yf.Ticker(ticker)
+    real_time_price = stock.history(period="1d")  # 실시간 주식 가격
+    stock_info = stock.info  # 주식 정보
+    financial_data = stock.history(period="max")  # 주식 데이터 다운로드 및 분석
+    financials = stock.financials  # 재무제표 다운로드 및 분석
+    sector_data = stock.sustainability  # 섹터 및 산업 데이터 조회
+    news_links = stock.news  # 최신 뉴스 조회
+
+    # 간단한 출력 형식 정리
+    return f"""**실시간 주식 가격**: {real_time_price.tail(1)}
+**기본 주식 정보**: {stock_info['longName']} ({stock_info['sector']})
+**재무 데이터 (최근 5일)**: {financial_data.tail(5)}
+**재무제표**: {financials.head()}
+**섹터 및 산업 데이터**: {sector_data}
+**최신 뉴스 링크**: {news_links}
+"""
+
+
+
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./botm.png"],
     bubble_full_width=False,