Update app.py

app.py

@@ -12,7 +12,8 @@ def get_statistic_codes():
     if response.status_code == 200:
         data = response.json()
         df = pd.DataFrame(data['StatisticTableList']['row'])
-        return df[['STAT_CODE', 'STAT_NAME']]
+        df['코드와 이름'] = df['STAT_CODE'] + ": " + df['STAT_NAME']
+        return df[['STAT_CODE', '코드와 이름']]
     else:
         st.error(f"Failed to retrieve data: {response.status_code}")
         return None
@@ -64,8 +65,8 @@ if menu == "통계 코드 조회 및 데이터 분석":
     # 통계 코드 조회 및 선택
     stat_codes_df = get_statistic_codes()
     if stat_codes_df is not None:
-        st.dataframe(stat_codes_df)
-        selected_code = st.selectbox("조회할 통계 코드를 선택하세요:", stat_codes_df['STAT_CODE'])
+        selected_code_name = st.selectbox("조회할 통계 코드를 선택하세요:", stat_codes_df['코드와 이름'])
+        selected_code = selected_code_name.split(": ")[0]  # 통계 코드만 추출
         
         # 통계 데이터 조회 및 출력
         period = st.text_input("조회 주기 입력 (예: A, Q, M):", "A")
@@ -77,6 +78,12 @@ if menu == "통계 코드 조회 및 데이터 분석":
             if stat_data_df is not None:
                 st.write(f"선택한 통계 코드: {selected_code}")
                 st.dataframe(stat_data_df)
+                
+                # Excel 파일로 저장
+                st.write("데이터를 Excel 파일로 저장하려면 아래 버튼을 클릭하세요.")
+                if st.button("Excel로 저장"):
+                    stat_data_df.to_excel(f"{selected_code}_data.xlsx", index=False)
+                    st.success(f"데이터가 {selected_code}_data.xlsx 파일로 저장되었습니다.")
 
 elif menu == "100대 통계지표 실시간 확인":
     st.header("100대 통계지표 실시간 확인")
@@ -84,3 +91,4 @@ elif menu == "100대 통계지표 실시간 확인":
     top_100_df = get_top_100_statistics()
     if top_100_df is not None:
         st.dataframe(top_100_df)
+