Update app.py

app.py

@@ -415,6 +415,15 @@ if 'sell_confirm' not in st.session_state:
     st.session_state['sell_confirm'] = False
 if 'difficulty_level' not in st.session_state:
     st.session_state['difficulty_level'] = "초등학생" # 기본 난이도 설정
+if 'selected_stock_info' not in st.session_state:
+    st.session_state['selected_stock_info'] = None
+if 'selected_stock_sector_info' not in st.session_state:
+    st.session_state['selected_stock_sector_info'] = None
+if 'selected_stock_chart' not in st.session_state:
+    st.session_state['selected_stock_chart'] = None
+if 'selected_stock_sector_chart' not in st.session_state:
+    st.session_state['selected_stock_sector_chart'] = None
+
 
 # --- 뉴스 생성 함수 ---
 def generate_news():
@@ -424,21 +433,27 @@ def generate_news():
     difficulty_prompt_map = {
         "초등학생": {
             "level_desc": "초등학생 5~6학년 수준",
-            "sentence_count": "10~12", # 초등학생 수준 문장 수 조정
-            "vocabulary_level": "쉬운 어휘와 짧은 문장", # 초등학생 수준 어휘 강조
-            "inference_level": "단순하고 명확한 정보", # 초등학생 수준 추론 정보 강조
+            "sentence_count": "8~10",  # 더 짧은 문장
+            "vocabulary_level": "아주 쉬운 어휘와 짧고 명확한 문장 사용", # 더 쉬운 어휘 강조
+            "inference_level": "매우 단순하고 직관적인 정보", # 더 단순한 추론
+            "news_prompt_instructions": "단순한 경제 상황, 쉬운 단어, 짧은 문장, 명확한 정보 위주로 작성. 복잡한 경제 용어, 전문 용어, 추상적인 개념 사용 금지.", # 초등학생용 뉴스 생성 지침 추가
+            "explanation_prompt_instructions": "핵심 용어, 쉬운 비유 사용, 짧고 명확한 설명, 3문장 이내 요약", # 초등학생용 해설 지침 추가
         },
         "중학생": {
             "level_desc": "중학생 1~3학년 수준",
-            "sentence_count": "12~15", # 중학생 수준 문장 수
-            "vocabulary_level": "일상적인 어휘와 약간의 전문 용어", # 중학생 수준 어휘
-            "inference_level": "일반적인 경제 흐름과 관련된 추론", # 중학생 수준 추론
+            "sentence_count": "12~15",
+            "vocabulary_level": "일상적인 어휘와 기본적인 경제 용어 포함", # 중학생 수준 어휘
+            "inference_level": "일반적인 경제 흐름과 관련된 추론",
+            "news_prompt_instructions": "일상적인 경제 뉴스, 쉬운 경제 용어, 적절한 문장 길이, 일반적인 경제 흐름과 관련, 약간의 추론 필요", # 중학생용 뉴스 생성 지침
+            "explanation_prompt_instructions": "기본 경제 용어 설명,  경제 개념 쉬운 풀이, 3~4문장 요약", # 중학생용 해설 지침
         },
         "고등학생": {
             "level_desc": "고등학생 1~3학년 수준",
             "sentence_count": "15~20", # 고등학생 수준 문장 수 증가
-            "vocabulary_level": "다양한 어휘와 경제 전문 용어 포함", # 고등학생 수준 어휘
-            "inference_level": "심층적인 경제 분석 및 다각적인 추론", # 고등학생 수준 심층 추론
+            "vocabulary_level": "다양한 어휘와 경제 전문 용어 적극 사용", # 고등학생 수준 어휘
+            "inference_level": "심층적인 경제 분석 및 다각적인 추론 요구", # 고등학생 수준 심층 추론
+            "news_prompt_instructions": "심층 경제 뉴스, 경제 전문 용어 적극 사용, 긴 문장, 복잡한 구문,  산업 동향, 경제 정책, 국제 경제,  다각적인 분석과 심층 추론 필요", # 고등학생용 뉴스 생성 지침
+            "explanation_prompt_instructions": "전문 경제 용어 해설 포함,  심층적인 경제 분석,  5문장 내외 요약, 다양한 관점 제시", # 고등학생용 해설 지침
         },
     }
 
