Implement dual-mode authentication strategy

- Space demo: Uses provided API key for demonstration (limited usage)
- MCP clients: Must provide own API key for production usage
- Check SPACE_ID environment variable to detect Hugging Face Space
- Update documentation to clarify API key requirements
- Add comprehensive authentication strategy logging
- Ensure security: client API keys required for production use

Features:
✅ Space demo works with pre-configured API key
✅ MCP clients must provide their own API key
✅ Clear documentation for both usage modes
✅ Production-ready authentication strategy

README.md

@@ -96,9 +96,9 @@ This Gradio app serves as both a **web interface** and a **full MCP server**, pr
 
 ## 🔧 MCP Client Configuration
 
-### Method 1: With API Key in Headers (Recommended)
+### ⚠️ API Key Required for Client Usage
 
-Add this to your Claude Desktop configuration file:
+When using with your own MCP client (Claude Desktop, Cursor, etc.), you **must** provide your API key:
 
 ```json
 {
@@ -119,23 +119,7 @@ Add this to your Claude Desktop configuration file:
 }
 ```
 
-### Method 2: With Environment Variable (Fallback)
-
-```json
-{
-  "mcpServers": {
-    "a1d-gradio": {
-      "command": "npx",
-      "args": [
-        "mcp-remote",
-        "http://localhost:7860/gradio_api/mcp/sse"
-      ]
-    }
-  }
-}
-```
-
-*Note: Requires `A1D_API_KEY` environment variable to be set.*
+**🔑 API key is mandatory for client usage.** Get your API key at [A1D.ai](https://a1d.ai/home/api).
 
 ### For Other MCP Clients
 - **Server URL:** `http://localhost:7860/gradio_api/mcp/sse`
@@ -169,7 +153,9 @@ To deploy this on your own Hugging Face Space:
 
 ### 🔗 MCP Client Configuration for Hosted Version
 
-**Method 1: With API Key in Headers (Recommended)**
+#### ⚠️ API Key Required for Client Usage
+
+When using with your own MCP client (Claude Desktop, Cursor, etc.), you **must** provide your API key:
 
 ```json
 {
@@ -190,21 +176,11 @@ To deploy this on your own Hugging Face Space:
 }
 ```
 
-**Method 2: Environment Variable (if Space has API key configured)**
+**🔑 API key is mandatory for client usage.** Get your API key at [A1D.ai](https://a1d.ai/home/api).
 
-```json
-{
-  "mcpServers": {
-    "a1d-hosted": {
-      "command": "npx",
-      "args": [
-        "mcp-remote",
-        "https://aigchacker-a1d-mcp-server.hf.space/gradio_api/mcp/sse"
-      ]
-    }
-  }
-}
-```
+#### 🌐 Using the Hosted Demo on Hugging Face Space
+
+The hosted demo at [https://huggingface.co/spaces/aigchacker/a1d-mcp-server](https://huggingface.co/spaces/aigchacker/a1d-mcp-server) uses our provided API key for demonstration purposes only, with limited usage. For production use, please obtain your own API key.
 
 ## 💡 How to Use
 

app.py

@@ -25,14 +25,21 @@ def get_api_client():
     except:
         pass
 
-    # 2. Fallback to environment variable
-    if not api_key:
-        api_key = os.getenv("A1D_API_KEY")
+    # 2. Check if running on Hugging Face Space
+    is_space = os.getenv("SPACE_ID") is not None
+    space_api_key = os.getenv("A1D_API_KEY")
+
+    # 3. Use environment variable only if on Hugging Face Space
+    if not api_key and is_space and space_api_key:
+        print("📡 Using API key from Space environment variable")
+        return A1DAPIClient(space_api_key)
 
+    # 4. If not on Space or no API key provided, require user API key
     if not api_key:
         raise ValueError(
-            "API key is required. Set A1D_API_KEY environment variable or provide API_KEY in request headers.")
+            "API key is required. Please provide API_KEY in request headers. Get your API key at https://a1d.ai")
 
+    print("🔑 Using API key from MCP client headers")
     return A1DAPIClient(api_key)
 
 
@@ -296,13 +303,14 @@ def create_mcp_docs():
     return gr.Markdown("""
 ## 🔧 MCP Client Configuration
 
-### For Claude Desktop:
-Add this configuration to your Claude Desktop settings:
+### ⚠️ API Key Required for Client Usage
+
+When using with your own MCP client (Claude Desktop, Cursor, etc.), you **must** provide your API key:
 
 ```json
 {
   "mcpServers": {
-    "a1d-hosted": {
+    "a1d": {
       "command": "npx",
       "args": [
         "mcp-remote@latest",
@@ -318,6 +326,16 @@ Add this configuration to your Claude Desktop settings:
 }
 ```
 
+**🔑 API key is mandatory for client usage.** Get your API key at [A1D.ai](https://a1d.ai/home/api).
+
+---
+
+### 🌐 Using the Hosted Demo on Hugging Face Space
+
+The hosted demo at [https://huggingface.co/spaces/aigchacker/a1d-mcp-server](https://huggingface.co/spaces/aigchacker/a1d-mcp-server) uses our provided API key for demonstration purposes only, with limited usage. For production use, please obtain your own API key.
+
+---
+
 ### 🔑 How to Get Your A1D API Key:
 
 1. **Visit A1D Website**: Go to [https://a1d.ai](https://a1d.ai)
@@ -327,6 +345,8 @@ Add this configuration to your Claude Desktop settings:
 5. **Copy Your Key**: Copy the generated API key
 6. **Replace in Config**: Replace `your_a1d_api_key_here` with your actual API key
 
+---
+
 ### 📋 Available MCP Tools:
 - `remove_bg_wrapper` - Remove background from images
 - `image_upscaler_wrapper` - Upscale images (2x/4x/8x/16x)
@@ -338,6 +358,10 @@ Add this configuration to your Claude Desktop settings:
 ### 🌐 MCP Endpoints:
 - **SSE Endpoint**: `https://aigchacker-a1d-mcp-server.hf.space/gradio_api/mcp/sse`
 - **Schema**: `https://aigchacker-a1d-mcp-server.hf.space/gradio_api/mcp/schema`
+
+### 💡 Usage Summary:
+- **Hosted Demo**: Works directly in browser with provided API key
+- **MCP Client**: Requires your own API key for production use
 """)
 
 # Create Gradio interfaces for each tool
@@ -535,22 +559,39 @@ def create_gradio_app():
 
 
 if __name__ == "__main__":
-    # Check for API key
-    if not os.getenv("A1D_API_KEY"):
-        print("❌ Error: A1D_API_KEY environment variable is required")
-        print("Please set your API key: export A1D_API_KEY=your_api_key_here")
-        exit(1)
-
     print("🚀 Starting A1D MCP Server...")
-    print("=" * 50)
-    print("✅ API key found")
-    print(f"🎯 Title: {GRADIO_CONFIG['title']}")
+    print("=" * 70)
+
+    # Check environment and show configuration
+    is_space = os.getenv("SPACE_ID") is not None
+    space_api_key = os.getenv("A1D_API_KEY")
+
+    if is_space and space_api_key:
+        print("✅ Running on Hugging Face Space with demo API key")
+        print("🌐 Web demo: Works directly with provided API key")
+        print("🔑 MCP clients: Must provide their own API key")
+    elif is_space:
+        print("⚠️  Running on Hugging Face Space without API key")
+        print("🔑 All users must provide their own API key")
+    else:
+        print("🖥️  Running locally")
+        if space_api_key:
+            print("✅ Local API key found")
+        else:
+            print("🔑 Users must provide their own API key")
+
+    print(f"\n🎯 Title: {GRADIO_CONFIG['title']}")
     print(
         f"🌐 Server: http://{GRADIO_CONFIG['server_name']}:{GRADIO_CONFIG['server_port']}")
     print(
         f"🔧 MCP Endpoint: http://{GRADIO_CONFIG['server_name']}:{GRADIO_CONFIG['server_port']}/gradio_api/mcp/sse")
     print(f"📋 Available Tools: {len(TOOLS_CONFIG)} AI tools")
-    print("=" * 50)
+
+    print(f"\n📖 Authentication Strategy:")
+    print("   - Space demo: Uses provided API key (limited usage)")
+    print("   - MCP clients: Must provide own API key (production usage)")
+    print("   - Get API key at: https://a1d.ai")
+    print("=" * 70)
 
     # Create and launch the app
     demo = create_gradio_app()