Update app.py

app.py

@@ -149,7 +149,7 @@ def submit(
     elif response.status_code == 202:
         return f"Endpoint {endpoint_name_input} created successfully on {provider_selector.lower()} using {repository_selector.lower()}@{revision_selector}. \n Please check out the progress at https://ui.endpoints.huggingface.co/endpoints."
     else:
-        return f"Something went wrong!, StatusCode:{response.status_code}, Error: {response.text}"
+        return f"Something went wrong! - StatusCode:{response.status_code} - Error: {response.text}"
 
 def delete_endpoint(
     hf_token_input,
@@ -166,10 +166,10 @@ def delete_endpoint(
         return "Invalid token"
     elif response.status_code == 404:
         return f"Error: {response.text}"
-    elif response.status_code == 202:
+    elif response.status_code == 202 or response.status_code == 200:
         return f"Endpoint {endpoint_name_input} deleted successfully."
     else:
-        return f"Something went wrong!, StatusCode:{response.status_code}, Error: {response.text}"
+        return f"Something went wrong! - StatusCode:{response.status_code} - Error: {response.text}"
 
 def get_all_endpoints(
     hf_token_input,
@@ -184,7 +184,9 @@ def get_all_endpoints(
     if response.status_code == 401:
         if method == "info":
             return gr.DataFrame.update(
-                value=[["Invalid token", "Invalid token", "Invalid token", "Invalid token", "Invalid token", "Invalid token", "Invalid token", "Invalid token", "Invalid token"]],
+                col_count = 1
+                headers = ["{ERROR}"]
+                value=["Invalid Token OR No Endpoint Found !"],
         )
         else:
             return gr.Dropdown.update(
@@ -208,6 +210,7 @@ def get_all_endpoints(
                 }, ignore_index=True)
             endpoints_df.columns = ["Endpoint Name", "Model Name @ Revision", "Provider", "Instance Type", "Status", "Min Replica", "Max Replica", "Created At", "Updated At"]
             return gr.DataFrame.update(
+                col_count = (9, "fixed")
                 value=endpoints_df,
             )
         else:
@@ -245,10 +248,10 @@ def update_endpoint(
         return "Invalid token"
     elif response.status_code == 404:
         return f"Error: {response.text}"
-    elif response.status_code == 202:
+    elif response.status_code == 202 or response.status_code == 200:
         return f"Endpoint {endpoint_name_input} updated successfully."
     else:
-        return f"Something went wrong!, StatusCode:{response.status_code}, Error: {response.text}"
+        return f"Something went wrong! - StatusCode:{response.status_code} - Error: {response.text}"
 
 def get_endpoint_logs(
     hf_token_input,