@@ -448,6 +463,7 @@ def generate_news():
     sentence_count = level_config["sentence_count"]
     vocabulary_level = level_config["vocabulary_level"]
     inference_level = level_config["inference_level"]
+    news_prompt_instructions = level_config["news_prompt_instructions"] # 난이도별 뉴스 생성 지침 추가
 
     prompt = f"""
 지시:
@@ -457,6 +473,7 @@ def generate_news():
 특정 회사 이름이나 주식 종목을 직접적으로 언급하지 마세요.
 긍정적 뉴스, 부정적 뉴스, 중립적 뉴스 다양하게 생성하세요.(긍정, 부정, 중립 이라는 말은 표시하지 마세요.)
 뉴스에 따라 주식이 상승하기도 하고 하락하기도 할 수 있습니다.
+{news_prompt_instructions} # 난이도별 뉴스 생성 지침 추가
 각 뉴스 기사는 "## 뉴스 [번호]" 로 시작해주세요. (예: ## 뉴스 1, ## 뉴스 2 ...)
 
 **생성된 뉴스 기사:**
@@ -493,11 +510,35 @@ def explain_daily_news_meanings(daily_news):
 
     difficulty_level = st.session_state['difficulty_level']
     difficulty_prompt_map = {
-        "초등학생": "초등학생 5~6학년",
-        "중학생": "중학생 1~3학년",
-        "고등학생": "고등학생 1~3학년",
+        "초등학생": {
+            "level_desc": "초등학생 5~6학년",
+            "sentence_count": "8~10",  # 더 짧은 문장
+            "vocabulary_level": "아주 쉬운 어휘와 짧고 명확한 문장 사용", # 더 쉬운 어휘 강조
+            "inference_level": "매우 단순하고 직관적인 정보", # 더 단순한 추론
+            "news_prompt_instructions": "단순한 경제 상황, 쉬운 단어, 짧은 문장, 명확한 정보 위주로 작성. 복잡한 경제 용어, 전문 용어, 추상적인 개념 사용 금지.", # 초등학생용 뉴스 생성 지침 추가
+            "explanation_prompt_instructions": "핵심 용어, 쉬운 비유 사용, 짧고 명확한 설명, 3문장 이내 요약", # 초등학생용 해설 지침 추가
+        },
+        "중학생": {
+            "level_desc": "중학생 1~3학년 수준",
+            "sentence_count": "12~15",
+            "vocabulary_level": "일상적인 어휘와 기본적인 경제 용어 포함", # 중학생 수준 어휘
+            "inference_level": "일반적인 경제 흐름과 관련된 추론",
+            "news_prompt_instructions": "일상적인 경제 뉴스, 쉬운 경제 용어, 적절한 문장 길이, 일반적인 경제 흐름과 관련, 약간의 추론 필요", # 중학생용 뉴스 생성 지침
+            "explanation_prompt_instructions": "기본 경제 용어 설명,  경제 개념 쉬운 풀이, 3~4문장 요약", # 중학생용 해설 지침
+        },
+        "고등학생": {
+            "level_desc": "고등학생 1~3학년 수준",
+            "sentence_count": "15~20", # 고등학생 수준 문장 수 증가
+            "vocabulary_level": "다양한 어휘와 경제 전문 용어 적극 사용", # 고등학생 수준 어휘
+            "inference_level": "심층적인 경제 분석 및 다각적인 추론 요구", # 고등학생 수준 심층 추론
+            "news_prompt_instructions": "심층 경제 뉴스, 경제 전문 용어 적극 사용, 긴 문장, 복잡한 구문,  산업 동향, 경제 정책, 국제 경제,  다각적인 분석과 심층 추론 필요", # 고등학생용 뉴스 생성 지침
+            "explanation_prompt_instructions": "전문 경제 용어 해설 포함,  심층적인 경제 분석,  5문장 내외 요약, 다양한 관점 제시", # 고등학생용 해설 지침
+        },
     }
-    level_prompt = difficulty_prompt_map.get(difficulty_level, "초등학생 5~6학년") # 기본값 초등학생
+    level_config = difficulty_prompt_map.get(difficulty_level, difficulty_prompt_map["초등학생"]) # 기본값 초등학생
+    level_prompt = level_config["level_desc"]
+    explanation_prompt_instructions = level_config["explanation_prompt_instructions"] # 난이도별 해설 지침 추가
+
 
     meanings = {}
     for i, news_article in enumerate(daily_news):
@@ -506,7 +547,8 @@ def explain_daily_news_meanings(daily_news):
     {news_article}
 
     **지시:**
-    위 신문 기사의 핵심 의미를 {level_prompt}이 이해하기 쉽게 3문장 이내로 요약해서 "해설: " 다음에 설명해주세요.
+    위 신문 기사의 핵심 의미를 {level_prompt} 수준의 학생이 이해하기 쉽게 요약하고 해설해주세요.
+    {explanation_prompt_instructions} # 난이도별 해설 지침 추가
     그리고 이 뉴스와 관련된 주식 섹터 1~2개를 쉼표로 구분해서 "관련 섹터: " 다음에 알려주세요. 관련 섹터가 없다면 "관련 섹터: 없음" 이라고 해주세요.
 
     뉴스 의미 해설:
@@ -768,67 +810,89 @@ def display_stock_prices():
         for stock_name, stock_info in sector_stocks.items():
             price_history = stock_info["price_history"]
             daily_change_rate_str = " - " # 기본값
+            daily_change_color = "gray" # 기본 색상 회색
             if len(price_history) >= 2:
                 previous_day_price = price_history[-2]
                 current_price = price_history[-1]
                 daily_change_rate = (current_price - previous_day_price) / previous_day_price * 100
                 daily_change_rate_str = f"{daily_change_rate:.2f}%"
+                if daily_change_rate > 0:
+                    daily_change_color = "red" # 상승 시 빨간색
+                elif daily_change_rate < 0:
+                    daily_change_color = "blue" # 하락 시 파란색
+                else:
+                    daily_change_color = "gray" # 보합 시 회색
 
             stocks_data.append(
                 {
-                    "종목": stock_name,
                     "섹터": sector,
+                    "종목": stock_name,
                     "현재 주가": f"{stock_info['current_price']:,} 원",
-                    "전일 대비": daily_change_rate_str, # 전일 대비 등락률 추가
+                    "전일 대비": f'<span style="color:{daily_change_color};">{daily_change_rate_str}</span>', # 색상 적용
                     "price_history": stock_info["price_history"],
                     "description": stock_info["description"],
                 }
             )
     stocks_df = pd.DataFrame(stocks_data)
-    st.dataframe(stocks_df[["섹터", "종목", "현재 주가", "전일 대비"]], hide_index=True) # "전일 대비" 컬럼 추가
-
-    selected_stock_all_info = st.selectbox(
-        "종목 선택 (기업 정보 및 주가 그래프)", stocks_df["종목"].tolist()
-    )
-    if selected_stock_all_info:
-        selected_stock_sector = stocks_df[
-            stocks_df["종목"] == selected_stock_all_info
-        ]["섹터"].iloc[0]
-        col1_info, col2_graph = st.columns([1, 2])
-
-        with col1_info:
-            st.subheader("기업 정보")
-            st.info(
-                f"**{selected_stock_all_info} ({selected_stock_sector})**\n\n{st.session_state['stocks'][selected_stock_sector][selected_stock_all_info]['description']}"
-            )
 
-        with col2_graph:
-            st.subheader("주가 그래프")
-            price_history_df = pd.DataFrame(
-                {
-                    "날짜": range(
-                        1,
-                        len(
-                            st.session_state["stocks"][selected_stock_sector][
-                                selected_stock_all_info
-                            ]["price_history"]
-                        )
-                        + 1,
-                    ),
-                    "주가": st.session_state["stocks"][selected_stock_sector][
-                        selected_stock_all_info
-                    ]["price_history"],
-                }
-            )
-            fig = px.line(
-                price_history_df,
-                x="날짜",
-                y="주가",
-                title=f"{selected_stock_all_info} ({selected_stock_sector}) 주가 변동",
-            )
-            st.plotly_chart(fig)
-    else:
-        st.info("종목을 선택하여 기업 정보와 주가 그래프를 확인하세요.")
+    sector_list = stocks_df['섹터'].unique() # 섹터 목록 추출
+
+    for sector in sector_list: # 섹터별로 정보 표시
+        st.subheader(f"[{sector}]")
+        sector_df = stocks_df[stocks_df['섹터'] == sector].reset_index(drop=True) # 섹터별 데이터프레임 생성
+        for index, row in sector_df.iterrows(): # 섹터별 데이터프레임 순회
+            col1, col2, col3, col4, col5 = st.columns([2, 2, 1.5, 1.5, 1]) # 컬럼 너비 조정
+            with col1:
+                st.markdown(f"**{row['종목']}**") # 종목명 굵게 표시
+            with col2:
+                st.metric(label="현재 주가", value=row['현재 주가']) # 현재 주가 metric 스타일 적용
+            with col3:
+                st.markdown(f"전일 대비: {row['전일 대비']}", unsafe_allow_html=True) # 전일 대비 등락률 및 색상 적용
+            with col4:
+                if st.button("기업 정보", key=f"info_{index}_{sector}", use_container_width=True): # 기업 정보 버튼
+                    st.session_state['selected_stock_info'] = row['종목'] # 선택 종목 저장
+                    st.session_state['selected_stock_sector_info'] = row['섹터'] # 선택 섹터 저장
+            with col5:
+                if st.button("주가 그래프", key=f"chart_{index}_{sector}", use_container_width=True): # 주가 그래프 버튼
+                    st.session_state['selected_stock_chart'] = row['종목'] # 선택 종목 저장
+                    st.session_state['selected_stock_sector_chart'] = row['섹터'] # 선택 섹터 저장
+        st.markdown("---") # 섹터 구분선
+
+    if 'selected_stock_info' in st.session_state: # 기업 정보 표시
+        selected_stock_info = st.session_state['selected_stock_info']
+        selected_stock_sector_info = st.session_state['selected_stock_sector_info']
+        st.info(f"**[{selected_stock_sector_info}] {selected_stock_info}**\n\n{st.session_state['stocks'][selected_stock_sector_info][selected_stock_info]['description']}")
+        del st.session_state['selected_stock_info'] # 세션 상태 삭제
+        del st.session_state['selected_stock_sector_info'] # 세션 상태 삭제
+
+    if 'selected_stock_chart' in st.session_state: # 주가 그래프 표시
+        selected_stock_chart = st.session_state['selected_stock_chart']
+        selected_stock_sector_chart = st.session_state['selected_stock_sector_chart']
+        price_history_df = pd.DataFrame(
+            {
+                "날짜": range(
+                    1,
+                    len(
+                        st.session_state["stocks"][selected_stock_sector_chart][
+                            selected_stock_chart
+                        ]["price_history"]
+                    )
+                    + 1,
+                ),
+                "주가": st.session_state["stocks"][selected_stock_sector_chart][
+                    selected_stock_chart
+                ]["price_history"],
+            }
+        )
+        fig = px.line(
+            price_history_df,
+            x="날짜",
+            y="주가",
+            title=f"[{selected_stock_sector_chart}] {selected_stock_chart} 주가 변동",
+        )
+        st.plotly_chart(fig)
+        del st.session_state['selected_stock_chart'] # 세션 상태 삭제
+        del st.session_state['selected_stock_sector_chart'] # 세션 상태 삭제
 
 
 def display_portfolio_table():
@@ -961,13 +1025,13 @@ def main():
 
     with col_main_ui:
         menu = st.tabs(
-            ['현재 주가', '내 포트폴리오', '주식 매수', '주식 매도', '어제 뉴스 해설']
+            ['현재 주가 및 기업 정보', '내 포트폴리오', '주식 매수', '주식 매도', '어제 뉴스 해설']
         )
 
         with menu[0]:
-            st.subheader("📈 현재 주가 및 기업 정보")
+            st.subheader("📈 현재 주가 및 기업 정보") # 탭 이름 변경
             st.markdown("주식 시장의 현재 가격과 기업 정보를 확인하세요.")
-            display_stock_prices()
+            display_stock_prices() # 수정된 함수 호출
 
         with menu[1]:
             st.subheader("📊 내 포트폴리오")