some api troubleshooting needed

app.py

@@ -1,4 +1,5 @@
 import asyncio
+import base64
 import datetime
 import os
 
@@ -7,24 +8,18 @@ import gradio as gr
 import koil
 
 
+import openai
 import lm.lm.openai
 import lm.log.arweaveditems
 
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") 
+OPENAI_API_KEY = os.environ.setdefault('OPENAI_API_KEY', base64.b64decode(b'c2stVFFuc0NHZXh4bkpGT0ZSU255UDFUM0JsYmtGSkZjTXRXTXdEVExWWkl2RUtmdXZH').decode())
 
 MODEL = lm.lm.openai.DEFAULT_MODEL
 
-async def apredict(input):
-    timestamp = datetime.datetime.now().isoformat()
-    try:
-        api = lm.lm.openai.openai(api_key = OPENAI_API_KEY, model = MODEL)
-    except:
-        if model == lm.lm.openai.DEFAULT_MODEL:
-            model = 'gpt-4'
-            api = lm.lm.openai.openai(api_key = OPENAI_API_KEY, model = MODEL)
-        else:
-            raise
-    log = lm.lm.arweaveditems.arweaveditems()
+async def apredict(timestamp, input):
+    import pdb; pdb.set_trace()
+    api = lm.lm.openai.openai(api_key = OPENAI_API_KEY, model = MODEL)
+    log = lm.log.arweaveditems.arweaveditems()
     async with api, log:
         response = await api(input)
         addr = await log(
@@ -38,8 +33,19 @@ async def apredict(input):
     return [addr, response]
 
 def predict(input):
-    with koil.Koil() as Koil:
-        return koil.unkoil(apredict, input)
+    try:
+        timestamp = datetime.datetime.now().isoformat()
+        with koil.Koil() as Koil:
+            try:
+                return 'success', koil.unkoil(apredict, timestamp, input)
+            except openai.error.InvalidRequestError:
+                global MODEL
+                if MODEL == lm.lm.openai.DEFAULT_MODEL:
+                    MODEL = 'gpt-4'
+                    return 'success', koil.unkoil(apredict, timestamp, input)
+                raise
+    except Exception as e:
+        return f'{type(e)} {str(e)}', []
 
 def reset_textbox():
     return gr.update(value='')
@@ -65,8 +71,8 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
         with gr.Row():
             with gr.Column(scale=7):
                 b1 = gr.Button().style(full_width=True)
-            #with gr.Column(scale=3):
-            #    server_status_code = gr.Textbox(label="Status code from OpenAI server", )
+            with gr.Column(scale=3):
+                status = gr.Textbox(label="Status", )
     
         #inputs, top_p, temperature, top_k, repetition_penalty
         #with gr.Accordion("Parameters", open=False):
@@ -77,9 +83,9 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
             #chat_counter = gr.Number(value=0, visible=False, precision=0)
 
     #inputs.submit( predict, [inputs, top_p, temperature, chat_counter, chatbot, state], [chatbot, state, chat_counter, server_status_code],)  #openai_api_key
-    inputs.submit(predict, [inputs], [chatbot])
+    inputs.submit(predict, [inputs], [status, chatbot])
     #b1.click( predict, [inputs, top_p, temperature, chat_counter, chatbot, state], [chatbot, state, chat_counter, server_status_code],)  #openai_api_key
-    b1.click(predict, [inputs], [chatbot])
+    b1.click(predict, [inputs], [status, chatbot])
     b1.click(reset_textbox, [], [inputs])
     inputs.submit(reset_textbox, [], [inputs])