Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 from gtts import gTTS
 from pydub import AudioSegment
-import speech_recognition as sr
+from pydub.playback import play
+import tempfile
 import os
 
 # 학생 리스트
@@ -13,48 +14,42 @@ absentees = []
 # Streamlit UI
 st.title("출석 확인 시스템")
 
-# 오디오 파일 생성 함수
-def create_combined_audio_file(students):
-    combined = AudioSegment.silent(duration=0)  # 시작에 짧은 무음 추가
-    for student in students:
-        tts = gTTS(student + " 학생")
-        tts.save(f"{student}.mp3")
-        audio = AudioSegment.from_mp3(f"{student}.mp3")
-        combined += audio + AudioSegment.silent(duration=500)  # 학생 이름 사이에 0.5초 간격 추가
-    combined.export("combined_students.wav", format="wav")  # WAV 형식으로 저장
-    return "combined_students.wav"
-
-# 오디오 파일 생성
-audio_file = create_combined_audio_file(student_list)
-
-# 음성 인식을 위한 함수
-def recognize_audio(file):
-    recognizer = sr.Recognizer()
-    with sr.AudioFile(file) as source:
-        audio = recognizer.record(source)
-        try:
-            return recognizer.recognize_google(audio, language="ko-KR")
-        except sr.UnknownValueError:
-            return "인식 불가"
-        except sr.RequestError:
-            return "STT 서비스 요청 실패"
+if 'current_index' not in st.session_state:
+    st.session_state.current_index = 0
 
-# 오디오 재생 및 음성 인식 실행
-if st.button("전체 오디오 재생"):
-    st.audio(audio_file)
-    recognized_text = recognize_audio(audio_file)
-    st.subheader("호명된 학생 이름")
-    st.write(recognized_text)
+# 오디오 파일 생성 함수
+def create_audio_for_student(student):
+    tts = gTTS(student + " 학생")
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp3')
+    tts.save(temp_file.name)
+    return temp_file.name
+
+# 현재 학생 이름 가져오기
+current_student = student_list[st.session_state.current_index]
+
+# 학생 이름 불러오기
+if st.button("다음 학생"):
+    # 현재 학생의 이름을 오디오로 재생
+    audio_file = create_audio_for_student(current_student)
+    audio = AudioSegment.from_mp3(audio_file)
+    play(audio)
+    os.remove(audio_file)  # 임시 파일 삭제
+    
+    # 화면에 현재 학생 이름 표시
+    st.subheader(f"현재 학생: {current_student}")
 
     # 결석 처리
-    if recognized_text:
-        selected_student = st.selectbox("결석자 선택", options=recognized_text.split())
-
-        if st.button("결석자로 등록"):
-            absentees.append(selected_student)
-            with open("absentees.txt", "a") as f:
-                f.write(f"{selected_student}\n")
-            st.success(f"{selected_student}가 결석자로 등록되었습니다.")
+    if st.radio(f"{current_student} 출석 여부", ("출석", "결석")) == "결석":
+        absentees.append(current_student)
+        with open("absentees.txt", "a") as f:
+            f.write(f"{current_student}\n")
+        st.success(f"{current_student}가 결석자로 등록되었습니다.")
+    
+    # 다음 학생으로 인덱스 이동
+    if st.session_state.current_index < len(student_list) - 1:
+        st.session_state.current_index += 1
+    else:
+        st.write("모든 학생의 이름을 불렀습니다.")
 
 # 결석자 리스트 보여주기
 st.subheader("결석자 리스트")