Update app.py

app.py

@@ -12,7 +12,6 @@ RateLimitPerModel = {
     "gpt-4-32k": 1000
 }
 
-
 def get_available_models(api_key):
     headers = {"Authorization": f"Bearer {api_key}"}
     url = f"{BaseApi}/engines"
@@ -44,7 +43,7 @@ def check_key_status(api_key, model):
 
         error_type = response_data.get("error", {}).get("type", "")
         if error_type == "insufficient_quota" and model in {"gpt-4", "gpt-4-32k"}:
-            return f"Error: The key for {model} is out of quota but has GPT4."
+            return f"Error: The key for {model} is out of quota, but has gpt4"
         elif error_type in ["insufficient_quota", "billing_not_active", "access_terminated"]:
             return f"Error: The key for {model} is either out of quota, inactive, or access is terminated."
 
@@ -52,7 +51,7 @@ def check_key_status(api_key, model):
         if (response.status_code == 400 and error_type == "invalid_request_error") or ratelimited:
             ratelimit = response.headers.get("x-ratelimit-limit-requests", "0")
             org = response.headers.get("openai-organization", "user-xyz")
-            is_trial_key = "Trial Key" if int(ratelimit) < RateLimitPerModel.get(model, 0) else "Paid key"
+            is_trial_key = "Trial Key" if int(ratelimit) < RateLimitPerModel.get(model, 0) else "Paid Key"
             return f"Key for {model} is working. Ratelimit: {ratelimit}, Organization: {org}, Key Type: {is_trial_key}"
 
     except HTTPError as http_err:
@@ -60,7 +59,7 @@ def check_key_status(api_key, model):
     except Exception as e:
         return f"Error occurred: {e}"
 
-def check_model(api_key):
+def check_models(api_key):
     available_models = get_available_models(api_key)
     if not available_models:
         return "Error occurred: Unable to retrieve available models. Please check your API key."
@@ -68,14 +67,18 @@ def check_model(api_key):
     model_info = "\n".join([check_key_status(api_key, model) for model in available_models])
     return model_info
 
-# Define Gradio interface
+# Define Gradio interface with a button to trigger model checking
+def trigger_model_check(api_key):
+    return check_models(api_key)
+
 iface = gr.Interface(
-    fn=check_model,
+    fn=trigger_model_check,
     inputs=gr.inputs.Textbox(placeholder="Enter your OpenAI API key", type="text"),
     outputs=gr.outputs.Textbox(),
     live=True,
     title="OKC",
-    description="Enter your OpenAI API key to check the available models.",
+    description="Enter your OpenAI API key and click 'Check Models' to see the available models and key status.",
+    allow_flagging=False,  # Disable flagging to prevent unnecessary reporting
 )
 
 iface.launch()