Update app.py

app.py

@@ -3,7 +3,7 @@ import speech_recognition as sr
 import difflib
 
 # Hàm chuyển giọng nói thành văn bản từ file
-def transcribe_speech(audio_file):
+def transcribe_speech(audio_file, language_code):
     recognizer = sr.Recognizer()
 
     # Load audio file
@@ -13,8 +13,8 @@ def transcribe_speech(audio_file):
         audio = recognizer.record(source)
 
         try:
-            # Chuyển giọng nói thành văn bản
-            text = recognizer.recognize_google(audio, language="vi-VN")
+            # Chuyển giọng nói thành văn bản với ngôn ngữ được chọn
+            text = recognizer.recognize_google(audio, language=language_code)
             return text
         except sr.UnknownValueError:
             return "Không thể nhận diện giọng nói"
@@ -38,8 +38,11 @@ def compare_transcription(transcribed_text, reference_text):
     return accuracy, incorrect_words
 
 # Hàm tích hợp để dùng trên Gradio
-def process_speech(reference_text, audio_file):
-    transcribed_text = transcribe_speech(audio_file)
+def process_speech(reference_text, audio_file, language):
+    # Map language selection to corresponding language code
+    language_code = "vi-VN" if language == "Vietnamese" else "en-US"
+    
+    transcribed_text = transcribe_speech(audio_file, language_code)
     
     if "Lỗi" in transcribed_text or "Không thể nhận diện" in transcribed_text:
         return transcribed_text, None, None
@@ -57,15 +60,22 @@ def build_interface():
         # Input cho file âm thanh
         audio_input = gr.Audio(type="filepath", label="Tải lên file âm thanh")
         
+        # Dropdown cho chọn ngôn ngữ
+        language_selector = gr.Dropdown(
+            choices=["English", "Vietnamese"],
+            label="Chọn ngôn ngữ",
+            value="Vietnamese"  # Default value
+        )
+        
         # Output hiển thị kết quả
         transcribed_text = gr.Textbox(label="Văn bản bạn nói")
         accuracy = gr.Textbox(label="Độ chính xác (%)")
         incorrect_words = gr.Textbox(label="Những từ nói sai")
 
-        # Nút ghi âm được kết nối với chức năng xử lý
+        # Nút kiểm tra được kết nối với chức năng xử lý
         gr.Button("Ghi âm và kiểm tra").click(
             fn=process_speech, 
-            inputs=[reference_text, audio_input], 
+            inputs=[reference_text, audio_input, language_selector], 
             outputs=[transcribed_text, accuracy, incorrect_words]
         )