Update app.py

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 import os
+import requests
 
 # Try multiple ways to get the token
 HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HF_API_TOKEN") or os.getenv("HUGGING_FACE_HUB_TOKEN")
 
+# Debug: Check if token exists (don't print the actual token!)
+TOKEN_STATUS = "✅ Token found" if HF_TOKEN else "❌ No token found"
+
 def generate_output(project_name, user_input, file_path=None):
     """Generates specialized output using the LLM based on the project context."""
     
@@ -51,13 +55,13 @@ def generate_output(project_name, user_input, file_path=None):
 
     # Call the HF Inference API using requests
     try:
-        import requests
-        
         API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
         
-        headers = {}
+        headers = {"Content-Type": "application/json"}
         if HF_TOKEN:
             headers["Authorization"] = f"Bearer {HF_TOKEN}"
+        else:
+            return "❌ ERROR: No HF_TOKEN found!\n\nPlease add your Hugging Face token:\n1. Go to Space Settings\n2. Click 'Variables and secrets'\n3. Add new secret: Name='HF_TOKEN', Value=your token from https://huggingface.co/settings/tokens"
         
         # Format the prompt for Mistral
         prompt = f"<s>[INST] {system_prompt}\n\n{user_input} [/INST]"
@@ -78,18 +82,18 @@ def generate_output(project_name, user_input, file_path=None):
             result = response.json()
             if isinstance(result, list) and len(result) > 0:
                 generated_text = result[0].get('generated_text', '').strip()
-                return generated_text
+                return generated_text if generated_text else "Error: Empty response from model"
             else:
-                return "Error: Unexpected response format from API"
+                return f"Error: Unexpected response format: {result}"
         elif response.status_code == 503:
-            return "⏳ Model is loading... Please wait 20-30 seconds and try again."
-        elif response.status_code == 401:
-            return "⚠️ Authentication Error: Your HF_TOKEN might be invalid. Please check it in Space Settings > Variables and secrets."
+            return "⏳ Model is loading... This can take 20-60 seconds on first use. Please try again in a moment."
+        elif response.status_code == 401 or response.status_code == 403:
+            return f"⚠️ Authentication Error (Code {response.status_code}): Your HF_TOKEN is invalid or expired.\n\n1. Get a new token from https://huggingface.co/settings/tokens\n2. Make sure to select 'Read' permission\n3. Update the HF_TOKEN secret in Space Settings"
         else:
-            return f"⚠️ API Error (Code {response.status_code}): {response.text}"
+            return f"⚠️ API Error (Code {response.status_code}):\n{response.text[:500]}"
 
-    except ImportError:
-        return "ERROR: 'requests' library not found. Please add 'requests' to requirements.txt"
+    except requests.exceptions.Timeout:
+        return "⏳ Request timed out. The model might be loading. Please try again in 30 seconds."
     except Exception as e:
         return f"ERROR: {str(e)}"
 
@@ -99,8 +103,18 @@ with gr.Blocks(title="VYZ Interstellar Multi-Tool AI", theme=gr.themes.Soft()) a
     gr.Markdown("# 🚀 VYZ Interstellar AI Assistant")
     gr.Markdown("A specialized, multi-tool assistant for VYZVERSE, ReVYZGen, and VYZENVO projects. **(Powered by Mistral 7B via HF Inference API)**")
     
+    # Show token status
+    gr.Markdown(f"**Status:** {TOKEN_STATUS}")
+    
     if not HF_TOKEN:
-        gr.Markdown("### ⚠️ Warning: No HF_TOKEN found. Add it in Settings > Variables and secrets > New secret (Name: HF_TOKEN)")
+        gr.Markdown("""
+        ### ⚠️ Setup Required:
+        1. Go to https://huggingface.co/settings/tokens
+        2. Create a new token with 'Read' permission
+        3. Go to your Space Settings → Variables and secrets
+        4. Add new secret: Name = `HF_TOKEN`, Value = your token
+        5. Restart the Space
+        """)
 
     with gr.Tabs() as tabs:
         # TAB 1: VYZVERSE (Animation/Lore Generation)