Update app.py

app.py

@@ -73,27 +73,30 @@ 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}"
 
-# 기존의 채팅 인터페이스에 티커 입력 기능을 통합
+
 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")
+            ticker_input = gr.Textbox(label="티커 입력", placeholder="예: AAPL")
+            submit_button = gr.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"
+            likeable=True
         )
         
-        # 결과를 채팅 인터페이스에 출력하도록 설정
+        # 버튼 클릭 이벤트를 통해 티커 정보를 채팅 창에 직접 출력
+        def query_and_show(ticker):
+            info = fetch_ticker_info(ticker)
+            return [("", f"티커 '{ticker}'의 정보 조회 결과:\n\n{info}")]  # 첫번째 항목은 사용자 메시지, 두번째는 봇의 응답
+        
         submit_button.click(
-            fn=lambda x: f"티커 '{x}'의 정보 조회 결과:\n\n{fetch_ticker_info(x)}",
+            fn=query_and_show,
             inputs=ticker_input,
             outputs=chatbot
         )
@@ -105,12 +108,8 @@ def setup_interface():
             ["추천 종목 알려줘", []],
             ["그 종목 투자 전망 예측해", []]
         ]
-        chatbot.update(
-            fn=generate,
-            inputs=[gr.Textbox(label="메시지 입력", elem_id="message_input")],
-            outputs=chatbot,
-            examples=examples
-        )
+        
+        chatbot.add_message("환영합니다! 어떤 주식 정보가 궁금하신가요?")
 
     return demo