fix: improve shell script for vscode compatability

app.py

@@ -10,11 +10,10 @@ import base64
 from memory_store import MemoryStore
 import logging
 
-# load_dotenv()
-# dsn = os.environ["DB_URL"]
-# schema = os.environ["DB_SCHEMA"]
-#
-# print("DEBUG: DB_URL =", os.environ["DB_URL"])
+
+# Load environment variables
+load_dotenv()
+
 
 # ======================================= Load DB configs
 def load_db_configs():

postgre_mcp_server.py

@@ -11,6 +11,7 @@ import matplotlib as plt
 import pandas as pd
 import logging
 
+
 # Constants
 DEFAULT_QUERY_LIMIT = 100
 
@@ -20,6 +21,9 @@ DEFAULT_QUERY_LIMIT = 100
 # get logger
 logger = logging.getLogger(__name__)
 
+# Load environment variables
+load_dotenv()
+
 # Define our own PromptMessage class if the MCP one isn't available
 @dataclass
 class PromptMessage:
@@ -45,8 +49,6 @@ async def db_lifespan(server: FastMCP) -> AsyncIterator[DbContext]:
     load_dotenv()
     dsn = os.environ["DB_URL"]
     schema = os.environ["DB_SCHEMA"]
-    
-    print("DEBUG: DB_URL =", os.environ["DB_URL"])
 
     pool = await asyncpg.create_pool(
         dsn,

run.sh

@@ -1,6 +1,11 @@
 #!/bin/bash
 
 # Replace 'myenv' with the name of your conda environment
-# conda activate myenv
+# conda deactivate
+# conda activate your_env_name
 
-python gradio_app.py
+if [ -f .env ]; then
+         export $(grep -v '^#' .env | xargs)
+       fi
+    #    echo "Starting Gradio app with DB_URL: $DB_URL"
+       python app.py