Update app.py

app.py

@@ -1,72 +1,73 @@
 import gradio as gr
-import speech_recognition as sr
-from Levenshtein import ratio
-import tempfile
-import numpy as np
-import soundfile as sf
-from transformers import pipeline
 
-# Load a grammar checking pipeline
-grammar_checker = pipeline("text-classification", model="textattack/bert-base-uncased-CoLA")
+# 전역 변수
+recordings = {}
 
-def transcribe_audio(file_info):
-    r = sr.Recognizer()
-    with tempfile.NamedTemporaryFile(delete=True, suffix=".wav") as tmpfile:
-        sf.write(tmpfile.name, data=file_info[1], samplerate=44100, format='WAV')
-        tmpfile.seek(0)
-        with sr.AudioFile(tmpfile.name) as source:
-            audio_data = r.record(source)
-    try:
-        text = r.recognize_google(audio_data)
-        return text
-    except sr.UnknownValueError:
-        return "Could not understand audio"
-    except sr.RequestError as e:
-        return f"Could not request results; {e}"
+# 음성 녹음 및 저장
+def record_and_submit_voice(student_name, voice):
+    if not student_name:
+        return "Please enter your name."
+    if student_name not in recordings:
+        recordings[student_name] = []
+    recordings[student_name].append({"voice": voice, "comments": []})
+    return f"Voice recorded and submitted successfully by {student_name}!"
 
-def pronunciation_correction(expected_text, file_info):
-    user_spoken_text = transcribe_audio(file_info)
-    similarity = ratio(expected_text.lower(), user_spoken_text.lower())
-    description = f"{similarity:.2f}"
+# 녹음 선택
+def get_student_recordings():
+    return list(recordings.keys())
 
-    if similarity >= 0.9:
-        feedback = "Excellent pronunciation!"
-    elif similarity >= 0.7:
-        feedback = "Good pronunciation!"
-    elif similarity >= 0.5:
-        feedback = "Needs improvement."
-    else:
-        feedback = "Poor pronunciation, try to focus more on clarity."
+# 녹음된 음성 재생
+def play_recording(selected_student):
+    if selected_student not in recordings:
+        return [], []
+    return [rec["voice"] for rec in recordings[selected_student]], recordings[selected_student][0]["comments"]
 
-    return feedback, description
-
-def check_sentence_accuracy(sentence):
-    results = grammar_checker(sentence)
-    if results[0]['label'] == 'LABEL_1':  # Assuming LABEL_1 indicates grammatical correctness
-        return "No errors found. Your sentence is correct!", []
-    else:
-        return "Errors found:", ["The sentence might be grammatically incorrect."]
+# 댓글 작성
+def write_comment(selected_student, commenter_name, comment):
+    if not selected_student or selected_student not in recordings:
+        return "Selected student's recording not found."
+    if not commenter_name:
+        return "Please enter your name."
+    if not comment:
+        return "Please enter a comment."
+    recordings[selected_student][0]["comments"].append(f"{commenter_name}: {comment}")
+    return f"Comment added successfully by {commenter_name}!"
 
+# Gradio 인터페이스 정의
 with gr.Blocks() as app:
-    with gr.Row():
-        user_input_textbox = gr.Textbox(label="Type a Sentence", placeholder="Type the sentence you want to practice here...")
-    audio_input = gr.Audio(label="Upload Audio File", type="numpy")
-    check_pronunciation_button = gr.Button("Check Pronunciation")
-    pronunciation_feedback = gr.Textbox(label="Pronunciation Feedback")
-    pronunciation_score = gr.Number(label="Pronunciation Accuracy Score: 0 (No Match) ~ 1 (Perfect)")
-    check_accuracy_button = gr.Button("Check Sentence Accuracy")
-    sentence_accuracy_feedback = gr.Textbox(label="Sentence Accuracy Feedback")
+    with gr.Tab("Record Voice"):
+        student_name_input = gr.Textbox(placeholder="Enter your name", label="Your Name")
+        voice_input = gr.Audio(source="microphone", type="numpy", label="Record your voice")
+        submit_voice_button = gr.Button("Submit Voice")
+        voice_output = gr.Textbox(label="Status")
+
+        submit_voice_button.click(
+            record_and_submit_voice, 
+            inputs=[student_name_input, voice_input], 
+            outputs=voice_output
+        )
+
+    with gr.Tab("Listen and Comment"):
+        student_selector = gr.Dropdown(choices=get_student_recordings, label="Select a student to listen")
+        play_voice_button = gr.Button("Play Voice")
+        recording_output = gr.Audio(label="Selected Recording", interactive=False)
+        comment_input = gr.Textbox(placeholder="Write your comment here...", label="Write Comment")
+        commenter_name_input = gr.Textbox(placeholder="Enter your name", label="Your Name")
+        submit_comment_button = gr.Button("Submit Comment")
+        comment_status_output = gr.Textbox(label="Comment Status")
+        comments_display = gr.Textbox(label="Comments", interactive=False)
 
-    check_pronunciation_button.click(
-        pronunciation_correction,
-        inputs=[user_input_textbox, audio_input],
-        outputs=[pronunciation_feedback, pronunciation_score]
-    )
+        play_voice_button.click(
+            play_recording, 
+            inputs=student_selector, 
+            outputs=[recording_output, comments_display]
+        )
 
-    check_accuracy_button.click(
-        check_sentence_accuracy,
-        inputs=[user_input_textbox],
-        outputs=[sentence_accuracy_feedback]
-    )
+        submit_comment_button.click(
+            write_comment, 
+            inputs=[student_selector, commenter_name_input, comment_input], 
+            outputs=comment_status_output
+        )
 
-app.launch(debug=True)
+# 인터페이스 실행
+app.launch()