Update app.py

app.py

@@ -7,20 +7,7 @@ import random
 openai.api_key = os.environ.get('SessionToken')     
 logger.info(f"session_token_: {openai.api_key}")
 
-def get_api():
-    api = None
-    try:
-      api = openai
-      # api.refresh_auth()
-    except Exception as e:
-      print(f'get_api_error:', e)
-      api = None
-    return api
-
-  
-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."
+def get_response_from_chatgpt(text):
     try:
       response = openai.Completion.create(engine='text-davinci-003', prompt=text, max_tokens=100)
       response_str = response["choices"][0]["text"].replace("\n", "")
@@ -246,21 +233,19 @@ for space_id in space_ids.keys():
     except Exception as e:
         logger.info(f"load_fail__{space_id}_{e}")
         
-def chat(api, input0, input1, chat_radio, chat_history):
+def chat(input0, input1, chat_radio, chat_history):
     out_chat = []
     if chat_history != '':
         out_chat = json.loads(chat_history)
     logger.info(f"out_chat_: {len(out_chat)} / {chat_radio}")
     if chat_radio == "Talk to chatGPT":
-        response = get_response_from_chatgpt(api, input0)
-        # response = get_response_from_microsoft(input0)
-        # response = get_response_from_skywork(input0)
+        response = get_response_from_chatgpt(input0)
         out_chat.append((input0, response))
         chat_history = json.dumps(out_chat)
-        return api, out_chat, input1, chat_history
+        return out_chat, input1, chat_history
     else:
         prompt_en = getTextTrans(input0, source='zh', target='en') + f',{random.randint(0,sys.maxsize)}'
-        return api, out_chat, prompt_en, chat_history
+        return out_chat, prompt_en, chat_history
         
 with gr.Blocks(title='Talk to chatGPT') as demo:
     gr.HTML("<p>You can duplicating this space and use your own session token: <a style='display:inline-block' href='https://huggingface.co/spaces/yizhangliu/chatGPT?duplicate=true'><img src='https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14' alt='Duplicate Space'></a></p>")
@@ -293,10 +278,10 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
                         rounded=(True, True, True, True),
                         width=100
                     )
-            #api = gr.State(value=get_api())
+
             submit_btn.click(fn=chat, 
-                             inputs=[api, prompt_input0, prompt_input1, chat_radio, chat_history], 
-                             outputs=[api, chatbot, prompt_input1, chat_history],
+                             inputs=[prompt_input0, prompt_input1, chat_radio, chat_history], 
+                             outputs=[chatbot, prompt_input1, chat_history],
                             )
         with gr.Row(elem_id='tab_img', visible=False).style(height=5):
            tab_img = gr.TabbedInterface(tab_actions, tab_titles)