feat: remove redundant code and improve visualization tool

langchain_mcp_client.py

@@ -17,7 +17,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-set_debug(True)
+# set_debug(True)
 
 
 # Set up logging

postgre_mcp_server.py

@@ -624,15 +624,17 @@ async def visualize_results(json_data: dict, vis_prompt: str) -> str:
             pllm = OpenAI(api_token=OPENAI_API_KEY)
             pai.config.set({"llm": pllm})
 
+        # =============== DEPRECATED CODE BLOCK==============================
         # Convert JSON to DataFrame
-        df = pd.DataFrame(json_data["data"], columns=json_data["columns"])
+        # df = pd.DataFrame(json_data["data"], columns=json_data["columns"])
         
         # Shorten long values in text columns
-        for column in df.select_dtypes(include=['object']).columns:
-            df[column] = df[column].apply(lambda x: str(x)[:20] + '...' if len(str(x)) > 20 else str(x))
+        # for column in df.select_dtypes(include=['object']).columns:
+        #     df[column] = df[column].apply(lambda x: str(x)[:20] + '...' if len(str(x)) > 20 else str(x))
+        # =============== END OF CODE BLOCK=====================================
 
         # Initialize PandasAI
-        df_ai = pai.DataFrame(df)
+        df_ai = pai.DataFrame(data=json_data["data"], columns=json_data["columns"])
 
         api_key = os.environ["PANDAS_KEY"]
         pai.api_key.set(api_key)