Fix state

app.py

@@ -4,10 +4,6 @@ from sentiment2d import Sentiment2D
 
 s2d = Sentiment2D()
 
-history = []
-
-FONT_COLOR = "#666666"
-FILL_COLOR = "rgb(.7,.7,.7)"
 TITLE = "COMPASS Pathways 2D Sentiment Model"
 EXAMPLES = [
     "This is so awesome!",
@@ -18,25 +14,25 @@ EXAMPLES = [
 ]
 
 
-def sentiment(text, clear_history):
-    global history
+def sentiment(text, state):
     valence, arousal = s2d(text)
-    # res = f"{text}: valence={valence:0.3f}, arousal={arousal:0.3f}"
     res = dict(text=text, valence=valence, arousal=arousal, words=len(text.split()))
-    if clear_history:
-        history = []
-    history.append(res)
-    df = pd.DataFrame(history)
+    #if clear_history:
+    #    state = []
+    if state == None:
+        state = []
+    state.append(res)
+    df = pd.DataFrame(state)
     res_txt = [
         f"{r['text']}: valence={r['valence']:0.3f}, arousal={r['arousal']:0.3f}"
-        for r in history
+        for r in state
     ]
-    return "\n".join(res_txt), df
+    return "\n".join(res_txt), df, state
 
 
 iface = gr.Interface(
     fn=sentiment,
-    inputs=[gr.Textbox(lines=1, placeholder="Text for 2d sentiment..."), "checkbox"],
+    inputs=[gr.Textbox(lines=1, placeholder="Text for 2d sentiment..."), "state"],
     outputs=[
         gr.Textbox(lines=5, max_lines=5, label="Results"),
         gr.ScatterPlot(
@@ -47,8 +43,9 @@ iface = gr.Interface(
             x_lim=[-1.05, 1.05],
             y_lim=[-1.05, 1.05],
         ),
+        "state",
     ],
     titel=TITLE,
-    #examples=EXAMPLES,
+    examples=EXAMPLES,
 )
 iface.launch()