Update app.py

app.py

@@ -88,10 +88,46 @@ 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 f"입력하신 '{prompt}'은(는) 지원되는 종목명이 아닙니다. 현재 지원되는 종목은 애플, 마이크로소프트, 아마존, 알파벳 (구글), 테슬라 등입니다. 정확한 종목명을 입력해주세요."
+            # 입력이 티커인 경우 처리
+            ticker = prompt.upper()
+            if ticker in ['AAPL', 'MSFT', 'AMZN', 'GOOGL', 'TSLA']:
+                stock_info = get_stock_info_by_ticker(ticker)
+                response_msg = f"{stock_info['name']}은(는) {stock_info['description']} 주력으로 생산하는 기업입니다. {stock_info['name']}의 티커는 {stock_info['ticker']}입니다. 원하시는 종목이 맞는가요?"
+                output_accumulated += response_msg
+                yield output_accumulated
+
+                # 추가적인 분석 요청이 있다면, yfinance로 데이터 수집 및 분석
+                stock_data = get_stock_data(stock_info['ticker'])  # 티커를 이용해 주식 데이터를 가져옵니다.
+                stream = client.text_generation(
+                    formatted_prompt,
+                    temperature=temperature,
+                    max_new_tokens=min(max_new_tokens, available_tokens),
+                    top_p=top_p,
+                    repetition_penalty=repetition_penalty,
+                    do_sample=True,
+                    seed=42,
+                    stream=True
+                )
+                for response in stream:
+                    output_part = response['generated_text'] if 'generated_text' in response else str(response)
+                    output_accumulated += output_part
+                    yield output_accumulated + f"\n\n---\nTotal tokens used: {total_tokens_used}\nStock Data: {stock_data}"
+            else:
+                yield f"입력하신 '{prompt}'은(는) 지원되는 종목명 또는 티커가 아닙니다. 현재 지원되는 종목은 애플(AAPL), 마이크로소프트(MSFT), 아마존(AMZN), 알파벳(GOOGL), 테슬라(TSLA) 등입니다. 정확한 종목명 또는 티커를 입력해주세요."
     except Exception as e:
         yield f"Error: {str(e)}\nTotal tokens used: {total_tokens_used}"
 
+# 티커를 토대로 종목 정보를 제공하는 함수
+def get_stock_info_by_ticker(ticker):
+    stock_info = {
+        "AAPL": {'ticker': 'AAPL', 'name': '애플', 'description': '아이폰을'},
+        "MSFT": {'ticker': 'MSFT', 'name': '마이크로소프트', 'description': '윈도우 운영체제와 오피스 소프트웨어를'},
+        "AMZN": {'ticker': 'AMZN', 'name': '아마존', 'description': '전자상거래 및 클라우드 서비스를'},
+        "GOOGL": {'ticker': 'GOOGL', 'name': '알파벳', 'description': '검색 엔진 및 온라인 광고를'},
+        "TSLA": {'ticker': 'TSLA', 'name': '테슬라', 'description': '전기자동차와 에너지 저장장치를'},
+    }
+    return stock_info.get(ticker, {'ticker': None, 'name': None, 'description': ''})
+
 # 종목명을 토대로 티커와 기업 정보를 제공하는 함수
 def get_stock_info(name):
     stock_info = {