Update app.py

app.py

@@ -1,33 +1,49 @@
-from pyChatGPT import ChatGPT
 import os
-session_token = os.environ.get('SessionToken') 
+import openai
+import whisper
+from pyChatGPT import ChatGPT
 
-print("------------Aqui----------------", session_token)
+from api_key import API_KEY
+openai.api_key = os.environ.get('SessionToken')
 
-import whisper
 whisper_model = whisper.load_model("small")
- 
+
+conversation = ""
+user_name = "MH"
+bot_name = "bbDemo"
 
 def chat_hf(audio, custom_token):
     try:
         whisper_text = translate(audio)
+
+        # Conversation rout
+        prompt = user_name + ": " + user_input + "\n" + bot_name+ ": "
+        conversation += prompt  # allows for context
+        
         api = ChatGPT(session_token) 
         resp = api.send_message(whisper_text)
+
+        # fetch the response from open AI api
+        response = openai.Completion.create(engine='text-davinci-003', prompt=conversation, max_tokens=100)
+        response_str = response["choices"][0]["text"].replace("\n", "")
+        response_str = response_str.split(user_name + ": ", 1)[0].split(bot_name + ": ", 1)[0]
         
+        conversation += response_str + "\n"
+    
         
-        api.refresh_auth()  # refresh the authorization token
-        api.reset_conversation()  # reset the conversation
-        gpt_response = resp['message']
+        #api.refresh_auth()  # refresh the authorization token
+        #api.reset_conversation()  # reset the conversation
+        gpt_response = response_str #resp['message']
 
     except:
-        whisper_text = translate(audio)
-        api = ChatGPT(custom_token) 
-        resp = api.send_message(whisper_text)
+        #whisper_text = translate(audio)
+        #api = ChatGPT(custom_token) 
+        #resp = api.send_message(whisper_text)
         
         
-        api.refresh_auth()  # refresh the authorization token
-        api.reset_conversation()  # reset the conversation
-        gpt_response = resp['message']
+        #api.refresh_auth()  # refresh the authorization token
+        #api.reset_conversation()  # reset the conversation
+        #gpt_response = resp['message']
     
     return whisper_text, gpt_response
 
@@ -46,11 +62,8 @@ def translate(audio):
     
     _, probs = whisper_model.detect_language(mel)
     
-    transcript_options = whisper.DecodingOptions(task="transcribe", fp16 = False)
-    #translate_options = whisper.DecodingOptions(task="translate", fp16 = False)
-    
+    transcript_options = whisper.DecodingOptions(task="transcribe", fp16 = False)    
     transcription = whisper.decode(whisper_model, mel, transcript_options)
-    #translation = whisper.decode(whisper_model, mel, translate_options)
     
     print("language spoken: " + transcription.language)
     print("transcript: " + transcription.text)