Update app.py

app.py

@@ -134,27 +134,22 @@ def random_response(message, history):
     summary = lida.summarize(df_data)
     
     print(f'*'*50)
-    goals = lida.goals(summary, n=1, textgen_config=text_gen_config,persona=f'Use color RoyalBlue#4169E1 for the line or bar.Show the average value line.An data analyst of the company who want to know {question}')
+    goals = lida.goals(summary, n=1, textgen_config=text_gen_config,persona=f'An data analyst of the company who want to know {question}')
 
     # ValueError: Unsupported library. Choose from 'matplotlib', 'seaborn', 'plotly', 'bokeh', 'ggplot', 'altair'.
+    print(f'goals: {goals[0]}')
+    
+    try:
+        temp_chart = lida.visualize(summary=summary, goal=goals[0].question+'set different color to the graph', textgen_config=text_gen_config,library='plotly')
+        # code = temp_chart[0].code
+        # instructions = ["change the color of the graph to #4169E1 if there is only one variable","change the background color to white but keep the grid lines grey","set the average line for the graph to be red"]
+        # edited_chart = lida.edit(code=code,summary=summary,instructions=instructions,library='plotly',textgen_config = text_gen_config)
+    except Exception as e:
+        print(f"Error while: {e}")
 
-    chart_result = []
-
-    final_explanation = []
-
-    for i in range(1):
-        try:
-            print(f"Goal{i}: {goals[i]}")
-            temp_chart = lida.visualize(summary=summary, goal=goals[i], textgen_config=text_gen_config,library='plotly')
-            temp_explanation = lida.explain(code=temp_chart[0].code)
-            final_explanation.append(temp_explanation)
-            chart_result.append(temp_chart)
-        except Exception as e:
-            print(f"Error while: {e}")
-    for i in range(len(chart_result)):
-        chart_result[i][0].savefig(f'chart_{i}.png')
-        print(f'*'*50)
-        print(f"Chart {i} saved")
+    temp_chart[0].savefig(f'chart_0.png')
+    print(f'*'*50)
+    print(f"Chart  saved")
 
     # Path to your image
     image_path = "chart_0.png"
@@ -195,7 +190,7 @@ with gr.Blocks() as demo:
             chat_output = gr.Textbox(label="Response", interactive=False)
 
             examples = gr.Examples(
-                examples=['Top 10 prod_cate sales rank', 'Top product in category Seafood','Sum sales trend of product name called Lobster by time per day'],
+                examples=['Top 10 prod_cate sales', 'Top product in category Seafood','Sum sales trend of product name called Lobster by time per day'],
                 inputs=chat_input
             )