Update app.py

app.py

@@ -26,12 +26,12 @@ def sentiment(text, state):
     if state == None:
         state = []
     state.append(res)
-    # df = pd.DataFrame(state)
+    df = pd.DataFrame(state)
     res_txt = [
         f"{r['text']}: valence={r['valence']:0.3f}, arousal={r['arousal']:0.3f}, confidence=arousal={r['confidence']:0.3f}"
         for r in state
     ]
-    return "\n".join(res_txt), state # df, state
+    return "\n".join(res_txt), df, state
 
 
 iface = gr.Interface(
@@ -39,6 +39,16 @@ iface = gr.Interface(
     inputs=[gr.Textbox(lines=1, placeholder="Text for 3d sentiment..."), "state"],
     outputs=[
         gr.Textbox(lines=5, max_lines=5, label="Results"),
+        gr.ScatterPlot(
+            x="valence",
+            y="arousal",
+            tooltip="text",
+            size="words",
+            size_legend_position="none",
+            interactive=False,
+            x_lim=[-1.05, 1.05],
+            y_lim=[-1.05, 1.05],
+        ),
         "state",
     ],
     title=TITLE,