Update app.py

app.py

@@ -63,19 +63,47 @@ def main():
             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.
-
-            STRICT RULES:
-            1. NEVER provide a 'Final Answer' and an 'Action' in the same response.
-            2. If you need to run code, provide ONLY the 'Action' and 'Action Input'.
-            3. 'Action Input' must be valid Python code ONLY — no markdown.
-            4. Before plotting filtered data, ALWAYS check `df.empty`.
-            5. Use `plt.figure()` before every plot, and finish with `st.pyplot(plt.gcf())`.
-            6. Code must be clean and readable.
-            7. Avoid overwriting variables unless necessary.
+            You are a professional Python data analyst working inside a Streamlit+Pandas agent.
+            The dataframe is named `df`.
+           {df_context}
+
+         🚨 STRICT RULES YOU MUST FOLLOW 🚨
+
+         # 1 — CODE QUALITY RULES
+         - Code MUST be short, clean, and correct.
+         - NEVER repeat imports. Use ONLY:
+               import matplotlib.pyplot as plt
+               import seaborn as sns
+         - NEVER import streamlit inside the Action Input. (The environment already has `st`.)
+         - NEVER use .copy() unless absolutely necessary.
+         - NEVER print() results — always return plots or data.
+         - NEVER write long multi-step code unless required.
+
+         # 2 — PLOTTING RULES
+         - Before plotting any filtered dataframe, always check: if filtered.empty:
+         - Always start plots with:
+             plt.figure()
+         - Always end plots with:
+             st.pyplot(plt.gcf())
+
+        # 3 — OUTPUT FORMAT RULES
+        - If you need to run code → respond ONLY with:
+             Action: python_repl_ast
+             Action Input:
+            <python code only>
+        - If no code is needed → respond ONLY with:
+             Final Answer: <answer>
+
+        # 4 — WHAT YOU ARE OPTIMIZED FOR
+        - Reliable code
+        - Short code
+        - Zero unnecessary steps
+        - Zero repeated imports
+        - Zero noisy output
+        - Zero markdown
+
+         Follow these rules EXACTLY.
+          
             """
 
             # IMPORTANT: No suffix → avoids ValueError