Update app.py

app.py

@@ -128,21 +128,22 @@ def move(repo,space,dest,token):
     move_repo(from_id=from_id,to_id=to_id,repo_type='space',token=token)
     print("REPO MOVED")
 
-def add_save(space,save_list):
-    save_list=[]if not save_list else save_list
+def add_save(space,save_state):
+    save_state=[]if not save_state else save_state
     this = True
     for ea in save_list:
         if space == ea:
             this = False
     if this:
-        save_list.append(space)
-    return gr.update(choices=[f for f in save_list],interactive=True)
+        #save_list.append(space)
+        save_state.append(space)
+    return save_state,gr.update(choices=[f for f in save_state],interactive=True)
     
-def rem_save(space,save_list):
-    for ea in save_list:
+def rem_save(space,save_state):
+    for ea in save_state:
         if not space == ea:
-            save_list.append(space)
-    return gr.update(choices=[f for f in save_list],interactive=True)
+            save_state.append(space)
+    return save_state, gr.update(choices=[f for f in save_state],interactive=True)
     
 with gr.Blocks(css=css) as build:
     save_state=gr.State()
@@ -186,8 +187,8 @@ with gr.Blocks(css=css) as build:
                             save_btn=gr.Button("Protect")                            
                             rem_btn=gr.Button("Remove")                            
 
-    save_btn.click(add_save,[space_radio,save_list],save_list).then(show_s,[r_name,token,save_list],[space_radio])
-    rem_btn.click(rem_save,[space_radio,save_list],save_list).then(show_s,[r_name,token,save_list],[space_radio])
+    save_btn.click(add_save,[space_radio,save_state],[save_state,save_list]).then(show_s,[r_name,token,save_list],[space_radio])
+    rem_btn.click(rem_save,[space_radio,save_list],[save_state,save_list]).then(show_s,[r_name,token,save_list],[space_radio])
     
     move_btn.click(move,[r_name,space_radio,dest,token]).then(show_s,[r_name,token,save_list],[space_radio])
     s_btn.click(show_s,[r_name,token,save_list],[space_radio])