Update app.py

app.py

@@ -6,6 +6,8 @@ import os
 import pandas as pd
 from io import StringIO, BytesIO
 import base64
+import json
+import plotly.io as pio
 # from linePlot import plot_stacked_time_series, plot_emotion_topic_grid
 
 # Define your Hugging Face token (make sure to set it as an environment variable)
@@ -37,14 +39,12 @@ def stream_chat_with_rag(
     print("top works from API:")
     print(fig)
     
-    plot_bytes = base64.b64decode(fig['plot'].split(',')[1])
-    img = plt.imread(BytesIO(fig), format='PNG')
-    plt.figure(dpi = 150)
-    plt.imshow(img)
-    plt.axis('off')
-    plt.show()
-    # return answer, fig
-    return answe, plt.gcf()
+    fig_dict = json.loads(plotly_data['plot'])
+    
+    # Render the figure
+    fig = pio.from_json(json.dumps(fig_dict))
+    fig.show() 
+    return answe, fig