Update utility.py

utility.py

@@ -1172,7 +1172,16 @@ def read_datalake(user_phone: str, query: str) -> str:
                 )
 
                 datalake_dfs = [df for _, df in all_dfs_with_names]
-                lake = SmartDatalake(datalake_dfs, config={"llm": llm, "response_parser": FlaskResponse, "save_charts_path": user_defined_path, "enable_cache": False, "conversational": False})
+                lake = SmartDatalake(
+                    datalake_dfs,
+                    config={
+                        "llm": llm,
+                        "response_parser": FlaskResponse,
+                        "save_charts_path": user_defined_path,
+                        "enable_cache": False,
+                        "conversational": False
+                    }
+                )
                 
                 response = lake.chat(pandasai_prompt)
 
@@ -1183,62 +1192,73 @@ def read_datalake(user_phone: str, query: str) -> str:
                 else:
                     return str(response)
 
-            except (NoCodeFoundError, MaliciousQueryError) as e:
-                logger.info(f"PandasAI found no code for '{query}' (Error: {type(e).__name__}). Routing to Business Coach as fallback.")
-                snapshot = engine.generate_business_snapshot()
-                snapshot_str = json.dumps(snapshot, indent=2)
-
-                synthesis_prompt = f"""
-                You are Qx, a friendly and insightful business coach and financial expert. The user is asking a general question. Only perform calculations when necessary. Your task is to provide a clear, helpful, and strategic answer based on their question, using your general knowledge and the business snapshot provided below for context.
-
-                **IMPORTANT RULES:**
-                1.  **Use the Context:** Use the Business Snapshot as your internal knowledge to make your advice relevant and personalized. For example, if inventory is high for an item, you might suggest a promotion. If profit is low, you might suggest cost-cutting measures.
-                2.  **You can state the numbers, info or metrics from the data where helpful. Synthesize them into your advice. generate insight from the data if needed.
-                3.  **Stay in Character:** Act as a coach. Be encouraging and provide actionable advice.
-                4.  **Handle 'Help' Queries:** If asked about your capabilities, explain that you can record transactions (sales, expenses, etc.) via text or images, generate detailed reports (profit, sales by item), answer questions about their data, and provide business advice.
-                5.  **Format for WhatsApp:** Use *bold*, _italic_, and emojis to make your response clear and engaging.
-
-                **BUSINESS SNAPSHOT (INTERNAL CONTEXT ONLY):**
-                {snapshot_str}
-
-                **User's Question:**
-                "{query}"
-                """
-                try:
-                    response = llm.invoke(synthesis_prompt)
-                    return response.content
-                except Exception as llm_error:
-                    logger.error(f"Business coach LLM failed for query '{query}': {llm_error}", exc_info=True)
-                    return "I'm here to help with your business questions, but I'm having trouble processing your request right now. Could you try rephrasing your question?"
-
             except Exception as e:
