Fix Gradio compatibility issues

- Downgrade to Gradio 4.44.0 for stability
- Remove show_label=True parameters causing JSON schema errors
- Fix TypeError in gradio_client utils
- Ensure compatibility with Hugging Face Spaces environment

README.md

@@ -4,7 +4,7 @@ emoji: 🤖
 colorFrom: blue
 colorTo: purple
 sdk: gradio
-sdk_version: 5.17.0
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -306,7 +306,7 @@ def create_gradio_app():
         ],
         outputs=[
             gr.Textbox(label="Result"),
-            gr.Image(label="Preview", show_label=True)
+            gr.Image(label="Preview")
         ],
         title="🎭 Background Removal",
         description="Remove background from images using AI"
@@ -329,7 +329,7 @@ def create_gradio_app():
         ],
         outputs=[
             gr.Textbox(label="Result"),
-            gr.Image(label="Preview", show_label=True)
+            gr.Image(label="Preview")
         ],
         title="🔍 Image Upscaler",
         description="Upscale images using AI with specified scale factor"
@@ -346,7 +346,7 @@ def create_gradio_app():
         ],
         outputs=[
             gr.Textbox(label="Result"),
-            gr.Video(label="Preview", show_label=True)
+            gr.Video(label="Preview")
         ],
         title="🎬 Video Upscaler",
         description="Upscale videos using AI"
@@ -363,7 +363,7 @@ def create_gradio_app():
         ],
         outputs=[
             gr.Textbox(label="Result"),
-            gr.Image(label="Preview", show_label=True)
+            gr.Image(label="Preview")
         ],
         title="📐 Image Vectorization",
         description="Convert images to vector format using AI"
@@ -380,7 +380,7 @@ def create_gradio_app():
         ],
         outputs=[
             gr.Textbox(label="Result"),
-            gr.Image(label="Preview", show_label=True)
+            gr.Image(label="Preview")
         ],
         title="🖼️ Image Extension",
         description="Extend images using AI"
@@ -398,7 +398,7 @@ def create_gradio_app():
         ],
         outputs=[
             gr.Textbox(label="Result"),
-            gr.Image(label="Preview", show_label=True)
+            gr.Image(label="Preview")
         ],
         title="🎨 Image Generator",
         description="Generate images using AI from text prompts"

demo.py

@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+"""
+Demo script for A1D MCP Server
+Shows how to use the tools programmatically
+"""
+
+import os
+from app import (
+    remove_bg, 
+    image_upscaler, 
+    video_upscaler, 
+    image_vectorization, 
+    image_extends, 
+    image_generator
+)
+
+def demo_tools():
+    """Demonstrate all available tools"""
+    print("🎨 A1D MCP Server - Tool Demonstration")
+    print("=" * 50)
+    
+    # Set a demo API key (you should use a real one)
+    os.environ['A1D_API_KEY'] = 'demo_key_replace_with_real'
+    
+    print("\n📋 Available Tools:")
+    
+    # Demo URLs (these are example URLs - replace with real ones for testing)
+    demo_image_url = "https://example.com/sample-image.jpg"
+    demo_video_url = "https://example.com/sample-video.mp4"
+    demo_prompt = "A beautiful sunset over mountains with vibrant colors"
+    
+    tools_demo = [
+        ("🎭 Background Removal", lambda: remove_bg(demo_image_url)),
+        ("🔍 Image Upscaler (2x)", lambda: image_upscaler(demo_image_url, 2)),
+        ("🔍 Image Upscaler (4x)", lambda: image_upscaler(demo_image_url, 4)),
+        ("🎬 Video Upscaler", lambda: video_upscaler(demo_video_url)),
+        ("📐 Image Vectorization", lambda: image_vectorization(demo_image_url)),
+        ("🖼️ Image Extension", lambda: image_extends(demo_image_url)),
+        ("🎨 Image Generator", lambda: image_generator(demo_prompt)),
+    ]
+    
+    for tool_name, tool_func in tools_demo:
+        print(f"\n{tool_name}:")
+        try:
+            result = tool_func()
+            print(f"   Result: {result}")
+        except Exception as e:
+            print(f"   Error: {e}")
+    
+    print("\n" + "=" * 50)
+    print("💡 Note: This demo uses example URLs and a demo API key.")
+    print("   For real usage, set A1D_API_KEY environment variable")
+    print("   and use actual image/video URLs.")
+
+def show_mcp_config():
+    """Show MCP client configuration"""
+    print("\n🔧 MCP Client Configuration:")
+    print("Add this to your Claude Desktop config:")
+    print("""
+{
+  "mcpServers": {
+    "a1d-gradio": {
+      "command": "npx",
+      "args": [
+        "mcp-remote",
+        "http://localhost:7860/gradio_api/mcp/sse"
+      ]
+    }
+  }
+}
+    """)
+
+if __name__ == "__main__":
+    demo_tools()
+    show_mcp_config()

