Update app.py

app.py

@@ -58,9 +58,14 @@ def main():
                 max_retries=6,
             )
 
-            # Updated Prefix to strictly separate Thought, Action, and Final Answer
-            custom_prefix = """
+            # Updated Prefix: We manually include the DF context here since we are using a custom suffix
+            # This satisfies the requirement to have DF info without triggering the ValueError
+            df_context = f"The dataframe 'df' has the following columns: {', '.join(df.columns)}"
+            
+            custom_prefix = f"""
             You are working with a pandas dataframe in Python. The name of the dataframe is `df`.
+            {df_context}
+            
             You should use the tools below to answer the question posed of you.
             
             STRICT FORMATTING RULES:
@@ -69,7 +74,7 @@ def main():
             3. Your 'Action Input' must be valid Python code and ONLY Python code. No markdown backticks.
             4. Before creating a visualization for filtered data, always check if the filtered dataframe is empty.
             5. For plots, use `plt.figure()` and end with `st.pyplot(plt.gcf())`.
-            6. The visuals must be clear and perfect .
+            6. The visual and code must be clear and beautiful.
             7. Avoide over writing .
             """
 
@@ -81,6 +86,7 @@ def main():
             """
 
             # Create the Pandas Agent
+            # FIX: Set include_df_in_prompt=False to allow the use of a custom suffix
             agent = create_pandas_dataframe_agent(
                 llm,
                 df,
@@ -89,6 +95,7 @@ def main():
                 allow_dangerous_code=True,
                 prefix=custom_prefix,
                 suffix=custom_suffix,
+                include_df_in_prompt=False, # Mandatory when using custom suffix
                 handle_parsing_errors=True,
                 agent_executor_kwargs={
                     "handle_parsing_errors": True