fix: long name problem fixed in chart
Browse files- postgre_mcp_server.py +4 -0
postgre_mcp_server.py
CHANGED
|
@@ -630,6 +630,10 @@ async def visualize_results(json_data: dict, vis_prompt: str) -> str:
|
|
| 630 |
|
| 631 |
# Convert JSON to DataFrame
|
| 632 |
df = pd.DataFrame(json_data["data"], columns=json_data["columns"])
|
|
|
|
|
|
|
|
|
|
|
|
|
| 633 |
|
| 634 |
# Initialize PandasAI
|
| 635 |
df_ai = pai.DataFrame(df)
|
|
|
|
| 630 |
|
| 631 |
# Convert JSON to DataFrame
|
| 632 |
df = pd.DataFrame(json_data["data"], columns=json_data["columns"])
|
| 633 |
+
|
| 634 |
+
# Shorten long values in text columns
|
| 635 |
+
for column in df.select_dtypes(include=['object']).columns:
|
| 636 |
+
df[column] = df[column].apply(lambda x: str(x)[:20] + '...' if len(str(x)) > 20 else str(x))
|
| 637 |
|
| 638 |
# Initialize PandasAI
|
| 639 |
df_ai = pai.DataFrame(df)
|