Update app.py

app.py

@@ -5,6 +5,9 @@ from pandasai.llm.starcoder import Starcoder
 import matplotlib.pyplot as plt
 #import gradio as gr
 
+def make_plot(x):
+    return x.figure
+
 # Define List of Models
 
 models = {"Starcoder": Starcoder}
@@ -27,4 +30,12 @@ prompt = 'Plot the histogram of countries showing for each the gdp, using differ
 response = pandas_ai.run(df, prompt=prompt,
                           is_conversational_answer=False) 
 
-response.figure.savefig('chart.jpg')
+with gr.Blocks(title=title) as demo:
+    gr.Markdown('''
+            <div>
+            <h1 style='text-align: center'>Plot</h1>
+            </div>
+        ''')
+    o1=gr.Plot()
+    btn = gr.Button(value = 'Submit')
+    btn.click(make_plot,inputs=[response],outputs=[o1])