AgentPay integrated

.gitignore

@@ -3,3 +3,4 @@ _context/
 .DS_Store
 .env
 .gradio/
+__pycache__/

agentpay_sdk/__init__.py

@@ -0,0 +1,5 @@
+# Expose the client class at the package level
+from .client import AgentPayClient
+
+# Optionally define __all__ to control wildcard imports
+__all__ = ["AgentPayClient"] 

agentpay_sdk/client.py

@@ -0,0 +1,225 @@
+# Simple client to interact with AgentHub API 
+
+import requests
+from typing import Optional, Dict, Any
+import json # For decoding JSON response
+
+from .types import ConsumptionResult
+from .exceptions import AgentPayAPIError, AgentPayConnectionError # Import custom exceptions
+
+API_BASE_URL = "https://api.agentpay.me"
+API_VERSION_PREFIX = "/v1"
+
+class AgentPayClient:
+    """Client for interacting with the AgentPay API.
+
+    Handles authentication using a Service Token and provides methods
+    for validating user API keys and consuming balance.
+    """
+
+    def __init__(
+        self,
+        service_token: str,
+        api_url: Optional[str] = None,
+    ):
+        """Initializes the AgentPay Client.
+
+        Args:
+            service_token: The unique token issued to your service by AgentHub.
+            api_url: Optional. Manually override the base API URL.
+        """
+        if not service_token:
+            raise ValueError("service_token is required.")
+        
+        self.service_token = service_token
+        
+        if api_url:
+            # Use manual override if provided
+            self.base_api_url = api_url.rstrip('/')
+        else:
+            self.base_api_url = API_BASE_URL
+            
+        # Add the /api/v1 prefix
+        self.base_api_url += API_VERSION_PREFIX
+        
+        # Prepare headers for requests
+        self.headers = {
+            "Authorization": f"Bearer {self.service_token}",
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+        } 
+
+    def _request(self, method: str, endpoint: str, data: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """Internal helper to make requests to the AgentPay API.
+
+        Args:
+            method: HTTP method (e.g., "POST", "GET").
+            endpoint: API endpoint path (e.g., "/balances/consume").
+            data: Optional dictionary payload for POST/PUT/PATCH requests.
+
+        Returns:
+            The JSON response body as a dictionary.
+
+        Raises:
+            AgentPayConnectionError: If a network error occurs.
+            AgentPayAPIError: If the API returns an error status code.
+        """
+        full_url = self.base_api_url + endpoint
+        
+        try:
+            response = requests.request(
+                method=method,
+                url=full_url,
+                headers=self.headers,
+                json=data, # requests library handles JSON serialization
+                timeout=10 # Add a reasonable timeout (e.g., 10 seconds)
+            )
+            
+            # Raise exception for bad status codes (4xx or 5xx)
+            response.raise_for_status()
+
+            # Attempt to parse successful response as JSON
+            # Handle cases where response might be empty (e.g., 204 No Content)
+            if response.status_code == 204:
+                return {} # Return empty dict for No Content
+            
+            return response.json()
+        
+        except requests.exceptions.Timeout as e:
+            raise AgentPayConnectionError(f"Request timed out connecting to {full_url}: {e}") from e
+        except requests.exceptions.ConnectionError as e:
+             raise AgentPayConnectionError(f"Could not connect to {full_url}: {e}") from e
+        except requests.exceptions.HTTPError as e:
+            # Handle API errors (4xx, 5xx)
+            status_code = e.response.status_code
+            error_code = None
+            error_message = None
+            try:
+                # Attempt to parse error details from response body
+                error_data = e.response.json()
+                error_code = error_data.get("error_code")
+                error_message = error_data.get("error_message") or error_data.get("detail") # FastAPI uses detail sometimes
+            except json.JSONDecodeError:
+                # If response body is not JSON or empty
+                error_message = e.response.text or f"HTTP error {status_code}"
+                
+            raise AgentPayAPIError(
+                status_code=status_code, 
+                error_code=error_code, 
+                error_message=error_message
+            ) from e
+        except json.JSONDecodeError as e:
+             # Handle cases where a 2xx response is not valid JSON
+             raise AgentPayAPIError(
+                 status_code=response.status_code, # Use the status code from the original response
+                 error_message=f"Failed to decode successful JSON response: {e}"
+            ) from e
+        except requests.exceptions.RequestException as e:
+            # Catch any other requests-related errors
+            raise AgentPayConnectionError(f"An unexpected network error occurred: {e}") from e 
+
+    def consume(
+        self,
+        api_key: str,
+        amount_cents: int,
+        usage_event_id: str,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> ConsumptionResult:
+        """Consumes balance for a user associated with an API key.
+
+        Args:
+            api_key: The User's API key to consume balance against.
+            amount_cents: The amount to consume in cents (must be >= 0).
+            usage_event_id: A unique identifier (e.g., UUID string) for this specific 
+                            consumption attempt, used for idempotency.
+            metadata: Optional dictionary containing additional data to associate
+                      with the usage event.
+
+        Returns:
+            A ConsumptionResult object indicating success or failure.
+        """
+        if amount_cents < 0:
+            # Prevent making an API call with an invalid amount
+            # The API would reject this anyway, but better to catch early.
+            return ConsumptionResult(
+                success=False,
+                error_code="INVALID_INPUT",
+                error_message="Consumption amount cannot be negative."
+            )
+            
+        endpoint = "/balances/consume"
+        payload = {
+            "api_key": api_key,
+            "amount_cents": amount_cents,
+            "usage_event_id": usage_event_id,
+        }
+        if metadata:
+            payload["metadata"] = metadata
+            
+        try:
+            # _request raises AgentPayAPIError for 4xx/5xx responses
+            # and AgentPayConnectionError for network issues.
+            response_data = self._request("POST", endpoint, data=payload)
+            
+            # If _request completes without raising an exception, it implies a 2xx response.
+            # The API's 200 OK response for consume just confirms success.
+            return ConsumptionResult(success=True)
+        
+        except AgentPayAPIError as e:
+            # API returned a non-2xx status code (e.g., 402, 403, 409)
+            return ConsumptionResult(
+                success=False,
+                error_code=e.error_code, 
+                error_message=e.error_message
+            )
+        except AgentPayConnectionError as e:
+            # Network error occurred during the request
+            return ConsumptionResult(
+                success=False,
+                error_code="NETWORK_ERROR", 
+                error_message=str(e)
+            )
+        except Exception as e:
+            # Catch any other unexpected errors during the process
+            return ConsumptionResult(
+                success=False,
+                error_code="SDK_ERROR", 
+                error_message=f"An unexpected error occurred in the SDK: {e}"
+            ) 
+
+    def validate_api_key(self, api_key: str) -> bool:
+        """Validates if a User API Key is active and valid for this service.
+
+        Args:
+            api_key: The User's API key to validate.
+
+        Returns:
+            True if the API key is valid for the service, False otherwise.
+            Returns False if any API or network error occurs during validation.
+        """
+        endpoint = "/api_keys/validate"
+        payload = {"api_key": api_key}
+        
+        try:
+            response_data = self._request("POST", endpoint, data=payload)
+            
+            # Expecting a response like {"is_valid": true | false}
+            is_valid = response_data.get("is_valid")
+            
+            if isinstance(is_valid, bool):
+                return is_valid
+            else:
+                # Unexpected response format from the API
+                # Log this? For now, treat unexpected format as invalid.
+                print(f"Warning: Unexpected response format from /validate endpoint: {response_data}")
+                return False
+        
+        except (AgentPayAPIError, AgentPayConnectionError) as e:
+            # If any API or network error occurs, treat the key as invalid for safety.
+            # Log the error for debugging if needed.
+            print(f"API or Connection Error during validation: {e}")
+            return False
+        except Exception as e:
+            # Catch any other unexpected errors
+            print(f"Unexpected SDK error during validation: {e}")
+            return False 

agentpay_sdk/exceptions.py

@@ -0,0 +1,20 @@
+class AgentPayError(Exception):
+    """Base exception class for AgentPay SDK errors."""
+    pass
+
+class AgentPayAPIError(AgentPayError):
+    """Raised when the AgentPay API returns an error response."""
+    def __init__(self, status_code: int, error_code: str | None = None, error_message: str | None = None):
+        self.status_code = status_code
+        self.error_code = error_code
+        self.error_message = error_message
+        detail = f"API returned status {status_code}"
+        if error_code:
+            detail += f" (Code: {error_code})"
+        if error_message:
+            detail += f": {error_message}"
+        super().__init__(detail)
+
+class AgentPayConnectionError(AgentPayError):
+    """Raised when there is a network issue connecting to the AgentPay API."""
+    pass 

agentpay_sdk/types.py

@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+from typing import Optional
+
+@dataclass
+class ConsumptionResult:
+    """Represents the outcome of a balance consumption attempt."""
+    success: bool
+    error_code: Optional[str] = None
+    error_message: Optional[str] = None 

app.py

@@ -5,6 +5,28 @@ import uuid
 import os # Added for path joining
 import copy # For deep copying message list
 from datetime import datetime, timezone # Added timezone for UTC consistency
+from dotenv import load_dotenv
+
+# AgentPay SDK imports
+from agentpay_sdk import AgentPayClient
+
+# Load environment variables
+load_dotenv()
+
+# --- AgentPay Configuration ---
+
+try:
+    agentpay_client = AgentPayClient(service_token=os.getenv("AGENTPAY_SERVICE_TOKEN", ""))
+    print("AgentPay client initialized successfully")
+except Exception as e:
+    print(f"Warning: AgentPay client initialization failed: {e}")
+    agentpay_client = None
+
+# AgentPay pricing in cents
+PRICING = {
+    "send_message": 100,     # $1.00 = 100 cents
+    "get_profile": 10  # $0.10 = 10 cents
+}
 
 # --- Start of JSON I/O Helper Functions ---
 
@@ -102,8 +124,8 @@ def new_profile(request: gr.Request):
         # Decide how to handle this error - maybe return an error to the client?
     
     # 4. Return IDs, questionnaire data, and instructions
-    instructions_for_agent = "You have received a `profile_id` (public identifier for this user\'s profile) and an `auth_id` (private key for authentication for this user). Please ask the user to store both securely. The `auth_id` must be sent as an `X-Auth-ID` header in subsequent requests that require authentication for this user (e.g., `update_profile_answers`, `get_matches`, `get_messages`, `get_profile` for own profile, `send_message`). The `profile_id` is used to publicly identify this user to others (e.g., in matches, or when sending/receiving messages). You must now walk the user through the questionnaire, preferably question by question, and collect the answers (in preparation to update the user's profile using update_profile_answers). If the user provides answers that are not in the list of valid answers, please ask them to provide a valid answer."
-    instructions_for_user = "Your profile creation process has started! You\'ve been assigned a unique Profile ID and a secret Auth ID. Your AI agent will use these to manage your profile and interactions. Please walk through the questionnaire with your Agent, and provide your answers. You can upload a profile picture later. Before you proceed, please update your MCP configuration to include your Auth ID by adding the following to your mcp.json (or similar) configuration file: `\"matchmaker\": {\"command\": \"npx\", \"args\": [\"mcp-remote\", \"http://localhost:7860/gradio_api/mcp/sse\", \"--allow-http\", \"--header\", \"X-AUTH-ID:<your-auth-id>\"]}`"
+    instructions_for_agent = "You have received a `profile_id` (public identifier for this user\'s profile) and an `auth_id` (private key for authentication for this user). Please ask the user to store both securely. The `auth_id` must be sent as an `X-Auth-ID` header in subsequent requests that require authentication for this user (e.g., `update_profile_answers`, `get_matches`, `get_messages`, `get_profile` for own profile, `send_message`). The `profile_id` is used to publicly identify this user to others (e.g., in matches, or when sending/receiving messages). Before you proceed, please ask the user to update their MCP configuration to include their new Auth ID by adding the following to their mcp.json (or similar) configuration file: `\"matchmaker\": {\"command\": \"npx\", \"args\": [\"mcp-remote\", \"http://localhost:7860/gradio_api/mcp/sse\", \"--allow-http\", \"--header\", \"X-AUTH-ID:<your-auth-id>\"]}`, then to RESTART THE CLIENT (especially if you are Claude), or to try RECONNECTING (e.g. if you are Cursor or otherwise). Once they have done so, you must then walk the user through the questionnaire, preferably question by question, and collect the answers (in preparation to update the user's profile using update_profile_answers). If the user provides answers that are not in the list of valid answers, please ask them to provide a valid answer."
+    instructions_for_user = "Your profile creation process has started! You\'ve been assigned a unique Profile ID and a secret Auth ID. Your AI agent will use these to manage your profile and interactions. Please walk through the questionnaire with your Agent, and provide your answers. You can upload a profile picture later."
     
     return {
         "profile_id": profile_id,
@@ -119,10 +141,10 @@ def update_profile_answers(answers_payload_str: str, request: gr.Request):
     Requires X-Auth-ID header for authentication.
     Answers_payload_str is expected to be a JSON string.
     """
-    auth_id_header = request.headers.get("x-auth-id") # Headers are lowercased by Gradio/Starlette
-    if not auth_id_header:
+    auth_id = request.headers.get("x-auth-id") # Headers are lowercased by Gradio/Starlette
+    if not auth_id:
         return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
-    print(f"update_profile_answers with Auth ID: {auth_id_header}")
+    print(f"update_profile_answers with Auth ID: {auth_id}")
 
     try:
         answers_payload = json.loads(answers_payload_str)
@@ -136,7 +158,7 @@ def update_profile_answers(answers_payload_str: str, request: gr.Request):
     target_profile_id = None
 
     for pid, profile_data in profiles.items():
-        if profile_data.get("auth_id") == auth_id_header:
+        if profile_data.get("auth_id") == auth_id:
             user_profile = profile_data
             target_profile_id = pid
             break
@@ -186,21 +208,21 @@ def get_matches(request: gr.Request):
     Requires X-Auth-ID header for authentication.
     For P1, this returns up to 3 random profiles, excluding the user's own.
     """
-    auth_id_header = request.headers.get("x-auth-id")
-    if not auth_id_header:
+    auth_id = request.headers.get("x-auth-id")
+    if not auth_id:
         return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
-    print(f"get_matches with Auth ID: {auth_id_header}")
+    print(f"get_matches with Auth ID: {auth_id}")
 
     profiles = load_json_data(PROFILES_FILE, default_data={})
     
     requester_profile_id = None
     requester_profile = None
     for pid, profile_data in profiles.items():
-        if profile_data.get("auth_id") == auth_id_header:
+        if profile_data.get("auth_id") == auth_id:
             requester_profile_id = pid
             requester_profile = profile_data
             break
-            
+    
     if not requester_profile:
         return {"status": "error", "message": "Authentication failed: Invalid X-Auth-ID."}
 
@@ -273,7 +295,7 @@ def get_matches(request: gr.Request):
         "status": "success",
         "matches": matches_list,
         "instructions_for_agent": "You have received a list of matches. For each match, you can call `get_profile` using their `profile_id` to get more details. You can also use `send_message` to send a message to a match's `profile_id`. For each match, please format it in a nice card-like display (not raw JSON), including the match's name, gender, profile summary, and profile image (in a way that it should render in the chat client).",
-        "instructions_for_user": "Here are some potential matches! Your AI agent can get more details on them ($0.10 per profile) or help you send a message ($1.00 per message). Or, if you don't like any of your matches, you can call `get_matches` again to get a new list."
+        "instructions_for_user": "Here are some potential matches! Your AI agent can get more details on them ($0.10 per profile) or help you send a message ($1.00 per message). Or, if you don't like any of your matches, you can call `get_matches` again to get a new list (free)."
     }
 
 def get_profile(profile_id_to_get: str, request: gr.Request):
@@ -281,18 +303,22 @@ def get_profile(profile_id_to_get: str, request: gr.Request):
     Gets a user's full profile.
     Requires X-Auth-ID header for authentication.
     Access is free for viewing one's own profile.
-    Accessing another user's profile has a cost (placeholder for P1).
+    Accessing another user's profile costs $0.10 and requires X-AGENTPAY-API-KEY.
     """
-    auth_id_header = request.headers.get("x-auth-id")
-    if not auth_id_header:
+    # Extract headers directly
+    auth_id = request.headers.get("x-auth-id")
+    agentpay_api_key = request.headers.get("x-agentpay-api-key")
+    
+    if not auth_id:
         return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
-    print(f"get_profile with Auth ID: {auth_id_header}")
+    
+    print(f"get_profile with Auth ID: {auth_id}")
 
     profiles = load_json_data(PROFILES_FILE, default_data={})
     
     requester_profile_id = None
     for pid, profile_data in profiles.items():
-        if profile_data.get("auth_id") == auth_id_header:
+        if profile_data.get("auth_id") == auth_id:
             requester_profile_id = pid
             break
             
@@ -304,6 +330,31 @@ def get_profile(profile_id_to_get: str, request: gr.Request):
     if not target_profile:
         return {"status": "error", "message": f"Profile with ID '{profile_id_to_get}' not found."}
 
+    # Check if viewing another user's profile (requires payment)
+    is_viewing_other = profile_id_to_get != requester_profile_id
+    
+    if is_viewing_other:
+        # Require AgentPay API key for viewing other profiles
+        if not agentpay_api_key:
+            return {"status": "error", "message": "Payment failed: X-AGENTPAY-API-KEY header is required to view other profiles."}
+        
+        # Validate AgentPay API key directly
+        if not agentpay_client or not agentpay_client.validate_api_key(agentpay_api_key):
+            return {"status": "error", "message": "Payment failed: Invalid AgentPay API key"}
+        
+        # Charge for viewing other profile via AgentPay
+        usage_id = str(uuid.uuid4())
+        amount_cents = PRICING["get_profile"]
+        result = agentpay_client.consume(
+            api_key=agentpay_api_key,
+            amount_cents=amount_cents,
+            usage_event_id=usage_id,
+            metadata={"tool": "get_profile", "requester": requester_profile_id, "target": profile_id_to_get}
+        )
+        
+        if not result.success:
+            return {"status": "error", "message": f"Payment failed: {result.error_message}"}
+
     # For security, never return the auth_id
     target_profile.pop("auth_id", None)
     
@@ -317,16 +368,15 @@ def get_profile(profile_id_to_get: str, request: gr.Request):
     target_profile['profile_image_url'] = image_url
     target_profile.pop("profile_image_filename", None)
 
+    # Update cost and messaging based on whether payment was charged
     cost_incurred = 0.0
-    instructions_for_agent = f"You have retrieved the profile for {profile_id_to_get}."
-    instructions_for_user = f"Here is the profile for {profile_id_to_get}."
-
-    if profile_id_to_get != requester_profile_id:
-        # This is where a real payment would be processed.
-        # For P1, we just note the cost.
-        cost_incurred = 0.10 
-        instructions_for_agent += f" As this was not your own profile, a cost of ${cost_incurred:.2f} was incurred (for PoC, this is just a note)."
-        instructions_for_user += " Viewing other profiles may have a cost."
+    if is_viewing_other:
+        cost_incurred = PRICING["get_profile"] / 100.0  # Convert cents to dollars
+        instructions_for_agent = f"You have retrieved the profile for {profile_id_to_get}. A cost of {cost_incurred} was charged via AgentPay."
+        instructions_for_user = f"Here is the profile for {profile_id_to_get}."
+    else:
+        instructions_for_agent = f"You have retrieved your own profile ({profile_id_to_get}). No cost was incurred."
+        instructions_for_user = f"Here is your profile."
 
     return {
         "status": "success",
@@ -338,20 +388,31 @@ def get_profile(profile_id_to_get: str, request: gr.Request):
 
 def send_message(receiver_profile_id: str, content: str, request: gr.Request = None):
     """
-    Sends a message to a match. 
-    Requires X-Auth-ID header for authentication.
-    Costs $1.00 per message (placeholder for P1).
+    Sends a message to a match.
+    Requires X-Auth-ID header for authentication and X-AGENTPAY-API-KEY for payment.
+    Costs $1.00 per message.
     """
-    auth_id_header = request.headers.get("x-auth-id")
-    if not auth_id_header:
+    # Extract headers directly
+    auth_id = request.headers.get("x-auth-id")
+    agentpay_api_key = request.headers.get("x-agentpay-api-key")
+    
+    if not auth_id:
         return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
-    print(f"send_message with Auth ID: {auth_id_header}")
+    
+    if not agentpay_api_key:
+        return {"status": "error", "message": "Payment failed: X-AGENTPAY-API-KEY header is missing."}
+    
+    # Validate AgentPay API key directly
+    if not agentpay_client or not agentpay_client.validate_api_key(agentpay_api_key):
+        return {"status": "error", "message": "Payment failed: Invalid AgentPay API key"}
+    
+    print(f"send_message with Auth ID: {auth_id}")
 
     profiles = load_json_data(PROFILES_FILE, default_data={})
     
     sender_profile_id = None
     for pid, profile_data in profiles.items():
-        if profile_data.get("auth_id") == auth_id_header:
+        if profile_data.get("auth_id") == auth_id:
             sender_profile_id = pid
             break
             
@@ -361,9 +422,18 @@ def send_message(receiver_profile_id: str, content: str, request: gr.Request = N
     if receiver_profile_id not in profiles:
         return {"status": "error", "message": "Receiver profile ID not found."}
 
-    # For P1, we are not integrating a real payment system.
-    # We will integrate AgentPay here late.
-    cost_incurred = 100
+    # Charge for sending message via AgentPay
+    usage_id = str(uuid.uuid4())
+    amount_cents = PRICING["send_message"]
+    result = agentpay_client.consume(
+        api_key=agentpay_api_key,
+        amount_cents=amount_cents,
+        usage_event_id=usage_id,
+        metadata={"tool": "send_message", "sender": sender_profile_id, "receiver": receiver_profile_id}
+    )
+    
+    if not result.success:
+        return {"status": "error", "message": f"Payment failed: {result.error_message}"}
     
     messages = load_json_data(MESSAGES_FILE, default_data=[])
     
@@ -384,8 +454,8 @@ def send_message(receiver_profile_id: str, content: str, request: gr.Request = N
     return {
         "status": "success",
         "message_id": new_message["message_id"],
-        "cost_incurred_usd_cents": cost_incurred,
-        "instructions_for_agent": f"Message sent successfully to {receiver_profile_id}. A cost of {cost_incurred/100:.2f} cents was incurred (for PoC, this is just a note). You can get all messages for the user with `get_messages`.",
+        "cost_incurred_usd_cents": amount_cents,
+        "instructions_for_agent": f"Message sent successfully to {receiver_profile_id}. A cost of ${amount_cents/100:.2f} was charged via AgentPay. You can get all messages for the user with `get_messages`.",
         "instructions_for_user": f"Your message has been sent to {receiver_profile_id}!"
     }
 
@@ -395,16 +465,16 @@ def get_messages(request: gr.Request):
     Marks retrieved messages where the user is the receiver as read for subsequent calls.
     Requires X-Auth-ID header for authentication.
     """
-    auth_id_header = request.headers.get("x-auth-id")
-    if not auth_id_header:
+    auth_id = request.headers.get("x-auth-id")
+    if not auth_id:
         return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
-    print(f"get_messages with Auth ID: {auth_id_header}")
+    print(f"get_messages with Auth ID: {auth_id}")
 
     profiles = load_json_data(PROFILES_FILE, default_data={})
     
     user_profile_id = None
     for pid, profile_data in profiles.items():
-        if profile_data.get("auth_id") == auth_id_header:
+        if profile_data.get("auth_id") == auth_id:
             user_profile_id = pid
             break
             
@@ -451,8 +521,8 @@ def upload_profile_picture(image_upload, request: gr.Request):
     The new filename will be based on the user's profile_id.
     Requires X-Auth-ID header for authentication.
     """
-    auth_id_header = request.headers.get("x-auth-id")
-    if not auth_id_header:
+    auth_id = request.headers.get("x-auth-id")
+    if not auth_id:
         return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
 
     if image_upload is None:
@@ -463,7 +533,7 @@ def upload_profile_picture(image_upload, request: gr.Request):
     user_profile = None
     target_profile_id = None
     for pid, profile_data in profiles.items():
-        if profile_data.get("auth_id") == auth_id_header:
+        if profile_data.get("auth_id") == auth_id:
             user_profile = profile_data
             target_profile_id = pid
             break
@@ -553,7 +623,7 @@ get_matches_demo = gr.Interface(
     inputs=None,
     outputs=gr.JSON(label="Matches"),
     title="Get Matches",
-    description="Gets a list of potential matches for the authenticated user. Requires X-Auth-ID header."
+    description="Gets a list of potential matches for the authenticated user. Requires X-Auth-ID header. FREE service."
 )
 # --- End of Get Matches Interface ---
 
@@ -563,7 +633,7 @@ get_profile_demo = gr.Interface(
     inputs=[gr.Textbox(label="Profile ID to Get")],
     outputs=gr.JSON(label="Profile Details"),
     title="Get Profile",
-    description="Gets the full profile for a given Profile ID. Requires X-Auth-ID header."
+    description="Gets the full profile for a given Profile ID. Requires X-Auth-ID header. Free for own profile, $0.10 for others (requires X-AGENTPAY-API-KEY)."
 )
 # --- End of Get Profile Interface ---
 
@@ -576,7 +646,7 @@ send_message_demo = gr.Interface(
     ],
     outputs=gr.JSON(label="Send Status"),
     title="Send Message",
-    description="Sends a message to another user. Requires X-Auth-ID header. ($1.00 placeholder cost)"
+    description="Sends a message to another user. Requires X-Auth-ID and X-AGENTPAY-API-KEY headers. ($1.00 per message)"
 )
 # --- End of Send Message Interface ---
 

data/messages.json

@@ -63,5 +63,29 @@
     "content": "Hey there",
     "timestamp": "2025-06-06T09:28:13.285402+00:00",
     "read_status": false
+  },
+  {
+    "message_id": "044124be-3081-4ce4-a4fa-a80245dafc2f",
+    "sender_profile_id": "user_e6chs11v",
+    "receiver_profile_id": "user_d8x3v7n1",
+    "content": "Hi-acynth",
+    "timestamp": "2025-06-09T08:46:21.258372+00:00",
+    "read_status": false
+  },
+  {
+    "message_id": "62f5d6ac-33f5-4f86-93f1-5d84b9fe73f9",
+    "sender_profile_id": "user_5ru2ht5q",
+    "receiver_profile_id": "user_l9z4f2x7",
+    "content": "Hi Maya! I'm John - I saw you're a data scientist who loves cooking Italian food. As someone who appreciates things being clean and organized (I'm a janitor!), I have a lot of respect for both data visualization and a good homemade pasta recipe. I'm more of a homebody, but I'd love to hear about your rock climbing adventures over a cozy podcast listening session. Maybe you could teach me to make that famous pasta of yours?",
+    "timestamp": "2025-06-10T15:54:56.209149+00:00",
+    "read_status": false
+  },
+  {
+    "message_id": "2bc9a582-e677-45e3-91bc-cd63472f4922",
+    "sender_profile_id": "user_5ru2ht5q",
+    "receiver_profile_id": "user_m4k7r9n2",
+    "content": "Hi Emma! I'm John - I really admire your balanced approach to life with yoga and farmers markets. As someone who takes pride in keeping things clean and organized (I'm a janitor), I have a deep appreciation for the quality and freshness you find at farmers markets. I'm more of a homebody, but I'd love to have some genuine conversations about your travels and maybe learn about your favorite market finds. Your optimistic vibe sounds like it would bring great energy to cozy evenings in!",
+    "timestamp": "2025-06-10T16:12:01.428426+00:00",
+    "read_status": false
   }
 ]

data/profiles.json

@@ -280,5 +280,37 @@
       "q_vibe": "Caring and generous.",
       "q_gender_preference": "Men"
     }
+  },
+  "user_e6chs11v": {
+    "profile_id": "user_e6chs11v",
+    "auth_id": "b9b8e91f-8369-4eff-a804-5a28a83f3cf3",
+    "created_at": "2025-06-09T08:32:50.340435+00:00",
+    "updated_at": "2025-06-09T08:38:05.614570+00:00",
+    "name": "George",
+    "gender": "Man",
+    "profile_summary": "I'm George the great",
+    "profile_image_filename": "default.jpg",
+    "answers": {
+      "q_gender_preference": "Women",
+      "q_hobby": "Gardening",
+      "q_looking_for": "I'm looking for someone to grow my love with.",
+      "q_vibe": "Homebody"
+    }
+  },
+  "user_5ru2ht5q": {
+    "profile_id": "user_5ru2ht5q",
+    "auth_id": "e98bbb1e-0359-489b-b552-554a6eb22b21",
+    "created_at": "2025-06-10T15:45:43.284107+00:00",
+    "updated_at": "2025-06-10T15:48:45.874926+00:00",
+    "name": "John",
+    "gender": "Man",
+    "profile_summary": "I'm John, the janitor.",
+    "profile_image_filename": "default.jpg",
+    "answers": {
+      "q_gender_preference": "Women",
+      "q_hobby": "Cleaning things.",
+      "q_looking_for": "Someone neat and tidy to spend time with.",
+      "q_vibe": "Homebody"
+    }
   }
 }

requirements.txt

@@ -1 +1,3 @@
 gradio[mcp]
+requests
+python-dotenv