Update app.py

app.py

@@ -108,7 +108,6 @@ def save_radar_chart_image(fig):
   img_bytes = pio.to_image(fig, format="png", engine="kaleido")
   return img_bytes
 
-
 def main():
     st.title('學生成績雷達圖產生器')
 
@@ -140,15 +139,15 @@ def main():
               
               st.write("### 選擇要下載的圖片")
               
-              #初始化選取狀態
+              # 初始化選取狀態
               if 'selected_images' not in st.session_state:
-                st.session_state.selected_images = {student:False for student in df['姓名'] if student not in comparison_rows}
+                  st.session_state.selected_images = {}
 
               # 建立圖片核取方塊
               image_bytes = {}
               for student in df['姓名']:
                 if student in comparison_rows:
-                  continue
+                    continue
 
                 fig = create_radar_chart(df, [student] + comparison_rows, selected_columns)
                 fig = apply_font_to_all_text(fig)
@@ -158,9 +157,14 @@ def main():
               cols = st.columns(3) # 排成三列
               for i, student in enumerate(image_bytes.keys()):
                 with cols[i%3]:
-                  st.image(image_bytes[student],use_column_width=True)
-                  st.checkbox(f"下載 {student}", key=student, value=st.session_state.selected_images[student])
-                  st.session_state.selected_images[student] = st.session_state[student] # 更新選取狀態
+                  st.image(image_bytes[student],use_container_width=True)
+                  try:
+                      st.checkbox(f"下載 {student}", key=student, value=st.session_state.selected_images[student])
+                      st.session_state.selected_images[student] = st.session_state[student] # 更新選取狀態
+                  except KeyError:
+                       st.session_state.selected_images[student] = False  #如果還沒有這個學生則預設為不選
+                       st.checkbox(f"下載 {student}", key=student, value=st.session_state.selected_images[student])
+                       st.session_state.selected_images[student] = st.session_state[student] # 更新選取狀態
               
               #全部選取和全部不選
               if st.button("全部選取"):