Update app.py

app.py

@@ -62,18 +62,28 @@ function() {
   globalThis.getStorage = (key, value)=>{
     return JSON.parse(localStorage.getItem(key));
   }
-  const local_data = getStorage('local_data');
+  var local_data = getStorage('local_data');
   const history = [];
-  console.log(1);
   if(local_data) {
-    console.log(2);
     console.log(local_data);
     local_data[0].pingpongs.forEach(element => 
       history.push([element.ping, element.pong])
     );
   }
+  else {
+    local_data = [
+      {'ctx': '', 'pingpongs':[]},
+      {'ctx': '', 'pingpongs':[]},
+      {'ctx': '', 'pingpongs':[]},
+      {'ctx': '', 'pingpongs':[]},
+      {'ctx': '', 'pingpongs':[]},
+    ];
+    local_data[0].pingpongs.forEach(element => 
+      history.push([element.ping, element.pong])
+    );    
+  }
   console.log(history);
-  return [local_data, history];
+  return [history, local_data];
 }
 """
 
@@ -106,42 +116,9 @@ def set_chatbot(btn, ld):
   ]
   return res[choice].build_uis(), choice
 
-def initialize(local_data):
-  if local_data is None:
-    list_conv = [
-        GradioAlpacaChatPPManager()
-        for _ in range(5)
-    ]
-    local_data = str(list_conv)
-
-  return local_data
-
-def initialize_t(ld):
-  print(ld)
-
-  if ld == {} or id is None:
-    list_conv = [
-        GradioAlpacaChatPPManager()
-        for _ in range(5)
-    ]
-    local_data = str(list_conv)
-  else:
-    list_conv = [
-        GradioAlpacaChatPPManager.from_json(json.dumps(ppm))
-        for ppm in ld
-    ]
-    local_data = str(list_conv)
-
-  return list_conv[0].build_uis(), local_data
-
 with gr.Blocks(css=STYLE, elem_id='container-col') as block:
   idx = gr.State(0)
-  ttt = gr.Textbox(visible=False)
-  local_data = gr.JSON(
-      {},
-      label="Local Storage",
-      visible=False
-  )
+  local_data = gr.JSON({},visible=False)
   
   with gr.Row():
     with gr.Column(scale=1, min_width=180):
@@ -162,61 +139,15 @@ with gr.Blocks(css=STYLE, elem_id='container-col') as block:
           elem_id="prompt-txt"
       )
 
-  first.click(
-    initialize,
-    local_data,
-    local_data
-  ).then(
-    set_chatbot,
-    [first, local_data],
-    [chatbot, idx]
-  )
-
-  second.click(
-    initialize,
-    local_data,
-    local_data
-  ).then(
-    set_chatbot,
-    [second, local_data],
-    [chatbot, idx]
-  )
-
-  third.click(
-    initialize,
-    local_data,
-    local_data
-  ).then(
-    set_chatbot,
-    [third, local_data],
-    [chatbot, idx]
-  )
+  btns = [first, second, third, fourth, fifth]
+  for btn in btns:
+    btn.click(
+      set_chatbot,
+      [btn, local_data],
+      [chatbot, idx]
+    )
   
-  fourth.click(
-    initialize,
-    local_data,
-    local_data
-  ).then(
-    set_chatbot,
-    [fourth, local_data],
-    [chatbot, idx]
-  )   
-
-  fifth.click(
-    initialize,
-    local_data,
-    local_data
-  ).then(
-    set_chatbot,
-    [fifth, local_data],
-    [chatbot, idx]
-  )     
-
   instruction_txtbox.submit(
-    initialize,
-    local_data,
-    local_data
-  ).then(        
     add_pingpong,
     [idx, local_data, instruction_txtbox],
     [chatbot, local_data]
@@ -228,7 +159,7 @@ with gr.Blocks(css=STYLE, elem_id='container-col') as block:
   block.load(
       None,
       inputs=None,
-      outputs=[local_data, chatbot],
+      outputs=[chatbot, local_data],
       _js=get_local_storage,
   )