Include option to transcribe and use same language as spoken

app.py

@@ -8,7 +8,7 @@ import torch
 
 session_token = os.environ.get('SessionToken')      
 # logger.info(f"session_token_: {session_token}")
-
+api = ChatGPT(session_token)
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -23,22 +23,21 @@ all_special_ids = whisper_model.tokenizer.all_special_ids
 transcribe_token_id = all_special_ids[-5]
 translate_token_id = all_special_ids[-6]
 
-def translate(audio):
-    task = "translate"
+def translate_or_transcribe(audio, task):
     whisper_model.model.config.forced_decoder_ids = [[2, transcribe_token_id if task=="transcribe" else translate_token_id]]
     text = whisper_model(audio)["text"]
     return text
     
 def get_response_from_chatbot(text):
     try:
-        api = ChatGPT(session_token)    
+        if reset_conversation:
+            api.refresh_auth() 
+            api.reset_conversation() 
         resp = api.send_message(text)    
-        api.refresh_auth() 
-        api.reset_conversation() 
         response = resp['message']
       # logger.info(f"response_: {response}")
     except:
-        response = "Sorry, I'm tired. Please try again in some time"
+        response = "Sorry, chatGPT queue is full. Please try again in some time"
     return response
 
 def chat(message, chat_history):      
@@ -162,8 +161,10 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
                                     )
             translate_btn = gr.Button("Check Whisper first ? 👍")
 
+        reset_conversation = gr.Checkbox(label="Reset conversation?", value=False)
+        whisper_task = gr.Radio(["Translate to English", "Transcribe in Spoken Language"], value="Translate to English", show_label=False)
         with gr.Row(elem_id="prompt_row"):
-            prompt_input = gr.Textbox(lines=2, label="Input text (translated)",show_label=True)
+            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 = "Send to chatGPT",elem_id="submit-btn").style(
                     margin=True,
@@ -173,8 +174,8 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
             
              
 
-        translate_btn.click(fn=translate, 
-                         inputs=prompt_input_audio, 
+        translate_btn.click(fn=translate_or_transcribe, 
+                         inputs=[prompt_input_audio,whisper_task], 
                          outputs=prompt_input
                         )