Update streamlit_app.py

streamlit_app.py

@@ -1,78 +1,105 @@
 import streamlit as st
 import yfinance as yf
 import FinanceDataReader as fdr
+from huggingface_hub import InferenceClient
 
 # 1. 페이지 설정
 st.set_page_config(page_title="Pocket Quant", page_icon="📈", layout="centered")
+st.title("🇨🇳 Qwen 투자 자문가")
+st.caption("Alibaba Qwen2.5-72B 모델이 분석합니다. (Zero Cost)")
 
-st.title("🤖 AI 주식 자문가")
-st.caption("이제 한글로 검색해도 다 알아듣습니다!")
+# 2. 허깅페이스 토큰 & 클라이언트 설정
+# Secrets에 HF_TOKEN이 없으면 에러 방지
+if "HF_TOKEN" in st.secrets:
+    client = InferenceClient(api_key=st.secrets["HF_TOKEN"])
+else:
+    st.error("🚨 설정 오류: Settings > Secrets에 'HF_TOKEN'을 넣어주세요!")
+    st.stop()
 
-# 2. [핵심] 한국 주식 목록 가져오기 (캐싱: 한 번만 로딩해서 속도 빠름)
+# 3. 주식 데이터 로딩
 @st.cache_data
 def get_stock_list():
-    # KRX 전체 리스트 다운로드 (약 2~3초 소요)
-    df = fdr.StockListing('KRX')
-    return df
+    return fdr.StockListing('KRX')
 
-# 데이터 로딩 중 표시
-with st.spinner('한국 주식 "전화번호부"를 불러오는 중...'):
+with st.spinner('전화번호부 뒤지는 중...'):
     df_krx = get_stock_list()
 
-# 3. 똑똑한 검색 함수 (한글 -> 티커 변환)
+# 4. 종목 검색 함수
 def search_ticker(input_text):
-    # 3-1. 입력값이 영어/숫자면 그냥 티커로 간주 (미국 주식 등)
-    if input_text.isascii():
-        return input_text.upper()
+    if input_text.isascii(): return input_text.upper()
     
-    # 3-2. 한글이면 KRX 리스트에서 검색
-    else:
-        # 정확히 일치하는 종목 찾기
-        res = df_krx[df_krx['Name'] == input_text]
-        
-        # 없으면 포함하는 종목 찾기 (예: "삼성" -> 삼성전자, 삼성생명...)
-        if res.empty:
-            return None
-        
-        # 찾았으면 티커 코드 변환
-        code = res.iloc[0]['Code']
-        market = res.iloc[0]['Market']
-        
-        # 야후 파이낸스용 접미사 붙이기 (.KS 코스피, .KQ 코스닥)
-        if market == 'KOSPI':
-            return f"{code}.KS"
-        elif market == 'KOSDAQ':
-            return f"{code}.KQ"
-        else:
-            return code # 기타
+    res = df_krx[df_krx['Name'] == input_text]
+    if res.empty: return None
+    
+    code = res.iloc[0]['Code']
+    market = res.iloc[0]['Market']
+    return f"{code}.KS" if market == 'KOSPI' else f"{code}.KQ"
 
-# 4. 사용자 입력
-user_input = st.text_input("종목명 또는 코드를 입력하세요", "삼성전자")
+# 5. UI 및 실행
+user_input = st.text_input("종목명 (예: 삼성전자, 카카오, TSLA)", "삼성전자")
 
-# 5. 분석 시작
-if st.button("🔍 분석 실행", use_container_width=True):
-    
-    # 한글 이름을 티커로 변환!
+if st.button("🚀 Qwen에게 물어보기", use_container_width=True):
     ticker = search_ticker(user_input)
     
     if ticker:
-        st.success(f"'{user_input}'을(를) 찾았습니다! 👉 코드: {ticker}")
-        
-        try:
-            stock = yf.Ticker(ticker)
-            hist = stock.history(period="1mo")
-            
-            if not hist.empty:
-                curr = hist['Close'].iloc[-1]
-                prev = hist['Close'].iloc[-2]
-                pct = ((curr - prev) / prev) * 100
+        with st.spinner(f"📊 1단계: '{ticker}' 시세 조회 중..."):
+            try:
+                # 데이터 수집
+                stock = yf.Ticker(ticker)
+                hist = stock.history(period="1mo")
+                info = stock.info
                 
-                st.metric(label="현재 주가", value=f"{curr:,.0f}원", delta=f"{pct:.2f}%")
-                st.line_chart(hist['Close'])
-            else:
-                st.error("데이터가 없습니다. 상장 폐지되었거나 코드가 잘못되었습니다.")
-                
-        except Exception as e:
-            st.error(f"에러 발생: {e}")
+                if not hist.empty:
+                    # 기본 차트 표시
+                    curr = hist['Close'].iloc[-1]
+                    start = hist['Close'].iloc[0]
+                    change = ((curr - start) / start) * 100
+                    
+                    st.metric("현재 주가", f"{curr:,.0f}원", f"{change:.2f}% (1달)")
+                    st.line_chart(hist['Close'])
+                    
+                    # --- Qwen AI 파트 ---
+                    with st.spinner("🐉 2단계: 큐웬(Qwen)이 차트를 해석하는 중..."):
+                        
+                        # 프롬프트 작성
+                        prompt = f"""
+                        당신은 냉철한 주식 투자 전문가입니다. 아래 데이터를 분석해서 한국어로 보고서를 써주세요.
+                        
+                        [종목 데이터]
+                        - 종목명: {user_input} ({ticker})
+                        - 현재가: {curr:,.0f}
+                        - 1달 변동률: {change:.2f}%
+                        - 52주 최고가: {info.get('fiftyTwoWeekHigh', 'N/A')}
+                        - 52주 최저가: {info.get('fiftyTwoWeekLow', 'N/A')}
+                        
+                        [분석 요청]
+                        1. 최근 1달간의 추세를 한 문장으로 요약하세요.
+                        2. 현재 가격이 고점 대비 어떤 위치인지 설명하세요.
+                        3. 투자 관점(매수/매도/관망)을 솔직하게 제시하고 이유를 덧붙이세요.
+                        
+                        말투는 '해요체'를 쓰고, 이모지를 적절히 사용하세요.
+                        """
+
+                        # Qwen 호출 (채팅 메시지 형식)
+                        messages = [
+                            {"role": "system", "content": "You are a helpful financial analyst."},
+                            {"role": "user", "content": prompt}
+                        ]
+                        
+                        # 스트리밍 방식으로 출력 (타자 치는 것처럼 나옴)
+                        stream = client.chat.completions.create(
+                            model="Qwen/Qwen2.5-72B-Instruct", 
+                            messages=messages, 
+                            max_tokens=500,
+                            stream=True
+                        )
+                        
+                        st.subheader("🐉 Qwen의 분석 리포트")
+                        st.write_stream(stream)
+                        
+                else:
+                    st.error("데이터가 없어요. 상장폐지 됐거나 티커가 틀렸을 수도?")
+            except Exception as e:
+                st.error(f"에러가 났습니다: {e}")
     else:
-        st.error(f"❌ '{user_input}'라는 종목을 한국 시장에서 못 찾겠어요. 정확한 이름을 입력해주세요.")
+        st.error(f"'{user_input}'? 그런 주식은 한국에 없는데요.")