Update app.py

app.py

@@ -5,7 +5,7 @@ import time
 
 # from elevenlabs import set_api_key
 
-
+import requests
 import whisper
 
 
@@ -91,7 +91,26 @@ remember to not to include question number.
 """)
 
 
+from gradio_client import Client
+API_URL = "https://sanchit-gandhi-whisper-jax.hf.space/"
+
+# set up the Gradio client
+client = Client(API_URL)
+
 
+def transcribe_audio(audio_path, task="transcribe", return_timestamps=False):
+    """Function to transcribe an audio file using the Whisper JAX endpoint."""
+    if task not in ["transcribe", "translate"]:
+        raise ValueError("task should be one of 'transcribe' or 'translate'.")
+
+    text, runtime = client.predict(
+        audio_path,
+        task,
+        return_timestamps,
+        api_name="/predict_1",
+    )
+    return text
+    
 
 def convert_tuples_to_list(tuples_list):
     result = []
@@ -128,8 +147,11 @@ def run_text_prompt(message, chat_history):
 def run_audio_prompt(audio, chat_history):
     if audio is None:
         return None, chat_history
+    
+    message_transcription = transcribe_audio(audio)
 
-    message_transcription = model.transcribe(audio)["text"]
+
+    # message_transcription = model.transcribe(audio)["text"]
     _, chat_history = run_text_prompt(message_transcription, chat_history)
     return None, chat_history
 
@@ -164,9 +186,9 @@ with gr.Blocks(title="hi") as app2:
         # fn = run_audio_prompt,
         # inputs = [audio, chatbot],
         # outputs = [audio, chatbot]
-
-        send_audio_button = gr.Button("Send Audio", interactive=True)
-        send_audio_button.click(run_audio_prompt, [audio, chatbot], [audio, chatbot])
+        audio.change(run_audio_prompt, [audio, chatbot], [audio, chatbot])
+        # send_audio_button = gr.Button("Send Audio", interactive=True)
+        # send_audio_button.click(run_audio_prompt, [audio, chatbot], [audio, chatbot])
 
 demo = gr.TabbedInterface([app2], [ "Interview"])