Update app.py

app.py

@@ -1,35 +1,18 @@
-from pyChatGPT import ChatGPT
+import openai
 import gradio as gr
 import os, sys, json
 from loguru import logger
 import paddlehub as hub
 import random
 
-    
-language_translation_model = hub.Module(directory=f'./baidu_translate')
-def getTextTrans(text, source='zh', target='en'):
-    def is_chinese(string):
-        for ch in string:
-            if u'\u4e00' <= ch <= u'\u9fff':
-                return True
-        return False
-        
-    if not is_chinese(text) and target == 'en': 
-        return text
-        
-    try:
-        text_translation = language_translation_model.translate(text, source, target)
-        return text_translation
-    except Exception as e:
-        return text 
-
+openai.api_key = os.environ.get('SessionToken')
 session_token = os.environ.get('SessionToken')      
 logger.info(f"session_token_: {session_token}")
 
 def get_api():
     api = None
     try:
-      api = ChatGPT(session_token)
+      api = openai
       # api.refresh_auth()
     except Exception as e:
       print(f'get_api_error:', e)
@@ -41,14 +24,10 @@ def get_response_from_chatgpt(api, text):
     if api is None:
         return "Openai said: I'm too tired. Let me lie down for a few days. If you like, you can visit my home."
     try:
-      resp = api.send_message(text)    
-      # api.refresh_auth()
-      # api.reset_conversation() 
-      response = resp['message']
-      conversation_id = resp['conversation_id']
-      parent_id = resp['parent_id']
-      # logger.info(f"response_: {response}")
-      logger.info(f"conversation_id_: [{conversation_id}] / parent_id: [{parent_id}]")  
+      response = openai.Completion.create(engine='text-davinci-003', prompt=text, 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]
+      logger.info(f"conversation_id_: [{esponse}]")  
     except:
       response = "Openai said: I'm so tired. Let me lie down for a few days. If you like, you can visit my home."
     return response
@@ -255,8 +234,6 @@ start_work = """async() => {
 
 space_ids = {
             "spaces/stabilityai/stable-diffusion":"Stable Diffusion 2.1",
-            # "spaces/runwayml/stable-diffusion-v1-5":"Stable Diffusion 1.5",
-            # "spaces/stabilityai/stable-diffusion-1":"Stable Diffusion 1.0",
             }
 
 tab_actions = []