Update app.py

app.py

@@ -21,6 +21,29 @@ system_instruction = """
 # 누적 토큰 사용량을 추적하는 전역 변수
 total_tokens_used = 0
 
+
+def fetch_ticker_info(ticker):
+    stock = yf.Ticker(ticker)
+    try:
+        info = stock.info
+        # 선택적으로 출력할 정보를 정제합니다.
+        result = {
+            "종목명": info.get("longName"),
+            "시장 가격": info.get("regularMarketPrice"),
+            "전일 종가": info.get("previousClose"),
+            "시가": info.get("open"),
+            "고가": info.get("dayHigh"),
+            "저가": info.get("dayLow"),
+            "52주 최고": info.get("fiftyTwoWeekHigh"),
+            "52주 최저": info.get("fiftyTwoWeekLow"),
+            "시가총액": info.get("marketCap"),
+            "배당 수익률": info.get("dividendYield"),
+            "주식 수": info.get("sharesOutstanding")
+        }
+        return "\n".join([f"{key}: {value}" for key, value in result.items() if value is not None])
+    except ValueError:
+        return "유효하지 않은 티커입니다. 다시 입력해주세요."
+        
 def format_prompt(message, history):
     prompt = "<s>[SYSTEM] {} [/SYSTEM]".format(system_instruction)
     for user_prompt, bot_response in history:
@@ -50,38 +73,46 @@ 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}"
 
-mychatbot = gr.Chatbot(
-    avatar_images=["./user.png", "./botm.png"],
-    bubble_full_width=False,
-    show_label=False,
-    show_copy_button=True,
-    likeable=True,
-)
-
-
-examples = [
-    ["반드시 한글로 답변할것.", []],  # history 값을 빈 리스트로 제공    
-    ["분석 결과 보고서 다시 출력할것", []],
-    ["추천 종목 알려줘", []],
-    ["그 종목 투자 전망 예측해", []]
-]
-
-
-css = """
-h1 {
-    font-size: 14px; /* 제목 글꼴 크기를 작게 설정 */
-}
-footer {visibility: hidden;}
-"""
+# 기존의 채팅 인터페이스에 티커 입력 기능을 통합
+def setup_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("### 글로벌 자산(주식,지수,상품,가상자산,외환 등) 분석 LLM: BloombAI")
+        
+        with gr.Row():
+            ticker_input = gr.Textbox(label="티커 입력", placeholder="예: AAPL", elem_id="ticker_input")
+            submit_button = gr.Button("조회", elem_id="submit_button")
+        
+        chatbot = gr.Chatbot(
+            avatar_images=["./user.png", "./botm.png"],
+            bubble_full_width=False,
+            show_label=False,
+            show_copy_button=True,
+            likeable=True,
+            elem_id="chatbot_area"
+        )
+        
+        # 결과를 채팅 인터페이스에 출력하도록 설정
+        submit_button.click(
+            fn=lambda x: f"티커 '{x}'의 정보 조회 결과:\n\n{fetch_ticker_info(x)}",
+            inputs=ticker_input,
+            outputs=chatbot
+        )
+        
+        gr.Markdown("### 채팅")
+        examples = [
+            ["반드시 한글로 답변할것.", []],
+            ["분석 결과 보고서 다시 출력할것", []],
+            ["추천 종목 알려줘", []],
+            ["그 종목 투자 전망 예측해", []]
+        ]
+        chatbot.update(
+            fn=generate,
+            inputs=[gr.Textbox(label="메시지 입력", elem_id="message_input")],
+            outputs=chatbot,
+            examples=examples
+        )
 
-demo = gr.ChatInterface(
-    fn=generate,
-    chatbot=mychatbot,
-    title="글로벌 자산(주식,지수,상품,가상자산,외환 등) 분석 LLM: BloombAI",
-    retry_btn=None,
-    undo_btn=None,
-    css=css,
-    examples=examples    
-)
+    return demo
 
-demo.queue().launch(show_api=False)
+app = setup_interface()
+app.launch()