update

app.py

@@ -33,7 +33,17 @@ def setup_environment():
         else:
             print("✅ NVIDIA_API_KEY found in Space secrets")
             print(f"   Key length: {len(api_key)} characters")
-            print(f"   Key preview: {api_key[:10]}...{api_key[-4:] if len(api_key) > 14 else '***'}")
+            print(f"   Key starts with: {api_key[:8]}...")
+            print(f"   Key ends with: ...{api_key[-4:]}")
+            print(f"   Key is all uppercase: {'Yes' if api_key.isupper() else 'No'}")
+            
+            # Check for common API key patterns
+            if api_key.startswith('nvapi-'):
+                print("   ✅ Key format appears correct (starts with 'nvapi-')")
+            elif api_key.startswith('nv-'):
+                print("   ✅ Key format appears correct (starts with 'nv-')")
+            else:
+                print("   ⚠️  Key format may be incorrect (should start with 'nvapi-' or 'nv-')")
         
         # Set CUDA device for Spaces (usually limited resources)
         os.environ["CUDA_VISIBLE_DEVICES"] = "0"
@@ -50,6 +60,17 @@ def setup_environment():
             print("   or set the environment variable manually.")
         else:
             print(f"✅ NVIDIA_API_KEY found locally (length: {len(api_key)})")
+            print(f"   Key starts with: {api_key[:8]}...")
+            print(f"   Key ends with: ...{api_key[-4:]}")
+            print(f"   Key is all uppercase: {'Yes' if api_key.isupper() else 'No'}")
+            
+            # Check for common API key patterns
+            if api_key.startswith('nvapi-'):
+                print("   ✅ Key format appears correct (starts with 'nvapi-')")
+            elif api_key.startswith('nv-'):
+                print("   ✅ Key format appears correct (starts with 'nv-')")
+            else:
+                print("   ⚠️  Key format may be incorrect (should start with 'nvapi-' or 'nv-')")
 
 def main():
     """Main function to launch the application"""

demo.py

@@ -105,6 +105,23 @@ def initialize_nvidia_client():
         print("   For Hugging Face Spaces: Set NVIDIA_API_KEY in Repository Secrets")
         return None
     else:
+        # Validate the API key format
+        print("🔍 Validating API key format...")
+        validation = validate_api_key(api_key)
+        
+        if not validation['valid']:
+            print("❌ API key validation failed:")
+            for issue in validation['issues']:
+                print(f"   - {issue}")
+            print(f"   Key preview: {api_key[:8]}...{api_key[-4:]}")
+            print("   Please check your API key format and try again.")
+            return None
+        
+        print("✅ API key format validation passed")
+        print(f"   Key format: {'✅' if validation['format_ok'] else '❌'}")
+        print(f"   Key length: {'✅' if validation['length_ok'] else '❌'}")
+        print(f"   Key charset: {'✅' if validation['charset_ok'] else '❌'}")
+        
         try:
             client = OpenAI(
                 base_url=NVIDIA_BASE_URL,
@@ -118,6 +135,50 @@ def initialize_nvidia_client():
             print(f"❌ Failed to initialize NVIDIA API client: {e}")
             return None
 
+def validate_api_key(api_key: str) -> dict:
+    """Validate the NVIDIA API key format and return detailed analysis"""
+    analysis = {
+        'valid': True,
+        'issues': [],
+        'format_ok': False,
+        'length_ok': False,
+        'charset_ok': False
+    }
+    
+    if not api_key:
+        analysis['valid'] = False
+        analysis['issues'].append("API key is empty or None")
+        return analysis
+    
+    # Check length (NVIDIA API keys are typically 40-50 characters)
+    if len(api_key) < 30 or len(api_key) > 60:
+        analysis['length_ok'] = False
+        analysis['issues'].append(f"Key length ({len(api_key)}) seems unusual (expected 30-60 chars)")
+    else:
+        analysis['length_ok'] = True
+    
+    # Check format (should start with nvapi- or nv-)
+    if not (api_key.startswith('nvapi-') or api_key.startswith('nv-')):
+        analysis['format_ok'] = False
+        analysis['issues'].append("Key should start with 'nvapi-' or 'nv-'")
+    else:
+        analysis['format_ok'] = True
+    
+    # Check for whitespace or special characters
+    if any(c.isspace() for c in api_key):
+        analysis['charset_ok'] = False
+        analysis['issues'].append("Key contains whitespace characters")
+    elif not all(c.isalnum() or c in '-_' for c in api_key):
+        analysis['charset_ok'] = False
+        analysis['issues'].append("Key contains invalid characters")
+    else:
+        analysis['charset_ok'] = True
+    
+    # Overall validity
+    analysis['valid'] = analysis['format_ok'] and analysis['length_ok'] and analysis['charset_ok']
+    
+    return analysis
+
 # Initialize the client
 client = initialize_nvidia_client()
 
@@ -171,7 +232,17 @@ def model_prompting(
     # Debug information
     api_key = os.getenv("NVIDIA_API_KEY")
     if api_key:
-        print(f"🔑 API Key available: {api_key[:10]}...{api_key[-4:] if len(api_key) > 14 else '***'}")
+        print(f"🔑 API Key available: {api_key[:8]}...{api_key[-4:]}")
+        print(f"   Key length: {len(api_key)} characters")
+        
+        # Validate the key
+        validation = validate_api_key(api_key)
+        if validation['valid']:
+            print("   ✅ Key format validation passed")
+        else:
+            print("   ❌ Key format validation failed:")
+            for issue in validation['issues']:
+                print(f"      - {issue}")
     else:
         print("❌ No API key found in environment")