Update utility.py

utility.py

@@ -1185,12 +1185,35 @@ def read_datalake(user_phone: str, query: str) -> str:
                 
                 response = lake.chat(pandasai_prompt)
 
+                # --- Sanitize response so error text never reaches the user ---
+                def _looks_like_error_text(s: str) -> bool:
+                    if not s:
+                        return False
+                    s_low = s.lower()
+                    return any(tok in s_low for tok in [
+                        "traceback",
+                        "no code found",
+                        "nocodefounderror",
+                        "maliciousqueryerror",
+                        "pandasai.exceptions",
+                        "pipeline failed",
+                        "error:"
+                    ])
+
                 if isinstance(response, dict) and 'value' in response:
-                    return str(response['value'])
+                    text = str(response['value'])
+                    if _looks_like_error_text(text):
+                        raise RuntimeError("Unsafe PandasAI response text contained error markers")
+                    return text
                 elif isinstance(response, str):
+                    if _looks_like_error_text(response):
+                        raise RuntimeError("Unsafe PandasAI response text contained error markers")
                     return response
                 else:
-                    return str(response)
+                    text = str(response)
+                    if _looks_like_error_text(text):
+                        raise RuntimeError("Unsafe PandasAI response text contained error markers")
+                    return text
 
             except Exception as e:
                 # Unified handler: catch PandasAI edge-cases by CLASS NAME or MESSAGE,
@@ -1202,6 +1225,7 @@ def read_datalake(user_phone: str, query: str) -> str:
                     or "No code found" in err_msg
                     or "NoCodeFoundError" in err_name
                     or "MaliciousQueryError" in err_name
+                    or "pandasai.exceptions.NoCodeFoundError" in err_msg
                 )
 
                 if is_pandasai_no_code:
@@ -1293,6 +1317,7 @@ def read_datalake(user_phone: str, query: str) -> str:
 
 
 
+
 def _find_document_by_details(user_phone: str, collection_name: str, details: Dict) -> Optional[Any]:
     col_ref = db.collection("users").document(user_phone).collection(collection_name)
     if 'transaction_id' in details and details['transaction_id']: