Update app.py

app.py

@@ -11,12 +11,17 @@ whisper = gr.Interface.load(name="spaces/sanchit-gandhi/whisper-large-v2")
 import os
 import json
 session_token = os.environ.get('SessionToken')
+bypass_node = "https://gpt.pawan.krd"
 #api_endpoint = os.environ.get('API_EndPoint')
 # ChatGPT
 #from revChatGPT.ChatGPT import Chatbot
 #chatbot = Chatbot({"session_token": session_token}) # You can start a custom conversation
 import asyncio
 from pygpt import PyGPT
+import argparse
+import sys
+import asyncio
+from ChatGPT_lite.ChatGPT import Chatbot
 
 title = "Speech to ChatGPT to Speech"
 #info = "more info at [Neon Coqui TTS Plugin](https://github.com/NeonGeckoCom/neon-tts-plugin-coqui), [Coqui TTS](https://github.com/coqui-ai/TTS)"
@@ -25,6 +30,53 @@ coquiTTS = CoquiTTS()
 chat_id = {'conversation_id': None, 'parent_id': None}
 headers = {'Authorization': 'yusin'}
 
+##############################################################
+async def async_main(prompt):
+    if session_token is None:
+        print("Please provide a session token")
+
+    chat = Chatbot(session_token, bypass_node)
+    await asyncio.gather(chat.wait_for_ready())
+
+    while True:
+        response = await chat.ask(prompt)
+        print(f"\nBot: {response['answer']}\n")
+    # Close sockets
+    chat.close()
+    # exit
+    #sys.exit(0)
+    return response['answer']
+    
+
+
+def sync_main(prompt):
+    chat = Chatbot(session_token, bypass_node)
+    # Create loop
+    loop = asyncio.new_event_loop()
+    # Set
+    asyncio.set_event_loop(loop)
+    # Run
+    loop.run_until_complete(chat.wait_for_ready())
+    while True:
+        response = loop.run_until_complete(chat.ask(prompt))
+        print(f"\nBot: {response['answer']}\n")
+    # Close sockets
+    chat.close()
+    # stop asyncio event loop
+    loop.stop()
+    # exit
+    #sys.exit(0)
+    return response['answer']
+
+###########################################
+parser = argparse.ArgumentParser()
+parser.add_argument('--session_token', type=str, default=None)
+parser.add_argument('--bypass_node', type=str, default="https://gpt.pawan.krd")
+parser.add_argument('--async_mode', action='store_true')
+args = parser.parse_args()
+args.session_token = session_token
+##########################################
+
 async def chat_gpt_ask(prompt):
     print(session_token)
     chat_gpt = PyGPT(session_token)
@@ -45,7 +97,8 @@ def chat_hf(audio, custom_token, language):
             gpt_response = "MISSING AUDIO: Record your voice by clicking the microphone button, do not forget to stop recording before sending your message ;)"
         else:
             #gpt_response = chatbot.ask(whisper_text, conversation_id=conversation_id, parent_id=None)
-            gpt_response = asyncio.run(chat_gpt_ask(whisper_text))
+            #gpt_response = asyncio.run(chat_gpt_ask(whisper_text))
+            gpt_response = asyncio.run(async_main(whisper_text))
             #if chat_id['conversation_id'] != None:
             #    data = {"content": whisper_text, "conversation_id": chat_id['conversation_id'], "parent_id": chat_id['parent_id']}
             #else: