Update app.py

app.py

@@ -3,7 +3,6 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
 import matplotlib.font_manager as fm
-from PIL import Image
 
 # 폰트 설정
 font_path = r"H2GTRM.TTF"
@@ -76,13 +75,23 @@ elif page == "연령대 및 성별에 따른 상위 3개 장르":
     # 장르별 대출 건수 집계
     genre_age_sex = df.groupby(['연령대', '성별', '주제분류명'])['대출건수'].sum().unstack().fillna(0)
 
+    # 연령대와 성별 선택
+    ages = df['연령대'].unique()
+    genders = df['성별'].unique()
+    
+    selected_age = st.selectbox("연령대 선택", options=ages)
+    selected_gender = st.selectbox("성별 선택", options=genders)
+    
+    # 필터링된 데이터 추출
+    filtered_genres = genre_age_sex.loc[selected_age, selected_gender] if selected_age in genre_age_sex.index and selected_gender in genre_age_sex.columns else pd.Series(dtype=int)
+
     # 히트맵 시각화
     st.write("연령대 및 성별에 따른 장르 선호도 히트맵")
     fig, ax = plt.subplots(figsize=(12, 8))
-    sns.heatmap(genre_age_sex, annot=False, cmap='YlGnBu', linewidths=0.5, ax=ax)
-    ax.set_title('연령대 및 성별에 따른 장르 선호도')
+    sns.heatmap(genre_age_sex.loc[[selected_age], [selected_gender]], annot=False, cmap='YlGnBu', linewidths=0.5, ax=ax)
+    ax.set_title(f'{selected_age} - {selected_gender}의 장르 선호도')
     st.pyplot(fig)
 
-    # 연령대와 성별에 따른 주제분류명 출력
-    st.write("연령대 및 성별에 따른 주제 분류명:")
-    st.dataframe(genre_age_sex)
+    # 선택된 카테고리에 따른 주제 분류명 출력
+    st.write(f"{selected_age} - {selected_gender}에 따른 주제 분류명:")
+    st.write(filtered_genres)