Update app.py

app.py

@@ -10,16 +10,13 @@ whisper = gr.Interface.load(name="spaces/sanchit-gandhi/whisper-large-v2")
 #chatgpt = gr.Blocks.load(name="spaces/fffiloni/whisper-to-chatGPT")
 import os
 import json
-import requests
-import requests
-from urllib3.exceptions import InsecureRequestWarning
-# Suppress the warnings from urllib3
-requests.packages.urllib3.disable_warnings(category=InsecureRequestWarning)
-#session_token = os.environ.get('SessionToken')
-api_endpoint = os.environ.get('API_EndPoint')
+session_token = os.environ.get('SessionToken')
+#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
 
 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)"
@@ -28,20 +25,13 @@ coquiTTS = CoquiTTS()
 chat_id = {'conversation_id': None, 'parent_id': None}
 headers = {'Authorization': 'yusin'}
 
-import socket
-
-def detect_port(ip,port):
-    """检测ip上的端口是否开放
-    """
-    s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
-    try:
-        s.connect((ip,int(port)))
-        s.shutdown(2)
-        print('{0} is open'.format(port))
-        return True
-    except:
-        print('{0} is close'.format(port))
-        return False
+async def chat_gpt_ask(prompt):
+    chat_gpt = PyGPT(session_token)
+    await chat_gpt.connect()
+    await chat_gpt.wait_for_ready()
+    answer = await chat_gpt.ask(prompt)
+    print(answer)
+    await chat_gpt.disconnect()
 
 # ChatGPT
 def chat_hf(audio, custom_token, language):
@@ -53,7 +43,7 @@ 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)
-            print(whisper_text)
+            gpt_response = asyncio.run(chat_gpt_ask(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: 
@@ -66,8 +56,8 @@ def chat_hf(audio, custom_token, language):
             #chat_id['parent_id'] = response.json()["response_id"]
             #chat_id['conversation_id'] = response.json()["conversation_id"]
             #gpt_response = response.json()["content"]
-            response = requests.get('https://api.pawan.krd/chat/gpt?text=' + whisper_text + '&cache=false', verify=False, timeout=5)
-            print(response.text)
+            #response = requests.get('https://api.pawan.krd/chat/gpt?text=' + whisper_text + '&cache=false', verify=False, timeout=5)
+            #print(response.text)
         
         #whisper_text = translate(audio)
         #api = ChatGPT(session_token)