Update app.py

Probando el comportamiento con plotly, añadí instrucciones al propmt:

1. Usar plotly como herramienta para graficar
2. Dar ejemplo de uso de traces con marks para mejor visualización

app.py

@@ -13,6 +13,7 @@ from supabase import create_client, Client  # type: ignore
 from pandasai import SmartDatalake  # type: ignore # Porque ya usamos más de un df (más de una tabla de nuestra db)
 from pandasai.llm.local_llm import LocalLLM # type: ignore
 from pandasai import Agent
+import plotly.graph_objects as go
 import matplotlib.pyplot as plt
 import time
 
@@ -27,9 +28,12 @@ def generate_graph_prompt(user_query):
 
             Given the user's request: "{user_query}"
 
-            1. Plot the relevant data using matplotlib:
+            1. Plot the relevant data using graph_objects plotly:
             - Use `df.query("geo == 'X'")` to filter the country, instead of chained comparisons.
             - Avoid using filters like `df[df['geo'] == 'Germany']`.
+            - Use traces with 'line+markers' mode. (e.g, fig.add_trace(go.Scatter(x='X', y='Y',
+                    mode='lines+markers',
+                    name='Country_name')))
             - Include clear axis labels and a descriptive title.
             - Save the plot as an image file (e.g., temp_chart.png).