Update app.py

app.py

@@ -159,6 +159,40 @@ def delete_endpoint(
     else:
         return f"something went wrong {response.status_code} = {response.text}"
 
+def update_endpoint(
+    hf_token_input,
+    endpoint_name_input,
+    min_node_selector,
+    max_node_selector,
+    instance_type,
+):
+    payload ={
+            "compute": {
+                "instanceSize": instance_type.split("·")[0].split("[")[1].split("]")[0],
+                "instanceType": instance_type.split("·")[-1].strip(),
+                "scaling": {
+                    "maxReplica": max_node_selector,
+                    "minReplica": min_node_selector
+                    }
+                }}
+
+    response = requests.put(
+            f"https://api.endpoints.huggingface.cloud/endpoint/{endpoint_name_input}",
+            headers = {
+            "Authorization": f"Bearer {hf_token_input.strip()}",
+            "Content-Type": "application/json",
+            },
+            data=payload,
+        )
+    if response.status_code == 401:
+        return "Invalid token"
+    elif response.status_code == 404:
+        return f"Error: {response.text}"
+    elif response.status_code == 202:
+        return f"Endpoint {endpoint_name_input} updated successfully."
+    else:
+        return f"something went wrong {response.status_code} = {response.text}"
+
 with gr.Blocks() as demo2:
     gr.Markdown("""
     #### Your 🤗 Access Token <span style="color:red;">(Required)</span>
@@ -321,6 +355,94 @@ with gr.Blocks() as demo2:
             security_selector],
         outputs=status_txt)    
 
+    # Update Endpoint
+    gr.Markdown("""
+    ### <br><center style="color:green">(Update Endpoint)</center>
+    """)
+    with gr.Row():
+        gr.Markdown("""    
+        #### Cloud Provider
+        """)
+
+        gr.Markdown("""    
+        #### Cloud Region
+        """)      
+
+    with gr.Row():
+        update_provider_selector = gr.Dropdown(
+            choices=providers,
+            interactive=True,
+            show_label=False,
+        )
+        
+        update_region_selector = gr.Dropdown(
+            [],
+            value="",
+            interactive=True,
+            show_label=False,
+        )
+
+    with gr.Row():
+        gr.Markdown("""
+        #### Endpoint Name
+        """)
+        gr.Markdown("""
+        #### Instance Type
+        """)
+
+    with gr.Row():
+        update_endpoint_name_input = gr.Textbox(
+            show_label=False
+        )
+        update_compute_selector = gr.Dropdown(
+            [],
+            value="",
+            interactive=True,
+            show_label=False,
+        )
+    with gr.Row():
+        gr.Markdown("""
+        #### Min Number of Nodes
+        """)
+        gr.Markdown("""
+        #### Max Number of Nodes
+        """)
+    with gr.Row():
+        update_min_node_input = gr.Number(
+            value=1,
+            interactive=True,
+            show_label=False,
+        )
+        
+        update_max_node_input = gr.Number(
+            value=1,
+            interactive=True,
+            show_label=False,
+        )
+    update_button = gr.Button(
+        value="Update",
+    )
+
+    update_status_txt = gr.Textbox(
+        value="status",
+        interactive=False
+    )
+
+    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(
+        update_endpoint,
+        inputs=[
+            hf_token_input,
+            update_endpoint_name_input,
+            update_min_node_input,
+            update_max_node_input,
+            update_compute_selector
+        ],
+        outputs=update_status_txt
+    )
+    
     # Delete Endpoint
     gr.Markdown("""
     ### <br><center style="color:green">(Delete 🗑️ Endpoint)</center>