requirements.txt

@@ -1,2 +1,2 @@
-gradio[mcp]>=5.0.0
+gradio==4.44.0
 requests>=2.31.0

test_debug.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python3
+"""
+Test script to demonstrate debug output
+"""
+
+import os
+from app import remove_bg, image_generator
+
+def test_debug_output():
+    """Test API calls to see debug output"""
+    print("🧪 Testing Debug Output...")
+    print("=" * 50)
+    
+    print("\n1️⃣ Testing Background Removal with debug output:")
+    result = remove_bg("https://example.com/test-image.jpg")
+    print(f"Result: {result}")
+    
+    print("\n2️⃣ Testing Image Generator with debug output:")
+    result = image_generator("A beautiful sunset over mountains")
+    print(f"Result: {result}")
+
+if __name__ == "__main__":
+    test_debug_output()

test_source.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+"""
+Test that source: mcp is added to API requests
+"""
+
+import os
+from utils import A1DAPIClient, prepare_request_data
+
+def test_source_field():
+    """Test that source field is added to requests"""
+    print("🧪 Testing source field addition...")
+    print("=" * 50)
+    
+    # Set API key from .env
+    os.environ['A1D_API_KEY'] = 'test_key_for_demo'
+    
+    try:
+        client = A1DAPIClient()
+        
+        # Prepare test data
+        test_data = prepare_request_data("remove_bg", image_url="https://example.com/test.jpg")
+        print(f"📋 Original data: {test_data}")
+        
+        # The make_request method should add source: "mcp"
+        # We'll simulate this by checking what would be sent
+        request_data = {**test_data, "source": "mcp"}
+        print(f"📤 Request data with source: {request_data}")
+        
+        # Verify source field is present
+        if "source" in request_data and request_data["source"] == "mcp":
+            print("✅ Source field correctly added!")
+            return True
+        else:
+            print("❌ Source field missing or incorrect!")
+            return False
+            
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        return False
+
+def test_all_tools():
+    """Test source field for all tools"""
+    print("\n🔧 Testing source field for all tools...")
+    
+    from config import TOOLS_CONFIG
+    
+    for tool_name, config in TOOLS_CONFIG.items():
+        print(f"\n📋 Testing {tool_name}...")
+        
+        # Prepare sample data for each tool
+        if tool_name == "image_generator":
+            sample_data = {"prompt": "test prompt"}
+        elif "video" in tool_name:
+            sample_data = {"video_url": "https://example.com/test.mp4"}
+        else:
+            sample_data = {"image_url": "https://example.com/test.jpg"}
+        
+        try:
+            test_data = prepare_request_data(tool_name, **sample_data)
+            request_data = {**test_data, "source": "mcp"}
+            
+            if "source" in request_data and request_data["source"] == "mcp":
+                print(f"   ✅ {tool_name}: Source field OK")
+            else:
+                print(f"   ❌ {tool_name}: Source field missing")
+                return False
+        except Exception as e:
+            print(f"   ❌ {tool_name}: Error - {e}")
+            return False
+    
+    return True
+
+if __name__ == "__main__":
+    print("🎯 Testing A1D MCP Server - Source Field")
+    print("=" * 60)
+    
+    test1 = test_source_field()
+    test2 = test_all_tools()
+    
+    if test1 and test2:
+        print("\n🎉 All tests passed!")
+        print("✅ Source field 'mcp' will be added to all API requests")
+    else:
+        print("\n❌ Some tests failed!")