Update app.py

app.py

@@ -5,13 +5,14 @@ from pandasai.llm.starcoder import Starcoder
 import matplotlib.pyplot as plt
 import gradio as gr
 
-def plot_chey():
+#function to plot
+def plot_response():
     a,b = response.figure, response.axes
     return a
     
 title='Plot'
-# Define List of Models
 
+# Define List of Models
 models = {"Starcoder": Starcoder}
 
 #@title Select Model to Run
@@ -24,35 +25,20 @@ df = pd.DataFrame({
     "country": ["United States", "United Kingdom", "France", "Germany", "Italy", "Spain", "Canada", "Australia", "Japan", "China"],
     "gdp": [21400000, 2940000, 2830000, 3870000, 2160000, 1350000, 1780000, 1320000, 516000, 14000000],
     "happiness_index": [7.3, 7.2, 6.5, 7.0, 6.0, 6.3, 7.3, 7.3, 5.9, 5.0]})
+
 # Model Initialisation
 llm = models[model_to_run](api_token=API_KEY)
 pandas_ai = PandasAI(llm, save_charts =True, conversational=False, verbose=True)
+
 # Enter Prompt related to data or Select from Pre-defined for demo purposes.
 prompt = 'Plot the histogram of countries showing for each the gdp, using different colors for each bar' #@param [ "What is the relation between GDP and Happines Index", "Plot the histogram of countries showing for each the gpd, using different colors for each bar", "GDP of Top 5 Happiest Countries?"]  {allow-input: true}
 response = pandas_ai.run(df, prompt=prompt,
                           is_conversational_answer=False) 
 
 heading = 'Pandas AI : Dataframe analytics using text prompt'
-with gr.Blocks(title = heading, theme= 'snehilsanyal/scikit-learn') as demo:
+with gr.Blocks(title = heading) as demo:
     gr.Markdown("# {}".format(heading))
-   
- 
-    
-    button = gr.Button(value = 'submit')
-    button.click(plot_chey, outputs = gr.Plot())
-
-demo.launch()
-"""
-
-with gr.Blocks(title=title) as demo:
-    gr.Markdown('''
-            <div>
-            <h1 style='text-align: center'>Plot</h1>
-            </div>
-        ''')
-    button = gr.Button(value = 'nokku')
-    
-    button.click(plot_chey, outputs = gr.Plot())
-demo.launch()
-"""    
-    
+    button = gr.Button(value = 'click to generate plot')
+    button.click(plot_response, outputs = gr.Plot())
+
+demo.launch()