support url reponame input

modules/restart.py

@@ -31,13 +31,17 @@ def self_restart():
 
 
 # UI func
-def test_hf_restart(repo):
+def test_hf_restart(repo: str):
     status = "Success"
     logs = None
     try:
         if not repo:
             raise ValueError("Empty repo!")
 
+        if f"{HF_DOMAIN}/spaces/" in repo:
+            repo = repo.split("/spaces/")[1].split("/")[:2]
+            repo = "/".join(repo)
+
         logs = restart_private_space(repo)
         logs += f"\nSuccessfully restart {HF_DOMAIN}/spaces/{repo}\n"
 
@@ -47,13 +51,17 @@ def test_hf_restart(repo):
     return status, logs
 
 
-def test_ms_restart(repo):
+def test_ms_restart(repo: str):
     status = "Success"
     logs = None
     try:
         if not repo:
             raise ValueError("Empty repo!")
 
+        if f"{MS_DOMAIN}/studios/" in repo:
+            repo = repo.split("/studios/")[1].split("/")[:2]
+            repo = "/".join(repo)
+
         logs = restart_private_studio(repo)
         logs += f"\nSuccessfully restart {MS_DOMAIN}/studios/{repo}\n"