Update app.py

app.py

@@ -58,6 +58,7 @@ class MyClient(discord.Client):
         )
 
 async def handle_message(message):
+    global conversation_history  # 전역 변수 사용을 명시
     user_input = message.content.strip()
     user_mention = message.author.mention
 
@@ -72,11 +73,14 @@ async def handle_message(message):
             return f"{user_mention}, '{user_input}'에 대한 티커 정보를 찾을 수 없습니다. 정확한 종목명이나 티커를 입력해 주세요."
 
 async def handle_ticker(ticker, user_mention):
+    global conversation_history  # 전역 변수 사용을 명시
     company_info = get_company_info(ticker)
 
+    # 대화 내역이 너무 길 경우, 최대 길이를 유지하기 위해 앞부분을 잘라냄
     if len(conversation_history) > 20:
         conversation_history = conversation_history[-20:]
 
+    # 대화 내역에서 역할이 제대로 교차하도록 확인
     filtered_conversation = []
     last_role = None
     for message in conversation_history:
@@ -84,6 +88,7 @@ async def handle_ticker(ticker, user_mention):
             filtered_conversation.append(message)
             last_role = message['role']
 
+    # 시스템 메시지와 필터링된 대화 히스토리를 결합하여 모델에 보낼 메시지 구성
     system_message = f"사용자가 티커 '{ticker}'에 대해 알고 싶어합니다."
     system_prefix = """
     너는 '글로벌 주식 정보 전문가'입니다. 티커에 대해 Yahoo Finance를 통해 회사 정보, 시세 정보, 1일/1주/1개월/1년/10년간의 차트 흐름 요약, 최신 관련 뉴스 요약을 제공하고, 최종적으로 향후 1주일간 시세 예측을 제공합니다. 리스크와 태스크를 구분하여 설명하며, 모든 투자 책임은 고객에게 있음을 반드시 고지하십시오.
@@ -94,14 +99,17 @@ async def handle_ticker(ticker, user_mention):
     messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}] + filtered_conversation
     logging.debug(f'모델에 보낼 메시지: {messages}')
 
+    # 모델 호출
     response = await asyncio.get_event_loop().run_in_executor(None, lambda: hf_client.chat_completion(
         messages=messages, max_tokens=1000, temperature=0.7, top_p=0.85))
 
     full_response_text = ''.join(response)
     logging.debug(f'모델 전체 응답: {full_response_text}')
 
+    # 어시스턴트 응답을 대화 히스토리에 추가
     conversation_history.append({"role": "assistant", "content": full_response_text})
 
+    # 최종 응답: 회사 정보와 모델의 응답 결합
     final_response = f"{user_mention}, {full_response_text}\n\n**회사 정보:**\n{company_info}"
 
     return final_response
@@ -109,11 +117,12 @@ async def handle_ticker(ticker, user_mention):
 def get_ticker_from_name(name):
     try:
         ticker_search = Ticker(name)
-        search_results = ticker_search.symbol
-        if isinstance(search_results, dict) and 'symbol' in search_results:
-            return search_results['symbol']
-        if isinstance(search_results, str):
-            return search_results
+        summary = ticker_search.summary_profile
+        # 티커가 올바르게 파싱되었는지 확인
+        if summary and isinstance(summary, dict):
+            # 가장 매칭된 키를 티커로 사용
+            ticker = next(iter(summary.keys()))
+            return ticker
         return None
     except Exception as e:
         logging.error(f'Error retrieving ticker for {name}: {e}')