import gradio as gr

# 저장된 음성 파일을 관리하는 딕셔너리
recordings = {}
# 저장된 댓글을 관리하는 딕셔너리
comments = {}
# 선생님이 제출한 질문을 저장하는 변수
submitted_question = ""

# 선생님의 질문 입력 받기
def teacher_question(question):
    global submitted_question
    submitted_question = question
    return question

# 학생들의 음성 녹음 및 저장
def record_student_voice(voice):
    global submitted_question
    if submitted_question:
        if submitted_question in recordings:
            recordings[submitted_question].append(voice)
        else:
            recordings[submitted_question] = [voice]
        return "Voice recorded successfully for question: " + submitted_question
    else:
        return "Please submit a question first."

# 댓글 저장
def write_comment(comment):
    global submitted_question
    if submitted_question:
        if submitted_question in comments:
            comments[submitted_question].append(comment)
        else:
            comments[submitted_question] = [comment]
        return "Comment added successfully for question: " + submitted_question
    else:
        return "Please submit a question first."

# Gradio 인터페이스 정의
with gr.Blocks() as demo:
    with gr.Tab("Teacher's Question"):
        question_input = gr.Textbox(lines=2, placeholder="Enter your question here...", label="Teacher's Question")
        submit_question = gr.Button("Submit")
        output_question = gr.Textbox(label="Submitted Question")
        submit_question.click(teacher_question, inputs=question_input, outputs=output_question)

    with gr.Tab("Record Voice"):
        question_display = gr.Textbox(label="Question", interactive=False)
        voice_input = gr.Audio(type="numpy", label="Record your voice")
        submit_voice = gr.Button("Submit Voice")
        output_voice = gr.Textbox(label="Status")
        submit_voice.click(record_student_voice, inputs=[voice_input], outputs=output_voice)

    with gr.Tab("Write Comment"):
        comment_input = gr.Textbox(placeholder="Your comment", label="Comment")
        submit_comment = gr.Button("Submit Comment")
        output_comment = gr.Textbox(label="Status")
        submit_comment.click(write_comment, inputs=[comment_input], outputs=output_comment)

    def update_question_display():
        return submitted_question

    demo.load(update_question_display, outputs=question_display)

demo.launch()