Update app.py

app.py

@@ -19,6 +19,7 @@ def query(api_url, payload=None, data=None):
         response = requests.post(api_url, headers=headers, json=payload)
     return response.json()
 
+# Define the function to translate speech
 # Define the function to translate speech
 def translate_speech(audio_file):
     print(f"Type of audio: {type(audio_file)}, Value of audio: {audio_file}")  # Debug line
@@ -26,7 +27,14 @@ def translate_speech(audio_file):
     # Use the ASR pipeline to transcribe the audio
     data = audio_file.read()
     output = query(ASR_API_URL, data=data)
-    transcription = output["text"]
+    print(f"Output: {output}")  # Debug line
+
+    # Check if 'text' key exists in the output
+    if 'text' in output:
+        transcription = output["text"]
+    else:
+        print("Key 'text' does not exist in the output.")
+        return
 
     # Use the translation pipeline to translate the transcription
     translated_text = query(TRANSLATION_API_URL, {"inputs": transcription})
@@ -38,6 +46,7 @@ def translate_speech(audio_file):
     # Display the audio output
     return Audio(audio_bytes)
 
+
 # Define the Gradio interface
 iface = gr.Interface(
     fn=translate_speech,