update

.env

@@ -1 +0,0 @@
-NVIDIA_API_KEY=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

.gitignore

@@ -21,10 +21,15 @@ wheels/
 *.egg
 MANIFEST
 
-# Environment variables
+# Environment variables and secrets
 .env
 .env.local
 .env.*.local
+.env.production
+.env.development
+*.env
+secrets.txt
+api_keys.txt
 
 # IDE
 .vscode/
@@ -49,8 +54,6 @@ Thumbs.db
 *.pth
 *.pt
 *.ckpt
-*.env
-.env
 
 # Logs
 *.log

GraphRouter_eval/run_exp.py

@@ -2,8 +2,17 @@ import yaml
 import argparse
 import sys
 import os
-from dotenv import load_dotenv
-load_dotenv()
+
+# Load environment variables from .env file (if it exists)
+try:
+    from dotenv import load_dotenv
+    load_dotenv()
+    print("✅ .env file loaded successfully")
+except ImportError:
+    print("Warning: python-dotenv not installed. Install with: pip install python-dotenv")
+    print("Or set NVIDIA_API_KEY environment variable manually")
+except FileNotFoundError:
+    print("ℹ️  No .env file found - using environment variables directly")
 
 sys.path.append(os.path.join(os.path.dirname(__file__), 'model'))
 from model.multi_task_graph_router import graph_router_prediction
@@ -18,7 +27,13 @@ parser.add_argument("--query", type=str, default="What is the derivative of f(x)
 args = parser.parse_args()
 
 # Load API key from environment variable
-NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY", "")
+NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY")
+
+if NVIDIA_API_KEY is None:
+    print("❌ NVIDIA_API_KEY not found in environment variables")
+    print("For local development: Create a .env file with: NVIDIA_API_KEY=your_api_key_here")
+    print("For Hugging Face Spaces: Set NVIDIA_API_KEY in Repository Secrets")
+    sys.exit(1)
 
 client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",

demo.py

@@ -51,6 +51,8 @@ try:
 except ImportError:
     print("Warning: python-dotenv not installed. Install with: pip install python-dotenv")
     print("Or set NVIDIA_API_KEY environment variable manually")
+except FileNotFoundError:
+    print("ℹ️  No .env file found - using environment variables directly")
 
 # Get API key from environment variable
 NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY")
@@ -58,7 +60,8 @@ NVIDIA_BASE_URL = "https://integrate.api.nvidia.com/v1"
 
 if NVIDIA_API_KEY is None:
     print("❌ NVIDIA_API_KEY not found in environment variables")
-    print("Please create a .env file with: NVIDIA_API_KEY=your_api_key_here")
+    print("For local development: Create a .env file with: NVIDIA_API_KEY=your_api_key_here")
+    print("For Hugging Face Spaces: Set NVIDIA_API_KEY in Repository Secrets")
 else:
     print("✅ NVIDIA API key loaded from environment")