Update src/streamlit_app.py

src/streamlit_app.py

@@ -139,45 +139,36 @@ def generate_llm_report(investor_style, ticker):
     {news_text}
     """
     system_prompt = """You are an expert financial analyst. Your mission is to write a concise, objective investment report for a client based on their specific risk profile.
-
     ANALYSIS INSTRUCTIONS: 
     - Use BOTH the provided financial metrics ("Facts") and recent news headlines ("News"). 
     - Always include the metrics listed under `core_metrics` in Facts. These are the most important indicators for the company/sector. 
     - Each Key Highlight should integrate at least one financial fact and one news item together (not listed separately). 
     - Do not hallucinate numbers that are not in Facts. 
     - Adjust the focus and tone strictly based on the investor's style: 
-
     If the style is SAFE (Conservative):
     * Focus: Capital preservation and stable income. 
     * Highlight: Balance sheet strength, liquidity, predictable returns. 
     * Mention risks (regulatory, legal, earnings decline) first, then cautiously note positives. 
     * Downplay speculative or uncertain news.
-
     If the style is NEUTRAL (Moderate):
     * Focus: A balance between growth and safety. 
     * Highlight: Strategic trade-offs. Analyze how growth initiatives (from news) interact with financial stability (from facts). 
     * Present risks and opportunities in equal measure. 
-
     If the style is RISKY (Aggressive):
     * Focus: High growth potential and maximum returns. 
     * Highlight: Exciting, forward-looking growth story. Emphasize innovation, expansion, competitive advantages. 
     * Frame risks as natural volatility on the path to high rewards. 
     * Place financial facts in the context of supporting aggressive growth. 
-
     OUTPUT REPORT TEMPLATE
     Report for: A (investor_style) Investor
     Company: (company_name)
-
     1. Executive Summary:
     (Provide a brief, one-paragraph summary that aligns with the investor's style, integrating at least one key metric and one recent news item.)
-
     2. Key Analysis & Highlights:
     (5–7 bullet points. Each bullet must combine a financial metric with a relevant news event, written from the perspective of the given investor style.)
-
     3. Concluding Remark:
     (One or two sentences, neutrally summarizing the company’s current standing for this type of investor. 
     Do NOT provide direct financial advice or buy/sell recommendations.)
-
     IMPORTANT:
     - Keep the tone professional and concise. 
     - Reports must be grounded in Facts and News only. 
@@ -200,9 +191,6 @@ def generate_llm_report(investor_style, ticker):
 
     return result
 
-def reset_search_input():
-    if "stock_search_input" in st.session_state:
-        st.session_state.stock_search_input = None
 
 # 1. 세션 상태(Session State) 초기화
 # st.session_state : 스트림릿이 재실행되어도 값을 유지하는 변수
@@ -233,41 +221,40 @@ full_market_df = load_full_df() # (Srisk 모듈용 데이터)
 # 3. 입력 섹션 (종목 추가)
 st.subheader("1. 보유 종목 추가하기")
 
-# 컬럼을 나눠서 UI를 깔끔하게 구성
-col1, col2 = st.columns([2, 1])
-
-with col1:
-    # `selectbox`를 검색 가능한 입력창으로 사용
-    selected_display = st.selectbox(
-        "종목 검색 (NASDAQ100 or S&P500 티커 또는 기업명)",
-        options=TICKER_OPTIONS_LIST,
-        index=None,
-        placeholder="티커를 검색하거나 선택하세요 (예: AAPL 또는 Apple)",
-        key="stock_search_input"
-    )
-with col2:
-    quantity = st.number_input("보유 수량 (주)", min_value=0.01, step=0.1, format="%.2f")
-
-# '종목 추가' 버튼
-if st.button("➕ 포트폴리오에 추가", use_container_width=True, on_click=reset_search_input):
-    selected_ticker = None
-    if selected_display:
-        selected_ticker = DISPLAY_TO_TICKER_MAP.get(selected_display)
-        current_price = TICKER_TO_PRICE_MAP.get(selected_ticker)
-
-    if selected_ticker:
-        st.session_state.portfolio.append({
-            "ticker": selected_ticker,
-            "quantity": quantity,
-            "price": current_price,
-            "total_value": quantity * current_price
-        })
-        st.success(f"{selected_ticker} {quantity}주 (현재가 ${current_price:,.2f})를 포트폴리오에 추가했습니다.")
-
-        st.rerun()
+with st.form(key='portfolio_add_form', clear_on_submit=True):
+    col1, col2 = st.columns([2, 1])
+    
+    with col1:
+        # `selectbox`를 검색 가능한 입력창으로 사용
+        selected_display = st.selectbox(
+            "종목 검색 (NASDAQ100 or S&P500 티커 또는 기업명)",
+            options=TICKER_OPTIONS_LIST,
+            index=None,
+            placeholder="티커를 검색하거나 선택하세요 (예: AAPL 또는 Apple)"
+        )
+    with col2:
+        quantity = st.number_input("보유 수량 (주)", min_value=0.01, step=0.1, format="%.2f")
         
-    else:
-        st.warning("종목, 수량을 모두 올바르게 입력하세요.")
+    submitted = st.form_submit_button("➕ 포트폴리오에 추가", use_container_width=True, type="primary")
+    
+    # '종목 추가' 버튼
+    if submitted:
+        selected_ticker = None
+        if selected_display:
+            selected_ticker = DISPLAY_TO_TICKER_MAP.get(selected_display)
+            current_price = TICKER_TO_PRICE_MAP.get(selected_ticker)
+    
+        if selected_ticker:
+            st.session_state.portfolio.append({
+                "ticker": selected_ticker,
+                "quantity": quantity,
+                "price": current_price,
+                "total_value": quantity * current_price
+            })
+            st.success(f"{selected_ticker} {quantity}주 (현재가 ${current_price:,.2f})를 포트폴리오에 추가했습니다.")
+            st.rerun()
+        else:
+            st.warning("종목, 수량을 모두 올바르게 입력하세요.")
 
 # 4. 포트폴리오 요약 및 보고서 생성 (스케치 레이아웃)
 st.subheader("2. 포트폴리오 요약 및 보고서 생성")
@@ -373,7 +360,7 @@ with col_controls:
             )
         
         with col_regen:
-            if st.button(f"'{new_style}' 스타일로 재생성", use_container_width=True, key="my_regen_button"):
+            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,