Update eb_agent_module.py

eb_agent_module.py

@@ -580,6 +580,8 @@ class EmployerBrandingAgent:
         else:
             return "general"
 
+# Replace the _generate_pandas_response method and everything after it with this properly indented code:
+
     async def _generate_pandas_response(self, query: str) -> tuple[str, bool]:
         """Generate response using PandasAI for data queries"""
         if not self.pandas_agent or not hasattr(self, 'pandas_dfs'):
@@ -647,7 +649,7 @@ class EmployerBrandingAgent:
             # Try to provide a more helpful error message
             if "Invalid output" in str(e) and "plot save path" in str(e):
                 return "I tried to create a visualization but encountered a formatting issue. Please try rephrasing your request or ask for specific data without requesting a chart.", False
-    
+            return f"Error processing data query: {str(e)}", False
 
     async def _generate_enhanced_response(self, query: str, pandas_result: str = "", query_type: str = "general") -> str:
         """Generate enhanced response combining PandasAI results with RAG context"""
@@ -767,7 +769,6 @@ class EmployerBrandingAgent:
                 logging.error(f"Error in _generate_enhanced_response: {e}", exc_info=True)
                 return f"I encountered an error while processing your request: {str(e)}"
 
-
     def _validate_query(self, query: str) -> bool:
         """Validate user query input"""
         if not query or not isinstance(query, str) or len(query.strip()) < 3:
@@ -803,10 +804,18 @@ class EmployerBrandingAgent:
             # For data-related queries, try PandasAI first
             if query_type in ["data", "hybrid"] and self.pandas_agent:
                 logging.info("Attempting PandasAI analysis...")
-                pandas_text_output, pandas_chart_path, pandas_success = await self._generate_pandas_response(user_query)
+                pandas_text_output, pandas_success = await self._generate_pandas_response(user_query)
                 
                 if pandas_success:
-                    logging.info(f"PandasAI analysis successful. Text: '{str(pandas_text_output)[:100]}...', Chart: '{pandas_chart_path}'")
+                    logging.info(f"PandasAI analysis successful. Text: '{str(pandas_text_output)[:100]}...'")
+                    # Check for chart generation in response
+                    if "Chart Generated" in pandas_text_output:
+                        # Extract chart path from response if present
+                        lines = pandas_text_output.split('\n')
+                        for line in lines:
+                            if "Chart saved at:" in line:
+                                pandas_chart_path = line.split("Chart saved at: ")[1].strip()
+                                break
                 else:
                     # pandas_text_output might contain the error message from PandasAI
                     logging.warning(f"PandasAI analysis failed or returned no specific result. Message from PandasAI: {pandas_text_output}")
@@ -820,7 +829,6 @@ class EmployerBrandingAgent:
             elif query_type in ["data", "hybrid"] and not self.pandas_agent:
                 llm_context_from_pandas += "Note: The data analysis tool is currently unavailable.\n"
 
-
             # Always call the LLM to formulate the final response
             final_llm_response = await self._generate_enhanced_response(
                 query=user_query,
@@ -837,7 +845,6 @@ class EmployerBrandingAgent:
             logging.error(f"Critical error in process_query: {e}", exc_info=True)
             return {"text": f"I encountered a critical error while processing your request: {type(e).__name__}. Please check the logs.", "image_path": None}
 
-
     def update_dataframes(self, new_dataframes: Dict[str, pd.DataFrame]):
         """Updates the agent's DataFrames and reinitializes PandasAI agent"""
         self.all_dataframes = {k: v.copy() for k, v in new_dataframes.items()}
@@ -877,7 +884,6 @@ class EmployerBrandingAgent:
             "charts_save_path_pandasai": pai.config.save_charts_path if pai.config.llm else "PandasAI not configured"
         }
 
-
     def get_available_analyses(self) -> List[str]:
         """Returns list of suggested analyses based on available data"""
         if not self.all_dataframes: