Update app.py

app.py

@@ -88,17 +88,22 @@ def generate(prompt, history=[], temperature=0.1, max_new_tokens=10000, top_p=0.
                 output_accumulated += output_part
                 yield output_accumulated + f"\n\n---\nTotal tokens used: {total_tokens_used}\nStock Data: {stock_data}"
         else:
-            yield "입력하신 종목명을 확인할 수 없습니다. 정확한 종목명을 입력해주세요."
+            yield f"입력하신 '{prompt}'은(는) 지원되는 종목명이 아닙니다. 현재 지원되는 종목은 애플, 마이크로소프트, 아마존, 알파벳 (구글), 테슬라 등입니다. 정확한 종목명을 입력해주세요."
     except Exception as e:
         yield f"Error: {str(e)}\nTotal tokens used: {total_tokens_used}"
 
 # 종목명을 토대로 티커와 기업 정보를 제공하는 함수
 def get_stock_info(name):
-    if name.lower() == "apple":
-        return {'ticker': 'AAPL', 'name': '애플', 'description': '아이폰을'}
-    # 추가적인 종목에 대한 정보를 이곳에 구현할 수 있습니다.
-    return {'ticker': None, 'name': name, 'description': ''}
-
+    stock_info = {
+        "apple": {'ticker': 'AAPL', 'name': '애플', 'description': '아이폰을'},
+        "microsoft": {'ticker': 'MSFT', 'name': '마이크로소프트', 'description': '윈도우 운영체제와 오피스 소프트웨어를'},
+        "amazon": {'ticker': 'AMZN', 'name': '아마존', 'description': '전자상거래 및 클라우드 서비스를'},
+        "google": {'ticker': 'GOOGL', 'name': '알파벳 (구글)', 'description': '검색 엔진 및 온라인 광고를'},
+        "tesla": {'ticker': 'TSLA', 'name': '테슬라', 'description': '전기자동차와 에너지 저장장치를'},
+        # 추가적인 종목에 대한 정보를 이곳에 구현할 수 있습니다.
+    }
+    return stock_info.get(name.lower(), {'ticker': None, 'name': name, 'description': ''})
+    
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./botm.png"],
     bubble_full_width=False,