updated code

app.py

@@ -59,7 +59,6 @@ def update_compute_options(provider, region):
         "Content-Type": "application/json",
     }
     endpoint_url = f"https://api.endpoints.huggingface.cloud/provider/{provider}/region/{region}/compute"
-    print(endpoint_url)
     response = requests.get(endpoint_url, headers=headers)
 
     for compute in response.json()['items']:
@@ -134,7 +133,6 @@ def submit(
     
 
     payload = json.dumps(payload)
-    print(f"Payload: {payload}")
 
     headers = {
         "Authorization": f"Bearer {hf_token_input.strip()}",
@@ -153,7 +151,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 {response.status_code} = {response.text}"
+        return f"Something went wrong!, StatusCode:{response.status_code}, Error: {response.text}"
 
 def delete_endpoint(
     hf_token_input,
@@ -173,10 +171,11 @@ def delete_endpoint(
     elif response.status_code == 202:
         return f"Endpoint {endpoint_name_input} deleted successfully."
     else:
-        return f"something went wrong {response.status_code} = {response.text}"
+        return f"Something went wrong!, StatusCode:{response.status_code}, Error: {response.text}"
 
 def get_all_endpoints(
     hf_token_input,
+    method,
 ):
     response = requests.get(
             f"https://api.endpoints.huggingface.cloud/endpoint",
@@ -185,23 +184,38 @@ def get_all_endpoints(
                 "Content-Type": "application/json",
             })
     if response.status_code == 401:
-        return "Invalid token"
+        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"]],
+        )
+        else:
+            return gr.Dropdown.update(
+                value="Invalid token or No endpoints found!",
+            )
     elif response.status_code == 200:
         endpoints_json = response.json()
-        print(endpoints_json)
-        endpoints_df = pd.DataFrame(endpoints_json["items"])
-        endpoints_df = endpoints_df[["name", "model", "provider", "compute", "status"]]
-        endpoints_df["model"] = endpoints_df["model"].apply(lambda x: x["repository"] + "@" + x["revision"])
-        endpoints_df["provider"] = endpoints_df["provider"].apply(lambda x: x["vendor"] + "/" + x["region"])
-        endpoints_df["compute"] = endpoints_df["compute"].apply(lambda x: x["instanceType"] + "·" + x["instanceSize"] + " [" + x["accelerator"] + "]")
-        endpoints_df["status"] = endpoints_df["status"].apply(lambda x: x["state"])
-        endpoints_df["minReplica"] = endpoints_df["compute"].apply(lambda x: x["scaling"]["minReplica"])
-        endpoints_df["maxReplica"] = endpoints_df["compute"].apply(lambda x: x["scaling"]["maxReplica"])
-        endpoints_df["createdAt"] = endpoints_df["status"].apply(lambda x: x["createdAt"])
-        endpoints_df["updatedAt"] = endpoints_df["status"].apply(lambda x: x["updatedAt"])
-        endpoints_df = endpoints_df[["name", "model", "provider", "compute", "status", "minReplica", "maxReplica", "createdAt", "updatedAt"]]
-        return gr.Dataframe.update(
-            value=endpoints_df
+        if method == "info":
+            endpoints_df = pd.DataFrame(columns=["name", "model", "provider", "compute", "status", "minReplica", "maxReplica", "createdAt", "updatedAt"])
+            for endpoint in endpoints_json["items"]:
+                endpoints_df = endpoints_df.append({
+                    "name": endpoint["name"],
+                    "model": endpoint["model"]["repository"] + "@" + endpoint["model"]["revision"],
+                    "provider": endpoint["provider"]["vendor"] + "/" + endpoint["provider"]["region"],
+                    "compute": endpoint["compute"]["instanceType"] + "·" + endpoint["compute"]["instanceSize"] + " [" + endpoint["compute"]["accelerator"] + "]",
+                    "status": endpoint["status"]["state"],
+                    "minReplica": endpoint["compute"]["scaling"]["minReplica"],
+                    "maxReplica": endpoint["compute"]["scaling"]["maxReplica"],
+                    "createdAt": endpoint["status"]["createdAt"],
+                    "updatedAt": endpoint["status"]["updatedAt"],
+                }, 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(
+                value=endpoints_df,
+            )
+        else:
+            return gr.Dropdown.update( 
+            choices=[endpoint["name"] for endpoint in endpoints_json["items"]],
+            value=endpoints_json["items"][0]["name"],
         )
 
 def update_endpoint(
@@ -216,11 +230,11 @@ def update_endpoint(
                 "instanceSize": instance_type.split("·")[0].split("[")[1].split("]")[0],
                 "instanceType": instance_type.split("·")[-1].strip(),
                 "scaling": {
-                    "maxReplica": max_node_selector,
-                    "minReplica": min_node_selector
+                    "maxReplica": int(max_node_selector),
+                    "minReplica": int(min_node_selector)
                     }
                 }}
-
+    payload = json.dumps(payload)
     response = requests.put(
             f"https://api.endpoints.huggingface.cloud/endpoint/{endpoint_name_input}",
             headers = {
@@ -236,7 +250,7 @@ def update_endpoint(
     elif response.status_code == 202:
         return f"Endpoint {endpoint_name_input} updated successfully."
     else:
-        return f"something went wrong {response.status_code} = {response.text}"
+        return f"Something went wrong!, StatusCode:{response.status_code}, Error: {response.text}"
 
 with gr.Blocks() as interface:
     gr.Markdown("""
@@ -253,7 +267,7 @@ with gr.Blocks() as interface:
         ### All Deployed Endpoints
         """)
         endpoints_table = gr.Dataframe(
-            headers=["Endpoint Name", "Revision", "Provider", "Instance Type", "Status", "Min Replica", "Max Replica", "Created At", "Updated At"],
+            headers=["Endpoint Name", "Model Name", "Provider", "Instance Type", "Status", "Min Replica", "Max Replica", "Created At", "Updated At"],
             col_count=(9, "fixed"),
         )
         endpoint_info_button = gr.Button(value="Get Info")
@@ -398,7 +412,7 @@ with gr.Blocks() as interface:
         gr.Markdown("""
         ### <br><center style="color:green">(Update 🔁 Endpoint)</center>
         """)
-
+        update_endpoint_info_button = gr.Button(value="Load Endpoints 🔃")
         with gr.Row():
             gr.Markdown("""    
             #### Cloud Provider
@@ -432,7 +446,9 @@ with gr.Blocks() as interface:
             """)
 
         with gr.Row():
-            update_endpoint_name_input = gr.Textbox(
+            update_endpoint_name_input = gr.Dropdown(
+                [],
+                value="",
                 show_label=False
             )
             update_compute_selector = gr.Dropdown(
@@ -534,7 +550,7 @@ with gr.Blocks() as interface:
     # Info Tab Events
     endpoint_info_button.click(
             get_all_endpoints, 
-            inputs=hf_token_input, 
+            inputs=[hf_token_input, gr.TextArea(value="info", interactive=False, visible=False)], 
             outputs=endpoints_table
         )
 
@@ -560,6 +576,11 @@ with gr.Blocks() as interface:
         outputs=status_txt)
 
     # Update Tab Events
+    update_endpoint_info_button.click(
+        get_all_endpoints, 
+        inputs=[hf_token_input, gr.TextArea(value="update", interactive=False, visible=False)], 
+        outputs=update_endpoint_name_input
+        )
     update_provider_selector.change(update_regions, inputs=update_provider_selector, outputs=update_region_selector)
     update_region_selector.change(update_compute_options, inputs=[update_provider_selector, update_region_selector], outputs=update_compute_selector)
     update_button.click(

requirements.txt

@@ -1,3 +1,3 @@
-pandas
+pandas==1.3.5
 numpy
 gradio