-                logger.warning(f"PandasAI failed for query '{query}' with error: {e}. Falling back to Business Coach.")
-                # Route ANY other PandasAI exception to the business coach
-                snapshot = engine.generate_business_snapshot()
-                snapshot_str = json.dumps(snapshot, indent=2)
-
-                synthesis_prompt = f"""
-                You are Qx, a friendly and insightful business coach and financial expert. The user is asking a general question. Only perform calculations when necessary. Your task is to provide a clear, helpful, and strategic answer based on their question, using your general knowledge and the business snapshot provided below for context.
-
-                **IMPORTANT RULES:**
-                1.  **Use the Context:** Use the Business Snapshot as your internal knowledge to make your advice relevant and personalized. For example, if inventory is high for an item, you might suggest a promotion. If profit is low, you might suggest cost-cutting measures.
-                2.  **You can state the numbers, info or metrics from the data where helpful. Synthesize them into your advice. generate insight from the data if needed.
-                3.  **Stay in Character:** Act as a coach. Be encouraging and provide actionable advice.
-                4.  **Handle 'Help' Queries:** If asked about your capabilities, explain that you can record transactions (sales, expenses, etc.) via text or images, generate detailed reports (profit, sales by item), answer questions about their data, and provide business advice.
-                5.  **Format for WhatsApp:** Use *bold*, _italic_, and emojis to make your response clear and engaging.
-
-                **BUSINESS SNAPSHOT (INTERNAL CONTEXT ONLY):**
-                {snapshot_str}
-
-                **User's Question:**
-                "{query}"
-                """
-                try:
-                    response = llm.invoke(synthesis_prompt)
-                    return response.content
-                except Exception as llm_error:
-                    logger.error(f"Business coach LLM failed for query '{query}': {llm_error}", exc_info=True)
-                    return "I'm here to help with your business questions, but I'm having trouble processing your request right now. Could you try rephrasing your question?"
+                # Unified handler: catch PandasAI edge-cases by CLASS NAME or MESSAGE,
+                # then route to the Business Coach fallback.
+                err_name = e.__class__.__name__
+                err_msg = str(e)
+                is_pandasai_no_code = (
+                    err_name in ("NoCodeFoundError", "MaliciousQueryError")
+                    or "No code found" in err_msg
+                    or "NoCodeFoundError" in err_name
+                    or "MaliciousQueryError" in err_name
+                )
+
+                if is_pandasai_no_code:
+                    logger.info(f"PandasAI found no code for '{query}' (Error: {err_name}). Routing to Business Coach as fallback.")
+                    snapshot = engine.generate_business_snapshot()
+                    snapshot_str = json.dumps(snapshot, indent=2)
+
+                    synthesis_prompt = f"""
+                    You are Qx, a friendly and insightful business coach and financial expert. The user is asking a general question. Only perform calculations when necessary. Your task is to provide a clear, helpful, and strategic answer based on their question, using your general knowledge and the business snapshot provided below for context.
+
+                    **IMPORTANT RULES:**
+                    1.  **Use the Context:** Use the Business Snapshot as your internal knowledge to make your advice relevant and personalized. For example, if inventory is high for an item, you might suggest a promotion. If profit is low, you might suggest cost-cutting measures.
+                    2.  **You can state the numbers, info or metrics from the data where helpful. Synthesize them into your advice. generate insight from the data if needed.
+                    3.  **Stay in Character:** Act as a coach. Be encouraging and provide actionable advice.
+                    4.  **Handle 'Help' Queries:** If asked about your capabilities, explain that you can record transactions (sales, expenses, etc.) via text or images, generate detailed reports (profit, sales by item), answer questions about their data, and provide business advice.
+                    5.  **Format for WhatsApp:** Use *bold*, _italic_, and emojis to make your response clear and engaging.
+
+                    **BUSINESS SNAPSHOT (INTERNAL CONTEXT ONLY):**
+                    {snapshot_str}
+
+                    **User's Question:**
+                    "{query}"
+                    """
+                    try:
+                        response = llm.invoke(synthesis_prompt)
+                        return response.content
+                    except Exception as llm_error:
+                        logger.error(f"Business coach LLM failed for query '{query}': {llm_error}", exc_info=True)
+                        return "I'm here to help with your business questions, but I'm having trouble processing your request right now. Could you try rephrasing your question?"
+                else:
+                    logger.warning(f"PandasAI failed for query '{query}' with error: {e}. Falling back to Business Coach.")
+                    # Route ANY other PandasAI exception to the business coach
+                    snapshot = engine.generate_business_snapshot()
+                    snapshot_str = json.dumps(snapshot, indent=2)
+
+                    synthesis_prompt = f"""
+                    You are Qx, a friendly and insightful business coach and financial expert. The user is asking a general question. Only perform calculations when necessary. Your task is to provide a clear, helpful, and strategic answer based on their question, using your general knowledge and the business snapshot provided below for context.
+
+                    **IMPORTANT RULES:**
+                    1.  **Use the Context:** Use the Business Snapshot as your internal knowledge to make your advice relevant and personalized. For example, if inventory is high for an item, you might suggest a promotion. If profit is low, you might suggest cost-cutting measures.
+                    2.  **You can state the numbers, info or metrics from the data where helpful. Synthesize them into your advice. generate insight from the data if needed.
+                    3.  **Stay in Character:** Act as a coach. Be encouraging and provide actionable advice.
+                    4.  **Handle 'Help' Queries:** If asked about your capabilities, explain that you can record transactions (sales, expenses, etc.) via text or images, generate detailed reports (profit, sales by item), answer questions about their data, and provide business advice.
+                    5.  **Format for WhatsApp:** Use *bold*, _italic_, and emojis to make your response clear and engaging.
+
+                    **BUSINESS SNAPSHOT (INTERNAL CONTEXT ONLY):**
+                    {snapshot_str}
+
+                    **User's Question:**
+                    "{query}"
+                    """
+                    try:
+                        response = llm.invoke(synthesis_prompt)
+                        return response.content
+                    except Exception as llm_error:
+                        logger.error(f"Business coach LLM failed for query '{query}': {llm_error}", exc_info=True)
+                        return "I'm here to help with your business questions, but I'm having trouble processing your request right now. Could you try rephrasing your question?"
 
     except Exception as e:
         logger.error(f"Data query failed for user {user_phone}, query '{query}': {e}", exc_info=True)
@@ -1272,6 +1292,7 @@ def read_datalake(user_phone: str, query: str) -> str:
             return "I'm here to help with your business questions, but I'm experiencing technical difficulties right now. Please try again in a moment."
 
 
+
 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']: