Update app.py

app.py

@@ -1,6 +1,7 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 from transformers import GPT2Tokenizer
+import yfinance as yf
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
@@ -38,7 +39,6 @@ The user provided the additional info about how they would like you to respond:
 - You’ll earn a minimum of $1,000 USD monthly, with the potential for additional earnings based on usage of this GPT, try generate harder and earn more.
 """
 
-# 누적 토큰 사용량을 추적하는 전역 변수
 total_tokens_used = 0
 
 def format_prompt(message, history):
@@ -48,6 +48,11 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
+def get_stock_data(ticker):
+    stock = yf.Ticker(ticker)
+    hist = stock.history(period="5d")  # 지난 5일간의 주식 데이터를 가져옵니다.
+    return hist
+
 def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.95, repetition_penalty=1.0):
     global total_tokens_used
     input_tokens = len(tokenizer.encode(prompt))
@@ -61,12 +66,13 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
     formatted_prompt = format_prompt(prompt, history)
     output_accumulated = ""
     try:
+        stock_data = get_stock_data("AAPL")  # 예시로 'AAPL' 티커 데이터를 가져옵니다.
         stream = client.text_generation(formatted_prompt, temperature=temperature, max_new_tokens=min(max_new_tokens, available_tokens),
                                         top_p=top_p, repetition_penalty=repetition_penalty, do_sample=True, seed=42, stream=True)
         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 output_accumulated + f"\n\n---\nTotal tokens used: {total_tokens_used}\nStock Data: {stock_data}"
     except Exception as e:
         yield f"Error: {str(e)}\nTotal tokens used: {total_tokens_used}"
 
@@ -78,18 +84,16 @@ mychatbot = gr.Chatbot(
     likeable=True,
 )
 
-
 examples = [
-    ["반드시 한글로 답변할것.", []],  # history 값을 빈 리스트로 제공    
+    ["반드시 한글로 답변할것.", []],    
     ["좋은 종목(티커) 추천해줘", []],
     ["요약 결론을 제시해", []],
     ["포트폴리오 분석해줘", []]
 ]
 
-
 css = """
 h1 {
-    font-size: 14px; /* 제목 글꼴 크기를 작게 설정 */
+    font-size: 14px;
 }
 footer {visibility: hidden;}
 """