Update plt.savefig('img/df.png')

app.py

@@ -10,6 +10,10 @@ import subprocess as sp
 from shlex import split
 import pandas as pd
 
+import matplotlib
+import matplotlib.pyplot as plt
+import seaborn as sns
+
 # from textwrap import dedent
 from inspect import cleandoc
 import gradio as gr
@@ -18,6 +22,9 @@ from logzero import logger
 
 from gradiobee.seg_text import seg_text
 
+matplotlib.use("Agg")  # non-interactive for plt.savefig
+sns.set()
+sns.set_style("darkgrid")
 logzero.loglevel()  # default to 10
 
 
@@ -56,17 +63,18 @@ def process(command):
 
     # quick test altair altair-save tooltip
     # from PIL import Image
-    import altair as alt
-    from altair_saver import save
-    alt.renderers.enable('altair_saver', fmts=['vega-lite', 'png'])
+
+    # import altair as alt
+    # from altair_saver import save
+    # alt.renderers.enable('altair_saver', fmts=['vega-lite', 'png'])
 
     df_ = pd.DataFrame(data={'x': [1, 2], 'y': [3, 4], "cos": [0.1, 0.5]})
-    chart_df = alt.Chart(df_).mark_circle(size=60).encode(
-        x='x',
-        y='y',
-        color='cos',
+    # chart_df = alt.Chart(df_).mark_circle(size=60).encode(
+        # x='x',
+        # y='y',
+        # color='cos',
         # tooltip=['x', 'y', 'cos', ]
-    )
+    # )
     # .interactive()
 
     # save(chart_df, "chart_df.html")
@@ -80,8 +88,9 @@ def process(command):
     # scatter_plot.save('simple_scatter_plot_with_altairchart.html')
     # chart_df.save("chart_df.html")  # does not work, constains js
     # chart_df_html = Path("chart_df.html").read_text("utf")
-    chart_df.save("chart_df.png")  #
-    chart_df_png = "chart_df.png"
+
+    # chart_df.save("chart_df.png")  #
+    # chart_df_png = "chart_df.png"
 
     # not is_command or not flag: text, do seg_text
     _ = "\n\n".join(seg_text(command.strip()))
@@ -100,7 +109,15 @@ def process(command):
 
     # return _, chart_df_html
 
-    return _, chart_df_png
+    plt.figure()
+    sns.scatterplot(data=df_, x='x', y='y')
+    Path("img").mkdir(exist_ok=True)
+    plt.savefig("img/df.png")
+    plt.close()
+    df_png = "img/df.png"
+
+    # return _, chart_df_png
+    return _, df_png
 
 
 iface = gr.Interface(