Update app.py

app.py

@@ -9,11 +9,11 @@ import torch
 session_token = os.environ.get('SessionToken')      
 # logger.info(f"session_token_: {session_token}")
 
-whisper_model = whisper.load_model("base")
+whisper_model = whisper.load_model("medium")
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-def translate(audio):
+def transcribe(audio):
     audio = whisper.load_audio(audio)
     audio = whisper.pad_or_trim(audio)
 
@@ -39,11 +39,10 @@ def get_response_from_chatbot(text):
         response = "Sorry, I'm tired. Please try again in some time"
     return response
 
-def chat(audio, chat_history):      
+def chat(message, chat_history):      
     out_chat = []
     if chat_history != '':
         out_chat = json.loads(chat_history)
-    message = translate(audio)
     response = get_response_from_chatbot(message)
     out_chat.append((message, response))
     chat_history = json.dumps(out_chat)
@@ -152,12 +151,23 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
             chatbot1 = gr.Chatbot(elem_id="chat_bot1").style(color_map=("green", "blue"))
         with gr.Row(elem_id="prompt_row"):
             prompt_input_audio = gr.Audio(label = 'Record Audio Input',source="microphone",type="filepath")
+            prompt_input = gr.Textbox(lines=2, label="Input text",show_label=True)
             chat_history = gr.Textbox(lines=4, label="prompt", visible=False)
-            submit_btn = gr.Button(value = "submit",elem_id="submit-btn").style(
+            transcribe_btn = gr.Button(value = "Transcribe").style(
                     margin=True,
                     rounded=(True, True, True, True),
                     width=100
                 )
+            transcribe_btn_btn.click(fn=transcribe, 
+                             inputs=prompt_input_audio 
+                             outputs=prompt_input
+                            )
+            submit_btn = gr.Button(value = "Submit",elem_id="submit-btn").style(
+                    margin=True,
+                    rounded=(True, True, True, True),
+                    width=100
+                )
+            
             submit_btn.click(fn=chat, 
                              inputs=[prompt_input_audio, chat_history], 
                              outputs=[chatbot, chat_history],