Update app.py

app.py

@@ -6,11 +6,8 @@ import random
 from transformers import pipeline
 import torch
 
-from webdriver_manager.chrome import ChromeDriverManager
-# import chromedriver_autoinstaller
-# chromedriver_autoinstaller.install()
-
-session_token = os.environ.get('SessionToken')      
+session_token = os.environ.get('SessionToken')
+api = ChatGPT(session_token)
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -25,36 +22,29 @@ all_special_ids = whisper_model.tokenizer.all_special_ids
 transcribe_token_id = all_special_ids[-5]
 translate_token_id = all_special_ids[-6]
 
-def get_api():
-    api = ChatGPT(session_token)
-    return api
-    
 def translate_or_transcribe(audio, task):
     whisper_model.model.config.forced_decoder_ids = [[2, transcribe_token_id if task=="Transcribe in Spoken Language" else translate_token_id]]
     text = whisper_model(audio)["text"]
     return text
     
-def get_response_from_chatbot(api,text):
-    if api is None:
-        return "Sorry, the chatGPT API has some issues. Please try again later"
+def get_response_from_chatbot(text):
     try:
-      resp = api.send_message(text)    
-      api.refresh_auth()
-      # api.reset_conversation() 
-      response = resp['message']
+        resp = api.send_message(text)    
+        response = resp['message']
+      # logger.info(f"response_: {response}")
     except:
-      response = "Sorry, the chatGPT queue is full. Please try again later"
+        response = "Sorry, the chatGPT queue is full. Please try again in some time"
     return response
 
-def chat(api,message, chat_history):      
+def chat(message, chat_history):      
     out_chat = []
     if chat_history != '':
         out_chat = json.loads(chat_history)
-    response = get_response_from_chatbot(api,message)
+    response = get_response_from_chatbot(message)
     out_chat.append((message, response))
     chat_history = json.dumps(out_chat)
     logger.info(f"out_chat_: {len(out_chat)}")
-    return api,out_chat, chat_history
+    return out_chat, chat_history
 
 start_work = """async() => {
     function isMobile() {
@@ -99,7 +89,6 @@ start_work = """async() => {
     
         page1.style.display = "none";
         page2.style.display = "block"; 
-
         window['div_count'] = 0;
         window['chat_bot'] = window['gradioEl'].querySelectorAll('#chat_bot')[0];
         window['chat_bot1'] = window['gradioEl'].querySelectorAll('#chat_bot1')[0];   
@@ -183,10 +172,10 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
                          outputs=prompt_input
                         )
         
-        api = gr.State(value=get_api())    
+          
         submit_btn.click(fn=chat, 
-                             inputs=[api,prompt_input, chat_history], 
-                             outputs=[api,chatbot, chat_history],
+                             inputs=[prompt_input, chat_history], 
+                             outputs=[chatbot, chat_history],
                             )
     gr.HTML('''
          <p>Note: Please be aware that audio records from iOS devices will not be decoded as expected by Gradio. For the best experience, record your voice from a computer instead of your smartphone ;)</p>
@@ -196,5 +185,6 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
                     </p>
         </div>
         ''')
+    gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=RamAnanth1.chatGPT_voice)")
 
 demo.launch(debug = True)