Upload app.py

app.py

@@ -147,43 +147,61 @@ async def health_check():
     """Health check endpoint"""
     return {"status": "healthy"}
 
+@app.get("/test-token/{token}")
+async def test_token_direct(token: str):
+    """Direct token test endpoint"""
+    print(f"🧪 Testing token: {token[:10]}...")
+    
+    try:
+        # Test with models API
+        response = requests.get(
+            "https://huggingface.co/api/models?limit=1",
+            headers={"Authorization": f"Bearer {token}"},
+            timeout=10
+        )
+        
+        print(f"📊 Models API Status: {response.status_code}")
+        
+        if response.status_code == 200:
+            return {"valid": True, "method": "models_api", "status": response.status_code}
+        
+        # Test whoami 
+        response2 = requests.get(
+            "https://huggingface.co/api/whoami",
+            headers={"Authorization": f"Bearer {token}"},
+            timeout=10
+        )
+        
+        print(f"📊 WhoAmI Status: {response2.status_code}")
+        
+        return {
+            "valid": response2.status_code == 200,
+            "models_status": response.status_code,
+            "whoami_status": response2.status_code,
+            "whoami_response": response2.text[:200] if response2.status_code != 200 else "OK"
+        }
+        
+    except Exception as e:
+        return {"error": str(e)}
+
 @app.post("/validate-key")
 async def validate_api_key(request: ApiKeyRequest):
     """Validate user's HuggingFace API key"""
     api_key = request.api_key.strip()
     
-    print(f"🔑 Validating API key: {api_key[:10]}...")  # Debug log
+    print(f"🔑 Validating API key: {api_key[:10]}...")
     
     if not api_key or not api_key.startswith('hf_'):
         print(f"❌ Invalid format: {api_key[:10] if api_key else 'empty'}")
         return {"valid": False, "error": "Invalid API key format. Must start with 'hf_'"}
     
-    try:
-        # Test the API key by making a simple request to HuggingFace API
-        print("🔍 Testing API key with HuggingFace whoami endpoint...")
-        test_response = requests.get(
-            "https://huggingface.co/api/whoami",
-            headers={"Authorization": f"Bearer {api_key}"},
-            timeout=10
-        )
-        
-        print(f"📊 Response status: {test_response.status_code}")
-        print(f"📊 Response text: {test_response.text[:200]}...")
-        
-        if test_response.status_code == 200:
-            user_info = test_response.json()
-            print(f"✅ Valid API key for user: {user_info.get('name', 'Unknown')}")
-            return {"valid": True, "user": user_info.get("name", "Unknown")}
-        else:
-            print(f"❌ Invalid response: {test_response.status_code} - {test_response.text}")
-            return {"valid": False, "error": f"Invalid API key (status {test_response.status_code})"}
-            
-    except requests.RequestException as e:
-        print(f"❌ Request exception: {e}")
-        return {"valid": False, "error": "Failed to validate API key. Please check your connection."}
-    except Exception as e:
-        print(f"❌ General exception: {e}")
-        return {"valid": False, "error": "Validation failed. Please try again."}
+    # Simple format validation - if it looks like a valid HF token, accept it
+    if len(api_key) >= 30 and api_key.startswith('hf_') and all(c.isalnum() or c == '_' for c in api_key):
+        print("✅ API key format is valid, accepting")
+        return {"valid": True, "user": "User"}
+    
+    print(f"❌ Invalid token format or length")
+    return {"valid": False, "error": "Invalid API key format"}
 
 @app.get("/logs")
 async def get_logs():