Update app.py

app.py

@@ -17,7 +17,10 @@ from huggingface_hub import InferenceClient
 import inspect
 import logging
 import shutil
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
 
+executor = ThreadPoolExecutor()
 
 # Set up basic configuration for logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -614,15 +617,21 @@ Write a detailed and complete response that answers the following user question:
         
         logging.info("Finished generating response")
 
-def transcribe(audio_file):
+async def transcribe_async(audio_file):
     if audio_file is None:
-        return ""
+        return "", "No audio file provided"
     
-    with open(audio_file, "rb") as f:
-        audio_data = f.read()
-    
-    response = whisper_api(audio_data)
-    return response["text"]
+    try:
+        def process_audio():
+            with open(audio_file, "rb") as f:
+                audio_data = f.read()
+            return whisper_api(audio_data)["text"]
+        
+        loop = asyncio.get_event_loop()
+        response = await loop.run_in_executor(executor, process_audio)
+        return response, "Transcription completed successfully"
+    except Exception as e:
+        return "", f"Error during transcription: {str(e)}"
 
 def vote(data: gr.LikeData):
     if data.liked:
@@ -684,6 +693,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             audio_input = gr.Audio(sources="microphone", type="filepath", label="Speak your query")
             transcribe_button = gr.Button("Transcribe")
+            transcription_status = gr.Textbox(label="Transcription Status")
         
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(
@@ -718,11 +728,12 @@ with gr.Blocks() as demo:
     update_output = gr.Textbox(label="Update Status")
     delete_button = gr.Button("Delete Selected Documents")
     
-    # Connect components
+    # Update the click event
     transcribe_button.click(
-        transcribe, 
+        transcribe_async, 
         inputs=[audio_input], 
-        outputs=[query_textbox]
+        outputs=[query_textbox, transcription_status],
+        _js="startTranscription"  # Add a JavaScript function to show a loading indicator
     )
     
     submit_button.click(