Fix: Add HF token support and error handling for API authentication

HF_TOKEN_SETUP.md

@@ -0,0 +1,58 @@
+# 🔑 How to Add HuggingFace Token to Your Space
+
+The AlgoSpeak AI app needs a HuggingFace token to access the Mistral-7B model for better translation quality.
+
+## Why do I need this?
+
+- **Without token**: Uses Google Flan-T5 (smaller, limited quality)
+- **With token**: Uses Mistral-7B (better understanding, higher quality)
+
+## Step-by-Step Instructions
+
+### 1. Create a HuggingFace Access Token
+
+1. Go to [HuggingFace Settings → Tokens](https://huggingface.co/settings/tokens)
+2. Click **"New token"**
+3. Give it a name like `algospeak-space`
+4. Select permission: **Read** (sufficient for inference)
+5. Click **"Generate"**
+6. **Copy the token** (you won't see it again!)
+
+### 2. Add Token as Secret to Your Space
+
+1. Go to your Space: https://huggingface.co/spaces/Veronyka/algospeak
+2. Click on **"Settings"** tab (⚙️)
+3. Scroll to **"Repository secrets"** section
+4. Click **"New secret"**
+5. Fill in:
+   - **Name:** `HF_TOKEN`
+   - **Value:** Paste your token here
+6. Click **"Add secret"**
+
+### 3. Restart Your Space
+
+1. Go to **"Settings"** → **"Factory reboot"**
+2. Or just wait ~1 minute for automatic restart
+
+### 4. Verify It's Working
+
+- The app header should now show: `🤖 Current Model: mistralai/Mistral-7B-Instruct-v0.3`
+- Translations should be more contextual and higher quality
+
+## Troubleshooting
+
+### Still seeing errors?
+
+- Check if the secret name is exactly `HF_TOKEN` (case-sensitive)
+- Make sure your token has "Read" permission
+- Try Factory reboot in Settings
+
+### Token not being used?
+
+- The app will automatically detect the token via environment variable
+- No code changes needed!
+
+---
+
+**Security Note:** Secrets in HF Spaces are encrypted and not visible to visitors. Only the Space owner can see/edit them.
+

app.py

@@ -1,13 +1,18 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 from algospeak_dictionary import get_algospeak_context, ALGOSPEAK_DICT
+import os
 
-# Initialize inference client
-client = InferenceClient()
+# Initialize inference client with HF token (automatically provided in Spaces)
+hf_token = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN")
+client = InferenceClient(token=hf_token)
 
 # Complete AlgoSpeak dictionary for prompts
 ALGOSPEAK_EXAMPLES = get_algospeak_context()
 
+# Model selection based on token availability
+MODEL = "mistralai/Mistral-7B-Instruct-v0.3" if hf_token else "google/flan-t5-large"
+
 def translate_to_algospeak(text):
     """Translates normal text to AlgoSpeak"""
     prompt = f"""You are an expert in AlgoSpeak - coded language used to circumvent algorithmic moderation on social media.
@@ -24,13 +29,16 @@ Original text: {text}
 
 Translation to AlgoSpeak:"""
 
-    response = client.text_generation(
-        prompt,
-        model="mistralai/Mistral-7B-Instruct-v0.3",
-        max_new_tokens=250,
-        temperature=0.7
-    )
-    return response
+    try:
+        response = client.text_generation(
+            prompt,
+            model=MODEL,
+            max_new_tokens=250,
+            temperature=0.7
+        )
+        return response
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}\n\nTip: The Space may need a HuggingFace token. Contact the Space owner."
 
 def interpret_algospeak(text):
     """Interprets AlgoSpeak to plain language"""
@@ -48,13 +56,16 @@ AlgoSpeak text: {text}
 
 Interpretation in plain language:"""
 
-    response = client.text_generation(
-        prompt,
-        model="mistralai/Mistral-7B-Instruct-v0.3",
-        max_new_tokens=250,
-        temperature=0.3
-    )
-    return response
+    try:
+        response = client.text_generation(
+            prompt,
+            model=MODEL,
+            max_new_tokens=250,
+            temperature=0.3
+        )
+        return response
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}\n\nTip: The Space may need a HuggingFace token. Contact the Space owner."
 
 def search_dictionary(query):
     """Search terms in the AlgoSpeak dictionary"""
@@ -72,7 +83,7 @@ def search_dictionary(query):
 
 # Interface Gradio
 with gr.Blocks(theme=gr.themes.Soft(), title="AlgoSpeak AI") as demo:
-    gr.Markdown("""
+    gr.Markdown(f"""
     # 🗣️ AlgoSpeak AI
     
     **AlgoSpeak** is a language used to circumvent content moderation algorithms on social media platforms.
@@ -80,7 +91,9 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AlgoSpeak AI") as demo:
     This AI can:
     - 📝 Translate plain text → AlgoSpeak
     - 🔍 Interpret AlgoSpeak → plain language
-    - 📖 Search through 50+ catalogued terms
+    - 📖 Search through 60+ catalogued terms
+    
+    🤖 **Current Model:** `{MODEL}`
     """)
     
     with gr.Tab("🌐 Translate to AlgoSpeak"):