Update app.py

app.py

@@ -5,24 +5,24 @@ from lid import identify  # Import Language Identification model
 
 # Function to detect language and transcribe speech
 def auto_transcribe(audio):
-    detected_lang = identify(audio)  # Detect language
-
-    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"]:
+    # Detect language
+    detected_lang = identify(audio)
+    
+    # Debug: Print detected language
+    print(f"Detected Language: {detected_lang}")  
+
+    # Ensure the detected language is Swahili or English
+    if detected_lang.startswith("swh"):
         detected_lang = "swh"
+    elif detected_lang.startswith("eng"):
+        detected_lang = "eng"
+    else:
+        return f"Error: Detected language '{detected_lang}' is not supported."
 
     # Transcribe using detected language
     return transcribe(audio, lang=detected_lang)
 
-# Speech-to-Text Interface
+# Speech-to-Text Interface with Auto Language Detection
 mms_transcribe = gr.Interface(
     fn=auto_transcribe,
     inputs=gr.Audio(),