update

GraphRouter_eval/run_exp.py

@@ -3,11 +3,11 @@ import argparse
 import sys
 import os
 
-# Load environment variables from .env file (if it exists)
+# Load environment variables from .env file (for local development only)
 try:
     from dotenv import load_dotenv
     load_dotenv()
-    print("✅ .env file loaded successfully")
+    print("✅ .env file loaded successfully (local development)")
 except ImportError:
     print("Warning: python-dotenv not installed. Install with: pip install python-dotenv")
     print("Or set NVIDIA_API_KEY environment variable manually")
@@ -26,18 +26,19 @@ parser.add_argument("--query", type=str, default="What is the derivative of f(x)
                     help="Input query to process")
 args = parser.parse_args()
 
-# Load API key from environment variable
-NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY")
-
-if NVIDIA_API_KEY is None:
+# Check for API key
+if os.getenv("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")
+    print("⚠️  Exiting due to missing API key")
     sys.exit(1)
 
+print("✅ NVIDIA API key loaded from environment")
+
 client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
-    api_key=NVIDIA_API_KEY,
+    api_key=os.getenv("NVIDIA_API_KEY"),
     timeout=60,
     max_retries=2
 )

app.py

@@ -24,10 +24,13 @@ def setup_environment():
     if os.getenv("SPACE_ID"):
         print("🚀 Running on Hugging Face Spaces")
         
-        # Set default values for required environment variables
+        # Check for NVIDIA API key
         if not os.getenv("NVIDIA_API_KEY"):
-            print("⚠️  NVIDIA_API_KEY not set. Some features may be limited.")
-            print("   Please set NVIDIA_API_KEY in the Space settings.")
+            print("⚠️  NVIDIA_API_KEY not set in Space secrets.")
+            print("   Please set NVIDIA_API_KEY in the Space Repository Secrets.")
+            print("   Some features may be limited without API access.")
+        else:
+            print("✅ NVIDIA_API_KEY found in Space secrets")
         
         # Set CUDA device for Spaces (usually limited resources)
         os.environ["CUDA_VISIBLE_DEVICES"] = "0"
@@ -37,6 +40,10 @@ def setup_environment():
         
     else:
         print("🏠 Running locally")
+        # Check for local .env file
+        if not os.getenv("NVIDIA_API_KEY"):
+            print("ℹ️  NVIDIA_API_KEY not found. For local development, create a .env file")
+            print("   or set the environment variable manually.")
 
 def main():
     """Main function to launch the application"""

demo.py

@@ -43,28 +43,28 @@ import sys
 import yaml
 from transformers import LongformerTokenizer as RouterTokenizer, LongformerModel as RouterModel
 
-# Load environment variables from .env file
+# Load environment variables from .env file (for local development only)
 try:
     from dotenv import load_dotenv
     load_dotenv()
-    print("✅ .env file loaded successfully")
+    print("✅ .env file loaded successfully (local development)")
 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")
-NVIDIA_BASE_URL = "https://integrate.api.nvidia.com/v1"
-
-if NVIDIA_API_KEY is None:
+# Check for API key
+if os.getenv("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")
+    print("⚠️  Some features will be limited without API access")
 else:
     print("✅ NVIDIA API key loaded from environment")
 
+NVIDIA_BASE_URL = "https://integrate.api.nvidia.com/v1"
+
 # Add GraphRouter_eval to path
 sys.path.append(os.path.join(os.path.dirname(__file__), 'GraphRouter_eval/model'))
 sys.path.append(os.path.join(os.path.dirname(__file__), 'GraphRouter_eval/data_processing'))
@@ -87,13 +87,13 @@ if torch.cuda.is_available():
     print(f"CUDA device name: {torch.cuda.get_device_name(0)}")
 
 # Initialize OpenAI client for NVIDIA API
-if NVIDIA_API_KEY is None:
+if os.getenv("NVIDIA_API_KEY") is None:
     print("❌ NVIDIA API key not found. Please create a .env file with your API key")
     client = None
 else:
     client = OpenAI(
         base_url=NVIDIA_BASE_URL,
-        api_key=NVIDIA_API_KEY,
+        api_key=os.getenv("NVIDIA_API_KEY"),
         timeout=60,
         max_retries=2
     )