seawolf2357 commited on
Commit
113d653
ยท
verified ยท
1 Parent(s): 9548efe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -12
app.py CHANGED
@@ -12,6 +12,7 @@ system_instruction = """
12
  ๋„ˆ์˜ ์ด๋ฆ„์€ 'BloombAI'์ด๋‹ค.
13
  ๋„ˆ์˜ ์—ญํ• ์€ '์ฃผ์‹ ๋ถ„์„ ์ „๋ฌธ๊ฐ€'์ด๋‹ค. ์˜ค๋Š˜์€ 2024๋…„ 04์›” 20์ผ์ด๋‹ค.
14
  '์ข…๋ชฉ' ์ด๋ฆ„์ด ์ž…๋ ฅ๋˜๋ฉด, yfinance์— ๋“ฑ๋ก๋œ 'ํ‹ฐ์ปค'๋ฅผ ์ถœ๋ ฅํ•˜๋ผ.
 
15
  ์˜ˆ๋ฅผ๋“ค์–ด, ์•„๋งˆ์กด 'AMZN' ์• ํ”Œ 'AAPL' ์‚ผ์„ฑ์ „์ž ๋“ฑ ํ•œ๊ตญ ๊ธฐ์—…์˜ ๊ฒฝ์šฐ KRX ๋“ฑ๋ก ํ‹ฐ์ปค์— .KS๊ฐ€ ํ‹ฐ์ปค๊ฐ€ ๋˜๊ณ 
16
  ์ด๊ฒƒ์„ yfinance๋ฅผ ํ†ตํ•ด ๊ฒ€์ฆํ•˜์—ฌ ์ถœ๋ ฅํ•˜๋ผ
17
  ์ด๋ฏธ์ง€์™€ ๊ทธ๋ž˜ํ”„๋Š” ์ง์ ‘ ์ถœ๋ ฅํ•˜์ง€ ๋ง๊ณ  '๋งํฌ'๋กœ ์ถœ๋ ฅํ•˜๋ผ
@@ -67,18 +68,6 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
67
  return f"์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}"
68
 
69
 
70
- def postprocess(history):
71
- user_prompt = history[-1][0]
72
- bot_response = history[-1][1]
73
-
74
- # ํ‹ฐ์ปค ์ •๋ณด ์ถœ๋ ฅ
75
- if "ํ‹ฐ์ปค" in bot_response:
76
- ticker = bot_response.split("ํ‹ฐ์ปค")[1].strip()
77
- ticker_info = fetch_ticker_info(ticker)
78
- bot_response += f"\n\nํ‹ฐ์ปค ์ •๋ณด:\n{ticker_info}"
79
-
80
- return [(user_prompt, bot_response)]
81
-
82
  mychatbot = gr.Chatbot(
83
  avatar_images=["./user.png", "./botm.png"],
84
  bubble_full_width=False,
 
12
  ๋„ˆ์˜ ์ด๋ฆ„์€ 'BloombAI'์ด๋‹ค.
13
  ๋„ˆ์˜ ์—ญํ• ์€ '์ฃผ์‹ ๋ถ„์„ ์ „๋ฌธ๊ฐ€'์ด๋‹ค. ์˜ค๋Š˜์€ 2024๋…„ 04์›” 20์ผ์ด๋‹ค.
14
  '์ข…๋ชฉ' ์ด๋ฆ„์ด ์ž…๋ ฅ๋˜๋ฉด, yfinance์— ๋“ฑ๋ก๋œ 'ํ‹ฐ์ปค'๋ฅผ ์ถœ๋ ฅํ•˜๋ผ.
15
+ ์‘๋‹ต๊ฐ’์— ํ‹ฐ์ปค ์ •๋ณด๊ฐ€ ํ˜ธ์ถœ๋˜๋ฉด ์ด ์ฝ”๋“œ๋‚ด์˜ ํ•จ์ˆ˜(fetch_ticker_info)๋ฅผ ํ˜ธ์ถœํ•˜์—ฌ ๊ฒฐ๊ณผ๋ฅผ ์ฐฝ์— ์ถœ๋ ฅํ•˜๋ผ
16
  ์˜ˆ๋ฅผ๋“ค์–ด, ์•„๋งˆ์กด 'AMZN' ์• ํ”Œ 'AAPL' ์‚ผ์„ฑ์ „์ž ๋“ฑ ํ•œ๊ตญ ๊ธฐ์—…์˜ ๊ฒฝ์šฐ KRX ๋“ฑ๋ก ํ‹ฐ์ปค์— .KS๊ฐ€ ํ‹ฐ์ปค๊ฐ€ ๋˜๊ณ 
17
  ์ด๊ฒƒ์„ yfinance๋ฅผ ํ†ตํ•ด ๊ฒ€์ฆํ•˜์—ฌ ์ถœ๋ ฅํ•˜๋ผ
18
  ์ด๋ฏธ์ง€์™€ ๊ทธ๋ž˜ํ”„๋Š” ์ง์ ‘ ์ถœ๋ ฅํ•˜์ง€ ๋ง๊ณ  '๋งํฌ'๋กœ ์ถœ๋ ฅํ•˜๋ผ
 
68
  return f"์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}"
69
 
70
 
 
 
 
 
 
 
 
 
 
 
 
 
71
  mychatbot = gr.Chatbot(
72
  avatar_images=["./user.png", "./botm.png"],
73
  bubble_full_width=False,