Update app.py

app.py

@@ -3,35 +3,43 @@ from gtts import gTTS
 from io import BytesIO
 import pandas as pd
 
-# 주어진 학생 명단
-students = [
-    "Hui Chin Nam", "Li Jiaying", "Mo Siyu", "Xu Tian", "Ye Wenting", "Zhou Qinyi",
-    "강수민", "김남훈", "김서연", "김성주", "김정하", "김효준", "남수민", "남예도", "민병기",
-    "박나원", "박수현", "박지은", "박채연", "박채원", "백승원", "서정현", "송수현", "송은영",
-    "안효정", "엄은수", "이수민", "이윤범", "이은영", "이정민", "이정민", "장한나", "전윤서",
-    "정윤서", "정진욱", "주선제", "한근효", "한은찬", "황민주"
-]
+# 학생 명단을 두 개의 수업으로 나눔
+students_dict = {
+    "매체철학": [
+        "Hui Chin Nam", "Li Jiaying", "Mo Siyu", "Xu Tian", "Ye Wenting", "Zhou Qinyi",
+        "강수민", "김남훈", "김서연", "김성주", "김정하", "김효준", "남수민", "남예도", "민병기"
+    ],
+    "경제저널리즘": [
+        "박나원", "박수현", "박지은", "박채연", "박채원", "백승원", "서정현", "송수현", "송은영",
+        "안효정", "엄은수", "이수민", "이윤범", "이은영", "이정민", "이정민", "장한나", "전윤서",
+        "정윤서", "정진욱", "주선제", "한근효", "한은찬", "황민주"
+    ]
+}
+
+# 사이드바에서 수업 선택
+selected_class = st.sidebar.selectbox("수업 선택", list(students_dict.keys()))
+
+# 선택된 수업의 학생 명단을 가져옴
+students = students_dict[selected_class]
 
 # 출석 기록 초기화
-if 'attendance' not in st.session_state:
+if 'attendance' not in st.session_state or 'current_class' not in st.session_state or st.session_state.current_class != selected_class:
     st.session_state.attendance = {student: None for student in students}
-
-# 현재 출석 체크 중인 학생 인덱스 초기화
-if 'current_index' not in st.session_state:
     st.session_state.current_index = 0
+    st.session_state.current_class = selected_class
 
-# 현재 학생의 이름 불러오기
+# 현재 출석 체크 중인 학생 인덱스 초기화
 current_student = students[st.session_state.current_index]
 
-st.title("자동 출석 체크 시스템")
+st.title(f"{selected_class} 출석 체크 시스템")
 
 # 이름을 음성으로 출력하는 함수
-def play_name(student_name):
+def generate_audio(student_name):
     tts = gTTS(text=student_name, lang='ko')
     fp = BytesIO()
     tts.write_to_fp(fp)
     fp.seek(0)
-    st.audio(fp, format='audio/mp3')
+    return fp
 
 # 4열로 학생 명단을 표시
 st.write("학생 명단:")
@@ -43,9 +51,6 @@ for i, student in enumerate(students):
     else:
         col.write(f"   {student}")
 
-# 현재 학생 이름을 음성으로 부름
-play_name(current_student)
-
 # 출석과 결석 버튼
 col1, col2 = st.columns(2)
 with col1:
@@ -53,18 +58,22 @@ with col1:
         st.session_state.attendance[current_student] = "출석"
         if st.session_state.current_index < len(students) - 1:
             st.session_state.current_index += 1
+            next_student = students[st.session_state.current_index]
+            audio_file = generate_audio(next_student)
+            st.audio(audio_file, format='audio/mp3')  # 다음 학생 이름을 자동으로 재생
         else:
             st.session_state.current_index = None  # 모든 학생 완료
-        st.session_state.query_params = {'index': st.session_state.current_index}
 
 with col2:
     if st.button("결석"):
         st.session_state.attendance[current_student] = "결석"
         if st.session_state.current_index < len(students) - 1:
             st.session_state.current_index += 1
+            next_student = students[st.session_state.current_index]
+            audio_file = generate_audio(next_student)
+            st.audio(audio_file, format='audio/mp3')  # 다음 학생 이름을 자동으로 재생
         else:
             st.session_state.current_index = None  # 모든 학생 완료
-        st.session_state.query_params = {'index': st.session_state.current_index}
 
 # 모든 학생의 출석 체크가 완료된 후 결석 학생들을 표시
 if st.session_state.current_index is None:
@@ -82,8 +91,12 @@ if st.session_state.current_index is None:
         st.download_button(
             label="결석 학생 목록 다운로드",
             data=csv,
-            file_name='결석_학생_목록.csv',
+            file_name=f'{selected_class}_결석_학생_목록.csv',
             mime='text/csv',
         )
     else:
         st.write("결석한 학생이 없습니다.")
+else:
+    # 현재 학생 이름을 음성으로 부름 (페이지 로드 시)
+    audio_file = generate_audio(current_student)
+    st.audio(audio_file, format='audio/mp3')