Update app.py

app.py

@@ -7,34 +7,41 @@ default_lang = "en"
 #import whisper
 #whisper_model = whisper.load_model("small")
 whisper = gr.Interface.load(name="spaces/sanchit-gandhi/whisper-large-v2")
-chatgpt = gr.Blocks.load(name="spaces/fffiloni/whisper-to-chatGPT")
+#chatgpt = gr.Blocks.load(name="spaces/fffiloni/whisper-to-chatGPT")
 import os
-session_token = os.environ.get('SessionToken')
-conversation_id = os.environ.get('conversation_id')
+import requests
+#session_token = os.environ.get('SessionToken')
 # ChatGPT
-from revChatGPT.ChatGPT import Chatbot
+#from revChatGPT.ChatGPT import Chatbot
 #chatbot = Chatbot({"session_token": session_token}) # You can start a custom conversation
 
 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)"
 #badge = "https://visitor-badge-reloaded.herokuapp.com/badge?page_id=neongeckocom.neon-tts-plugin-coqui"
-
 coquiTTS = CoquiTTS()
-
-
+chat_id = {'conversation_id': None, 'parent_id': None}
+proxy = {'https': 'http://10.236.250.208'}
+headers = {'Authorization': 'yusin'}
         
 # ChatGPT
 def chat_hf(audio, custom_token, language):
-    output = chatgpt(audio, "transcribe", fn_index=0)
-    whisper_text, gpt_response = output[0], output[1]
-    '''
+    #output = chatgpt(audio, "transcribe", fn_index=0)
+    #whisper_text, gpt_response = output[0], output[1]
     try:
         whisper_text = translate(audio)
         if whisper_text == "ERROR: You have to either use the microphone or upload an audio file":
             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 = chatbot.ask(whisper_text, conversation_id=conversation_id, parent_id=None)
+            if chat_id['conversation_id'] != None:
+                data = {"content": whisper_text, "conversation_id": chat_id['conversation_id'], "parent_id": chat_id['parent_id']}
+            else: 
+                data = {"content": whisper_text}
+            response = requests.post('http://192.168.163.215:7788/api/ask', headers=headers, json=data, proxies=proxy)
+            chat_id['parent_id'] = response.json()["response_id"]
+            chat_id['conversation_id'] = response.json()["conversation_id"]
+            gpt_response = response.json()["content"]
+        
         #whisper_text = translate(audio)
         #api = ChatGPT(session_token) 
         #resp = api.send_message(whisper_text)
@@ -53,7 +60,7 @@ def chat_hf(audio, custom_token, language):
         #api.refresh_auth()  # refresh the authorization token
         #api.reset_conversation()  # reset the conversation
         #gpt_response = resp['message']
-    '''  
+    
     # to voice
     with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
         coquiTTS.get_tts(gpt_response, fp, speaker = {"language" : language})