Update eb_agent_module.py

eb_agent_module.py

@@ -486,7 +486,6 @@ Relevant Benchmarks: {benchmarks_val}"""
         logging.info("EmployerBrandingAgent chat history cleared by request.")
 
 def get_all_schemas_representation(all_dataframes: dict) -> str:
-    # ... (implementation unchanged)
     if not all_dataframes: return "No DataFrames are currently loaded."
     schema_descriptions = ["DataFrames currently available in the application state:"]
     for key, df in all_dataframes.items():
@@ -496,8 +495,17 @@ def get_all_schemas_representation(all_dataframes: dict) -> str:
         if df.empty:
             schema = f"\n--- DataFrame: {df_name} ---\nStatus: Empty\nShape: {shape}\nColumns: {columns}"
         else:
-            sample_data_str = df.head(2).to_markdown(index=False)
-            schema = (f"\n--- DataFrame: {df_name} ---\nShape: {shape}\nColumns: {columns}\n\n<details><summary>Sample Data (first 2 rows of {df_name}):</summary>\n\n{sample_data_str}\n\n</details>")
+            try:
+                # Attempt to use to_markdown, with a fallback to to_string
+                sample_data_str = df.head(2).to_markdown(index=False)
+            except ImportError:
+                logging.warning("`tabulate` library not found. Falling back to `to_string()` for schema representation.")
+                sample_data_str = df.head(2).to_string(index=False)
+            except Exception as e:
+                logging.error(f"Error formatting DataFrame sample for {df_name} with to_markdown: {e}. Falling back to to_string().")
+                sample_data_str = df.head(2).to_string(index=False)
+                
+            schema = (f"\n--- DataFrame: {df_name} ---\nShape: {shape}\nColumns: {columns}\n\n<details><summary>Sample Data (first 2 rows of {df_name}):</summary>\n\n```text\n{sample_data_str}\n```\n\n</details>")
         schema_descriptions.append(schema)
     return "\n".join(schema_descriptions)