Update app.py

app.py

@@ -1,43 +1,57 @@
+import json
 import gradio as gr
+from pingpong import PingPong
+from pingpong.gradio import GradioAlpacaChatPPManager
 
 get_local_storage = """
-    function() {
-      globalThis.setStorage = (key, value)=>{
-        localStorage.setItem(key, JSON.stringify(value))
-      }
-       globalThis.getStorage = (key, value)=>{
-        return JSON.parse(localStorage.getItem(key))
-      }
-       const text_input =  getStorage('text_input')
-       const dropdown =  getStorage('dropdown')
-       const local_data =  getStorage('local_data')
-       return [text_input, dropdown, local_data];
-      }
-    """
+function() {
+  globalThis.setStorage = (key, value)=>{
+    localStorage.setItem(key, JSON.stringify(value));
+  }
+  globalThis.getStorage = (key, value)=>{
+    return JSON.parse(localStorage.getItem(key));
+  }
+  const local_data =  globalThis.getStorage('local_data');
+  console.log(local_data);
+  return local_data;
+}
+"""
 
+def add_pingpong(ld, ping, pong):
+  if ld is not None:
+    ppm = GradioAlpacaChatPPManager.from_json(json.dumps(ld))
+    ppm.add_pingpong(PingPong(ping, pong))
+    return str(ppm)
+  
+  else:
+    ppm = GradioAlpacaChatPPManager()
+    ppm.add_pingpong(PingPong(ping, pong))
+    return str(ppm)
 
-def predict(text_input, dropdown):
-    return {
-        "text": text_input,
-        "dropdown": dropdown,
-        "something_else": [text_input] * 3 + [dropdown],
-    }
+with gr.Blocks() as block:
+  local_data = gr.JSON(
+      {},
+      label="Local Storage"
+  )
 
+  ping = gr.Text(label="ping")
+  pong = gr.Text(label="pong")
 
-with gr.Blocks() as block:
-    text_input = gr.Text(label="Input")
-    dropdown = gr.Dropdown(["first", "second", "third"], type="index")
-    local_data = gr.JSON({}, label="Local Storage")
+  btn = gr.Button("Add new ping pong")
+  btn.click(
+      fn=add_pingpong, 
+      inputs=[local_data, ping, pong], 
+      outputs=local_data
+  ).then(
+      None, local_data, None, 
+      _js="(v)=>{ setStorage('local_data',v) }"
+  )
+
+  block.load(
+      None,
+      inputs=None,
+      outputs=local_data,
+      _js=get_local_storage,
+  )
 
-    dropdown.change(None, dropdown, None, _js="(v)=>{ setStorage('dropdown',v) }")
-    text_input.change(None, text_input, None, _js="(v)=>{ setStorage('text_input',v) }")
-    local_data.change(None, local_data, None, _js="(v)=>{ setStorage('local_data',v) }")
-    btn = gr.Button("Set New Data")
-    btn.click(fn=predict, inputs=[text_input, dropdown], outputs=[local_data])
-    block.load(
-        None,
-        inputs=None,
-        outputs=[text_input, dropdown, local_data],
-        _js=get_local_storage,
-    )
-block.launch(debug=True)
+block.launch(debug=True)