Updated Space.

app.py

@@ -221,10 +221,10 @@ def safe_eval(expression):
             raise NameError(f"Use of {name} is not allowed")
     return eval(code, {"__builtins__": {}}, allowed_names)
 
-def get_api_key(call="api_key"):
-    if call == "api_key":
+def get_api_key(call='api_key'):
+    if call == 'api_key':
         key = api_key
-    elif call == "oai_api_key":
+    elif call == 'oai_api_key':
         key = oai_api_key
     else:
         key = api_key
@@ -249,7 +249,7 @@ def moderate(messages):
             f"{base_url}/moderations",
             headers={
                 "Content-Type": "application/json",
-                "Authorization": f"Bearer {get_api_key(call="api_key")}"
+                "Authorization": f"Bearer {get_api_key(call='api_key')}"
             },
             json={"input": encodeChat(messages)}
         )
@@ -262,7 +262,7 @@ def moderate(messages):
                 "https://api.openai.com/v1/moderations",
                 headers={
                     "Content-Type": "application/json",
-                    "Authorization": f"Bearer {get_api_key(call="oai_api_key")}"
+                    "Authorization": f"Bearer {get_api_key(call='oai_api_key')}"
                 },
                 json={"input": encodeChat(messages)}
             )
@@ -284,7 +284,7 @@ def moderate(messages):
 async def streamChat(params):
     async with aiohttp.ClientSession() as session:
         try:
-            async with session.post(f"{base_url}/chat/completions", headers={"Authorization": f"Bearer {get_api_key(call="api_key")}", "Content-Type": "application/json"}, json=params) as r:
+            async with session.post(f"{base_url}/chat/completions", headers={"Authorization": f"Bearer {get_api_key(call='api_key')}", "Content-Type": "application/json"}, json=params) as r:
                 r.raise_for_status()
                 async for line in r.content:
                     if line:
@@ -300,7 +300,7 @@ async def streamChat(params):
                             continue
         except aiohttp.ClientError:
             try:
-                async with session.post("https://api.openai.com/v1/chat/completions", headers={"Authorization": f"Bearer {get_api_key(call="oai_api_key")}", "Content-Type": "application/json"}, json=params) as r:
+                async with session.post("https://api.openai.com/v1/chat/completions", headers={"Authorization": f"Bearer {get_api_key(call='oai_api_key')}", "Content-Type": "application/json"}, json=params) as r:
                     r.raise_for_status()
                     async for line in r.content:
                         if line:
@@ -323,7 +323,7 @@ def imagine(prompt):
             f"{base_url}/images/generations",
             headers={
                 "Content-Type": "application/json",
-                "Authorization": f"Bearer {get_api_key(call="api_key")}"
+                "Authorization": f"Bearer {get_api_key(call='api_key')}"
             },
             json={
                 "model": "dall-e-3",
@@ -340,7 +340,7 @@ def imagine(prompt):
                 "https://api.openai.com/v1/images/generations",
                 headers={
                     "Content-Type": "application/json",
-                    "Authorization": f"Bearer {get_api_key(call="oai_api_key")}"
+                    "Authorization": f"Bearer {get_api_key(call='oai_api_key')}"
                 },
                 json={
                 	"model": "dall-e-3",