some fixes

app.py

@@ -1,23 +1,25 @@
 import asyncio
 import datetime
+import os
 
 import gradio as gr
 
 import koil
 
+
 from lm.lm.openai import openai
 from lm.log.arweaveditems import arweaveditems
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") 
 
-@koil.unkoil
-async def predict(input):
+async def apredict(input):
     timestamp = datetime.datetime.now().isoformat()
     try:
         api = openai(api_key = OPENAI_API_KEY)
     except:
         api = openai(api_key = OPENAI_API_KEY, model = 'gpt-4')
-    async with api as api, arweaveditems() as log:
+    log = arweaveditems()
+    async with api, log:
         response = await api(input)
         addr = await log(
                 timestamp = timestamp,
@@ -29,6 +31,10 @@ async def predict(input):
     print(addr)
     return [addr, response]
 
+def predict(input):
+    with koil.Koil() as Koil:
+        return koil.unkoil(apredict, input)
+
 def reset_textbox():
     return gr.update(value='')
 
@@ -53,8 +59,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):
+            #    server_status_code = gr.Textbox(label="Status code from OpenAI server", )
     
         #inputs, top_p, temperature, top_k, repetition_penalty
         #with gr.Accordion("Parameters", open=False):