Update src/streamlit_app.py

사용자 투자 성향을 파이차트 위에도 적어주기(Srisk랑 같이)
재생성 버튼 두 개 되는 오류 수정

src/streamlit_app.py

@@ -1,4 +1,5 @@
 import os  
+
 os.environ['STREAMLIT_HOME'] = '/tmp/streamlit'  
 
 os.environ['HF_HOME'] = '/tmp/hf_cache'  
@@ -89,10 +90,17 @@ def generate_llm_reports(portfolio_list, override_style=None):
     if override_style:
         investor_style = override_style
         st.toast(f"'{investor_style}' 스타일(수동)로 재생성 시작...")
+        
+        st.session_state.investor_style = investor_style
+        st.session_state.srisk = None # 수동 선택 시 Srisk는 N/A 처리
+ 
     else:
         srisk, investor_style = classify_investment_style(full_market_df, portfolio_list)
         st.toast(f"srisk: {srisk:.2f} '{investor_style}' 스타일(자동)로 생성 시작...")
-
+        
+        st.session_state.investor_style = investor_style
+        st.session_state.srisk = srisk
+        
     reports = {}
 
     for item in portfolio_list:
@@ -193,23 +201,24 @@ def generate_llm_report(investor_style, ticker):
     return result
 
 
-# ----------------------------------------------------------------------
 # 1. 세션 상태(Session State) 초기화
-# ----------------------------------------------------------------------
-# st.session_state : 스트림릿이 재실행되어도 값을 유지하는 마법의 변수
+# st.session_state : 스트림릿이 재실행되어도 값을 유지하는 변수
 if 'portfolio' not in st.session_state:
     st.session_state.portfolio = []  # 사용자의 포트폴리오를 저장할 리스트
 if 'last_report' not in st.session_state:
     st.session_state.last_report = None # 생성된 보고서를 저장할 변수
 
+if 'srisk' not in st.session_state:
+    st.session_state.srisk = None
+if 'investor_style' not in st.session_state:
+    st.session_state.investor_style = None
+    
 if 'peft_model' not in st.session_state:
     st.session_state.peft_model = None
 if 'tokenizer' not in st.session_state:
     st.session_state.tokenizer = None
 
-# ----------------------------------------------------------------------
 # 2. 페이지 기본 설정
-# ----------------------------------------------------------------------
 st.set_page_config(page_title="AI 주식 포트폴리오 분석", layout="wide")
 st.title("🤖 AI 주식 포트폴리오 보고서 생성기")
 st.write("NASDAQ 100 종목을 검색하여 포트폴리오를 구성하고, 맞춤형 AI 보고서를 받아보세요.")
@@ -218,9 +227,7 @@ TICKER_OPTIONS_LIST, DISPLAY_TO_TICKER_MAP, TICKER_TO_PRICE_MAP = load_ticker_da
 company = load_company_metrics()
 full_market_df = load_full_df() # (Srisk 모듈용 데이터)
 
-# ----------------------------------------------------------------------
 # 3. 입력 섹션 (종목 추가)
-# ----------------------------------------------------------------------
 st.subheader("1. 보유 종목 추가하기")
 
 # 컬럼을 나눠서 UI를 깔끔하게 구성
@@ -255,15 +262,34 @@ if st.button("➕ 포트폴리오에 추가", use_container_width=True):
     else:
         st.warning("종목, 수량을 모두 올바르게 입력하세요.")
 
-# ----------------------------------------------------------------------
 # 4. 포트폴리오 요약 및 보고서 생성 (스케치 레이아웃)
-# ----------------------------------------------------------------------
 st.subheader("2. 포트폴리오 요약 및 보고서 생성")
-
 col_chart, col_controls = st.columns(2, gap="large")
 
 with col_chart:
     st.markdown("### 📊 포트폴리오 구성")
