Hi @gventuri

I do have the same issue, plot is generated and saved as a temp_chart.png.
for displaying the chart in the streamlit, the code generated by LLM failing and iterating till last execution.
so the chart generated or response is showing as Value must match with type plot a string object
By the way I am using StreamlitResponse as response parser

logger's

The user asked the following question:

QUERY

How does the distribution of item visibility vary across different outlet types?

You generated this python code:
df = pd.concat(dfs)
visibility_by_outlet_type = df.groupby('Outlet_Type')['Item_Visibility'].mean()
plt.figure(figsize=(10, 6))
visibility_by_outlet_type.plot(kind='bar', color='skyblue')
plt.title('Distribution of Item Visibility Across Different Outlet Types')
plt.xlabel('Outlet Type')
plt.ylabel('Mean Item Visibility')
plt.xticks(rotation=45)
plt.tight_layout()
plt.savefig('C:/Users/xx/OneDrive - xx xx/Desktop/QuickBooks/qbo_plot/exports/charts/temp_chart.png')
result = {'type': 'plot', 'value': 'C:/Users/xxx/OneDrive - xx xx/Desktop/QuickBooks/qbo_plot/exports/charts/temp_chart.png'}

It fails with the following error:
Traceback (most recent call last):
File "C:\Users\xx\OneDrive - xx xx\Desktop\QuickBooks\qbo_plot\myenv\lib\site-packages\pandasai\pipelines\chat\code_execution.py", line 98, in execute
raise InvalidOutputValueMismatch(
pandasai.exceptions.InvalidOutputValueMismatch: Value must match with type plot

Fix the python code above and return the new python code:
INFO:pandasai.helpers.logger:Executing Step 1: CodeGenerator
INFO:httpx:HTTP Request: POST https://api.openai.com/v1/chat/completions "HTTP/1.1 200 OK"
INFO:pandasai.helpers.logger:Prompt used:

dfs[0]:5681x11
Item_Identifier,Item_Weight,Item_Fat_Content,Item_Visibility,Item_Type,Item_MRP,Outlet_Identifier,Outlet_Establishment_Year,Outlet_Size,Outlet_Location_Type,Outlet_Type
FDA14,8.115,LF,0.058572263,Health and Hygiene,62.6536,OUT010,1987,Small,Tier 2,Supermarket Type1
NCN55,,Regular,0.114571006,Snack Foods,208.7638,OUT019,2002,Medium,Tier 3,Grocery Store
FDQ15,15.6,low fat,0.015434464,Canned,100.5358,OUT045,1985,,Tier 1,Supermarket Type3

The user asked the following question:

QUERY

How does the distribution of item visibility vary across different outlet types?

You generated this python code:
df = pd.concat(dfs)
visibility_by_outlet_type = df.groupby('Outlet_Type')['Item_Visibility'].mean()
plt.figure(figsize=(10, 6))
visibility_by_outlet_type.plot(kind='bar', color='skyblue')
plt.title('Distribution of Item Visibility Across Different Outlet Types')
plt.xlabel('Outlet Type')
plt.ylabel('Mean Item Visibility')
plt.xticks(rotation=45)
plt.tight_layout()
plt.savefig('C:/Users/xxxx/OneDrive - xx xxx/Desktop/QuickBooks/qbo_plot/exports/charts/temp_chart.png')
result = {'type': 'plot', 'value': 'C:/Users/xxx/xx - xxx Ltd/Desktop/QuickBooks/qbo_plot/exports/charts/temp_chart.png'}

It fails with the following error:
Traceback (most recent call last):
File "C:\Users\xx\xx - xx Ltd\Desktop\QuickBooks\qbo_plot\myenv\lib\site-packages\pandasai\pipelines\chat\code_execution.py", line 98, in execute
raise InvalidOutputValueMismatch(
pandasai.exceptions.InvalidOutputValueMismatch: Value must match with type plot

Fix the python code above and return the new python code:

INFO:pandasai.helpers.logger:Code generated:
```
import pandas as pd
import matplotlib.pyplot as plt

df = pd.concat(dfs)
visibility_by_outlet_type = df.groupby('Outlet_Type')['Item_Visibility'].mean()
plt.figure(figsize=(10, 6))
visibility_by_outlet_type.plot(kind='bar', color='skyblue')
plt.title('Distribution of Item Visibility Across Different Outlet Types')
plt.xlabel('Outlet Type')
plt.ylabel('Mean Item Visibility')
plt.xticks(rotation=45)
plt.tight_layout()
plt.show()
```

INFO:pandasai.helpers.logger:Executing Step 2: CodeCleaning
INFO:pandasai.helpers.logger:
Code running:

df = pd.concat(dfs)
visibility_by_outlet_type = df.groupby('Outlet_Type')['Item_Visibility'].mean()
plt.figure(figsize=(10, 6))
visibility_by_outlet_type.plot(kind='bar', color='skyblue')
plt.title('Distribution of Item Visibility Across Different Outlet Types')
plt.xlabel('Outlet Type')
plt.ylabel('Mean Item Visibility')
plt.xticks(rotation=45)
plt.tight_layout()
plt.show()
        ```
ERROR:pandasai.helpers.logger:Failed with error: Traceback (most recent call last):
  File "C:\Users\xx\xx - xx xx\Desktop\QuickBooks\qbo_plot\myenv\lib\site-packages\pandasai\pipelines\chat\code_execution.py", line 98, in execute
    raise InvalidOutputValueMismatch(
pandasai.exceptions.InvalidOutputValueMismatch: Value must match with type plot

Try using following in the prompt:
When plotting use appropriate Plotly chart, write plot to image file and return path. Display the plot using Streamlit (import streamlit as st; st.plotly_chart) if returning chart response. Do not run .show() on figure. Use pio.write_image to save image to file.

With OpenAI extension (does not work with LiteLLM - need investigation) it generates the following code:

df_top_products = execute_sql_query(sql_query)
fig = px.bar(df_top_products, x='Product', y='conversion_rate', title='Top 5 Products by Conversion Rate', labels={'conversion_rate': 'Conversion Rate', 'Product': 'Product'}, text=df_top_products['conversion_rate'].apply(lambda x: f'{x:.2%}'))
fig.update_traces(textposition='outside')
fig.update_layout(yaxis_tickformat='.2%')

plot_path = 'exports/charts/temp_chart_3b6149e0-1546-4615-ab4d-ef694ad72663.png'
pio.write_image(fig, plot_path)
st.plotly_chart(fig)

result = {'type': 'plot', 'value': plot_path}