Update app.py

app.py

@@ -5,17 +5,24 @@ from lid import identify  # Import Language Identification model
 
 # Function to detect language and transcribe speech
 def auto_transcribe(audio):
-    # Detect language (returns language code like "eng" or "swh")
-    detected_lang = identify(audio)
+    detected_lang = identify(audio)  # Detect language
 
-    # Ensure the detected language is Swahili or English
-    if detected_lang not in ["eng", "swh"]:
-        return "Error: Only English and Swahili are supported."
+    print(f"Detected Language: {detected_lang}")  # Debugging: Print detected language
+
+    # Check if the detected language is valid
+    if detected_lang not in ["eng", "swh", "en", "sw"]:
+        return f"Error: Only English and Swahili are supported. Detected: {detected_lang}"
+
+    # Normalize language codes if needed
+    if detected_lang in ["en"]:
+        detected_lang = "eng"
+    elif detected_lang in ["sw"]:
+        detected_lang = "swh"
 
     # Transcribe using detected language
     return transcribe(audio, lang=detected_lang)
 
-# Speech-to-Text Interface with Auto Language Detection
+# Speech-to-Text Interface
 mms_transcribe = gr.Interface(
     fn=auto_transcribe,
     inputs=gr.Audio(),