Update app.py

app.py

@@ -4,12 +4,12 @@ from huggingface_hub import HfApi
 
 repo_id = "duypro247/TestScrape"
 api = HfApi()
-btn=gr.Button(label="Restart Space")  # new button
 
-def restart_space():
-    api.restart_space(repo_id=repo_id)
-
-btn.click(restart_space)
+def restart_space(restart_button):
+    if restart_button:
+        api.restart_space(repo_id=repo_id)
+        return "Space restarted!"
+    return "Space not restarted."
 
 username_1=os.getenv('username_1')
 username_2=os.getenv('username_2')
@@ -18,4 +18,11 @@ password_2=os.getenv('password_2')
 
 hf_token=os.getenv('hf_token')
 iface = gr.load("duypro247/TestScrape",src="spaces",hf_token=hf_token)
+
+iface = gr.Interface(
+    fn=restart_space, 
+    inputs=gr.inputs.Button(label="Restart Space"), 
+    outputs="text"
+)
+
 iface.launch(auth=[(username_1,password_1 ), (username_2, password_2)])