import gradio as gr

# 상태를 관리하는 전역 변수
submitted_question = ""
recordings = {}
comments = {}

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

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

# 댓글 저장
def write_comment(comment):
    global submitted_question
    if submitted_question:
        comments[submitted_question].append(comment)
        return f"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, value=lambda: submitted_question)
        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_display = gr.Textbox(label="Question", interactive=False, value=lambda: submitted_question)
        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)

demo.launch()