Update app.py

app.py

@@ -54,23 +54,29 @@ def get_client():
     # Try multiple token sources for Spaces
     api_token = None
     
-    # Method 1: Spaces secrets (recommended)
+    # Method 1: Check HF_API_TOKEN (your current setup)
     api_token = os.getenv("HF_API_TOKEN")
+    if api_token:
+        logger.info("✅ Found HF_API_TOKEN")
     
-    # Method 2: Traditional environment variable
+    # Method 2: Fallback to HF_TOKEN 
     if not api_token:
-        api_token = os.getenv("HF_API_TOKEN")
+        api_token = os.getenv("HF_TOKEN")
+        if api_token:
+            logger.info("✅ Found HF_TOKEN")
     
     # Method 3: Hugging Face CLI token (if logged in)
     if not api_token:
         try:
             from huggingface_hub import get_token
             api_token = get_token()
+            if api_token:
+                logger.info("✅ Found CLI token")
         except:
             pass
     
     if not api_token:
-        logger.warning("⚠️ No HF token found. Some features may be limited.")
+        logger.warning("⚠️ No HF token found. Please set HF_API_TOKEN or HF_TOKEN in Spaces secrets.")
         return None
     
     try:
@@ -98,7 +104,7 @@ CLIENT = get_client()
 def safe_call(func_name: str, func, *args, **kwargs):
     """Spaces-optimized safe API calls."""
     if not CLIENT:
-        return "❌ API client not available. Please check if HF_TOKEN is set in Spaces secrets."
+        return "❌ API client not available. Please check if HF_API_TOKEN or HF_TOKEN is set in Spaces secrets."
     
     try:
         logger.info(f"🔄 {func_name}...")
@@ -449,7 +455,7 @@ with gr.Blocks(title="🚀 AI Research Hub", theme=gr.themes.Soft(), css=custom_
     else:
         gr.HTML("""
         <div class="status-indicator status-error">
-            ❌ <strong>Status:</strong> Please set HF_TOKEN in Spaces settings
+            ❌ <strong>Status:</strong> Please set HF_API_TOKEN or HF_TOKEN in Spaces settings
         </div>
         """)
     
@@ -613,7 +619,7 @@ with gr.Blocks(title="🚀 AI Research Hub", theme=gr.themes.Soft(), css=custom_
     
     **To use this Space:**
     1. Go to **Settings** → **Repository secrets**
-    2. Add **HF_TOKEN** with your Hugging Face token
+    2. Add **HF_API_TOKEN** (or **HF_TOKEN**) with your Hugging Face token
     3. Get your token at: [https://huggingface.co/settings/tokens](https://huggingface.co/settings/tokens)
     
     **Features:**