Update app.py

app.py

@@ -173,12 +173,13 @@ def random_response(message):
                                                     # "please give me the most related and useful possible question to get simple but useful insights for {question}."
                                                     "The data is sales order line is every transaction of the company."
                                                     "Base on the question:{question}, regenerate the output"
-                                                    "Your output will be used to guide the graph generation by python using ploty, so make it simple and easier to process data."
+                                                    # "Your output will be used to guide the graph generation by python using ploty, so make it simple and easier to process data."
                                                     "For example: 'Goal(question='What are the sales trends by product category?visualization='bar chart of prod_category against sum(trxn_item_qty) grouped by trxn_date'. and"
                                                     "'Goal(question='Who are the top customers based on transaction count?', visualization='Bar chart of customer_name vs. count(trxn_id)')"
+                                                    "The visualization should align with the question and the data."
                                                     "Usually, when group by age, show all the data."
                                                     "But for other datas: customer,prouct,sales,qty,etc show top 10."
-                                                    "Process the data at last when put in the graph."
+                                                    "Process the top data at last when put in the graph."
                                                     "When ask customers,customer, it means customer_name."
                                                     "When ask product, it means prod_name."
                                                     "When ask category, it means prod_category."
@@ -187,6 +188,7 @@ def random_response(message):
                                                     "Only output 1 question."
                                                     "")
 
+
             chain = (
                 {"str_summary": RunnablePassthrough(),"question": RunnablePassthrough()}
                 | prompt
@@ -203,7 +205,7 @@ def random_response(message):
 
             
             try:
-                temp_chart = lida.visualize(summary=summary, goal=str(insights)+'Graph heigh 800,width 1000.If there is statement in previous question using monthly data,other time related using daily.set different color to the graph,x label rotate 45 degree,do not use the guide line', textgen_config=text_gen_config,library='matplotlib')
+                temp_chart = lida.visualize(summary=summary, goal=str(insights)+"Graph heigh 800,width 1000.Set different colors to different varibles.x label rotate 45 degree,do not use the guide line", textgen_config=text_gen_config,library='matplotlib')
                 print(f'*'*50)
                 code = temp_chart[0].code
                 print(f"{code}")
@@ -222,6 +224,8 @@ def random_response(message):
             # Open the image file
             # img = Image.open(image_path)
             img = mpimg.imread('chart_1.png')
+            print(f'*'*50)
+            print(f"Image opened")
 
             base64_image = encode_image(image_path)
 
@@ -231,7 +235,7 @@ def random_response(message):
                 [
                     HumanMessage(
                         content=[
-                            {"type": "text", "text": f"Give me some business insights base on the graph and the data{str_summary}, contain exact number conclusion."},
+                            {"type": "text", "text": f"Give me some business insights base on the graph, contain exact number conclusion."},
                             {
                                 "type": "image_url",
                                 "image_url": {