Update eb_agent_module.py

eb_agent_module.py

@@ -594,31 +594,31 @@ class EmployerBrandingAgent:
                 dfs = list(self.pandas_dfs.values())
                 pandas_response = pai.chat(query, *dfs)
             
-        # Handle different response types (text, charts, etc.)
-        response_text = ""
-        chart_info = ""
-        
-        # Check if a chart was generated
-        import os
-        charts_dir = "./charts"
-        if os.path.exists(charts_dir):
-            chart_files = [f for f in os.listdir(charts_dir) if f.endswith(('.png', '.jpg', '.jpeg', '.svg'))]
-            if chart_files:
-                # Get the most recent chart file
-                chart_files.sort(key=lambda x: os.path.getmtime(os.path.join(charts_dir, x)), reverse=True)
-                latest_chart = chart_files[0]
-                chart_path = os.path.join(charts_dir, latest_chart)
-                chart_info = f"\n\n📊 **Chart Generated**: {latest_chart}\nChart saved at: {chart_path}"
-                logging.info(f"Chart generated: {chart_path}")
-        
-        # Combine text response with chart info
-        if pandas_response and str(pandas_response).strip():
-            response_text = str(pandas_response).strip()
-        else:
-            response_text = "Analysis completed"
-        
-        final_response = response_text + chart_info
-        return final_response, True
+            # Handle different response types (text, charts, etc.)
+            response_text = ""
+            chart_info = ""
+            
+            # Check if a chart was generated
+            import os
+            charts_dir = "./charts"
+            if os.path.exists(charts_dir):
+                chart_files = [f for f in os.listdir(charts_dir) if f.endswith(('.png', '.jpg', '.jpeg', '.svg'))]
+                if chart_files:
+                    # Get the most recent chart file
+                    chart_files.sort(key=lambda x: os.path.getmtime(os.path.join(charts_dir, x)), reverse=True)
+                    latest_chart = chart_files[0]
+                    chart_path = os.path.join(charts_dir, latest_chart)
+                    chart_info = f"\n\n📊 **Chart Generated**: {latest_chart}\nChart saved at: {chart_path}"
+                    logging.info(f"Chart generated: {chart_path}")
+            
+            # Combine text response with chart info
+            if pandas_response and str(pandas_response).strip():
+                response_text = str(pandas_response).strip()
+            else:
+                response_text = "Analysis completed"
+            
+            final_response = response_text + chart_info
+            return final_response, True
         
     except Exception as e:
         logging.error(f"Error in PandasAI processing: {e}", exc_info=True)