Refactor code

app.py

@@ -13,25 +13,26 @@ whisper_model = whisper.load_model("base")
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-def get_response_from_chatbot(audio):
-    try:
-        api = ChatGPT(session_token)
-        audio = whisper.load_audio(audio)
-        audio = whisper.pad_or_trim(audio)
-    
-        mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
-    
-        _, probs = whisper_model.detect_language(mel)
-        translate_options = whisper.DecodingOptions(task="translate", fp16 = False)
-    
-        translation = whisper.decode(whisper_model, mel, translate_options)
-    
-        print("Language Spoken: " + transcription.language)
-      
-        print("Translated: " + translation.text)
+def translate(audio):
+    audio = whisper.load_audio(audio)
+    audio = whisper.pad_or_trim(audio)
+
+    mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
+
+    _, probs = whisper_model.detect_language(mel)
+    translate_options = whisper.DecodingOptions(task="translate", fp16 = False)
+
+    translation = whisper.decode(whisper_model, mel, translate_options)
+
+    print("Language Spoken: " + transcription.language)
+  
+    print("Translated: " + translation.text)
+    return translation.text
     
-        
-        resp = api.send_message(translation.text)    
+def get_response_from_chatbot(text):
+    try:
+        api = ChatGPT(session_token)    
+        resp = api.send_message(text)    
         api.refresh_auth() 
         api.reset_conversation() 
         response = resp['message']
@@ -40,10 +41,11 @@ def get_response_from_chatbot(audio):
         response = "Sorry, I'm tired. Please try again in some time"
     return response
 
-def chat(message, chat_history):      
+def chat(audio, 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)
@@ -151,7 +153,7 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
             chatbot = gr.Chatbot(elem_id="chat_bot", visible=False).style(color_map=("green", "blue"))
             chatbot1 = gr.Chatbot(elem_id="chat_bot1").style(color_map=("green", "blue"))
         with gr.Row(elem_id="prompt_row"):
-            prompt_input_audio = audio_input_r = gr.Audio(label = 'Record Audio Input',source="microphone",type="filepath")
+            prompt_input_audio = gr.Audio(label = 'Record Audio Input',source="microphone",type="filepath")
             chat_history = gr.Textbox(lines=4, label="prompt", visible=False)
             submit_btn = gr.Button(value = "submit",elem_id="submit-btn").style(
                     margin=True,