+
+    if st.session_state.get("investor_style"):
+        style = st.session_state.investor_style
+        srisk = st.session_state.srisk
+
+        # (수정) 스타일에 따라 이모지(색상 동그라미) 추가
+        if style == "SAFE":
+            style_display = f"🔵 {style}"  # 파란색
+        elif style == "NEUTRAL":
+            style_display = f"🟢 {style}"  # 연두색
+        elif style == "RISKY":
+            style_display = f"🔴 {style}"  # 빨간색
+        else:
+            style_display = style # 기본값
+            
+        if srisk is not None:
+            # 자동 분석 (Srisk가 있음)
+            st.metric(label="AI 분석 투자 성향", value=style, delta=f"S-Risk: {srisk:.2f}")
+        else:
+            # 수동 선택 (Srisk가 None)
+            st.metric(label="선택된 투자 성향", value=style, delta="수동 선택")
+            
     if st.session_state.portfolio:
         df = pd.DataFrame(st.session_state.portfolio)
 
@@ -299,7 +325,6 @@ with col_controls:
         )
 
         if not df.equals(edited_df):
-            # 삭제되거나 수정된 DataFrame을 다시 세션 상태(list of dicts)로 변환
             st.session_state.portfolio = edited_df.to_dict('records')
             st.toast("포트폴리오가 수정(삭제)되었습니다.")
             st.rerun()
@@ -307,12 +332,14 @@ with col_controls:
     if st.button("🔄 포트폴리오 전체 초기화", use_container_width=True, type="secondary"):
         st.session_state.portfolio = []
         st.session_state.last_report = None
+
+        st.session_state.srisk = None
+        st.session_state.investor_style = None
+        
         st.toast("포트폴리오가 초기화되었습니다.")
         st.rerun()
 
-# ----------------------------------------------------------------------
 # 5. 보고서 생성 버튼 (메인 LLM 호출)
-# ----------------------------------------------------------------------
     if st.button("🚀 AI 보고서 생성하기", type="primary", use_container_width=True, disabled=(not st.session_state.portfolio)):
         if st.session_state.peft_model and st.session_state.tokenizer:
             with st.spinner("AI가 포트폴리오를 분석하고 보고서를 작성 중입니다..."):
@@ -324,9 +351,7 @@ with col_controls:
             st.warning("모델이 아직 로드 중입니다. 잠시 후 다시 시도해주세요.")
     st.divider()
 
-# ----------------------------------------------------------------------
 # 6. 보고서 재생성 버튼 (메인 LLM 호출)
-# ----------------------------------------------------------------------
     if st.session_state.last_report:
         st.markdown("##### 🔄 다른 성향으로 보고서 다시 뽑기")
         
@@ -341,7 +366,7 @@ with col_controls:
             )
         
         with col_regen:
-            if st.button(f"'{new_style}' 스타일로 재생성", use_container_width=True):
+            if st.button(f"'{new_style}' 스타일로 재생성", use_container_width=True, "my_regen_button"):
                 with st.spinner(f"'{new_style}' 스타일로 보고서를 다시 작성 중입니다..."):
                     regenerated_reports = generate_llm_reports(
                         st.session_state.portfolio,
@@ -350,9 +375,7 @@ with col_controls:
                     st.session_state.last_report = regenerated_reports
                     st.rerun() # 화면을 즉시 새로고침
 
-# ----------------------------------------------------------------------
 # 7. 보고서 출력 섹션
-# ----------------------------------------------------------------------
 st.divider()
 
 if st.session_state.last_report:
@@ -368,10 +391,7 @@ if st.session_state.last_report:
 else:
     st.info("보고서를 생성하면 이 곳에 결과가 표시됩니다.")
 
-# ----------------------------------------------------------------------
 # 8. (신규) LLM 모델 로딩 (모든 UI를 그린 후 마지막에 실행)
-# ----------------------------------------------------------------------
-
 # peft_model, tokenizer를 st.session_state로 관리
 if 'peft_model' not in st.session_state:
     st.session_state.peft_model = None