test:added logging

llm_agent.py

@@ -40,6 +40,15 @@ class LLM_Agent:
         data_path = data.get('file_path')
         model_choice = data.get('model', 'bart')
 
+        # Log file path and check existence
+        if data_path:
+            logging.info(f"Data path received: {data_path}")
+            import os
+            if not os.path.exists(data_path):
+                logging.error(f"File does not exist at path: {data_path}")
+            else:
+                logging.info(f"File exists at path: {data_path}")
+
         # Few-shot + persona prompt for Flan-UL2 (best model)
         flan_prompt = (
             "You are VizBot, an expert data visualization assistant. "
@@ -60,6 +69,9 @@ class LLM_Agent:
         # Re-initialize data processor and chart generator if a file is specified
         if data_path:
             self.data_processor = DataProcessor(data_path)
+            # Log loaded columns
+            loaded_columns = self.data_processor.get_columns()
+            logging.info(f"Loaded columns from data: {loaded_columns}")
             self.chart_generator = ChartGenerator(self.data_processor.data)
 
         if model_choice == 'bart':