PoC complete

app.py

@@ -3,6 +3,7 @@ import random
 import json
 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
 
 # --- Start of JSON I/O Helper Functions ---
@@ -71,7 +72,8 @@ def profile_questionnaire(request: gr.Request):
     # 1. Generate profile_id and auth_id
     profile_id = f"user_{''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=8))}"
     auth_id = str(uuid.uuid4())
-    
+    print(f"profile_questionnaire with Profile ID: {profile_id} and Auth ID: {auth_id}")
+
     # 2. Read questionnaire.json
     questionnaire_data = load_json_data(QUESTIONNAIRE_FILE, default_data={"title": "Error Loading Questionnaire", "questions": []})
     if not questionnaire_data.get("questions") or questionnaire_data.get("title") == "Error Loading Questionnaire":
@@ -118,6 +120,7 @@ def update_profile_answers(answers_payload_str: str, request: gr.Request):
     auth_id_header = request.headers.get("x-auth-id") # Headers are lowercased by Gradio/Starlette
     if not auth_id_header:
         return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
+    print(f"update_profile_answers with Auth ID: {auth_id_header}")
 
     try:
         answers_payload = json.loads(answers_payload_str)
@@ -175,23 +178,222 @@ def update_profile_answers(answers_payload_str: str, request: gr.Request):
         "instructions_for_user": "Your profile has been updated! You can now look for matches or view your profile."
     }
 
-# --- End of MCP Matchmaker Tools ---
+def get_matches(request: gr.Request):
+    """
+    Finds potential matches for the authenticated user.
+    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:
+        return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
+    print(f"get_matches with Auth ID: {auth_id_header}")
 
-# Original example functions from the user's app.py for context
-def print_headers(text, request: gr.Request):
+    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:
+            requester_profile_id = pid
+            break
+            
+    if not requester_profile_id:
+        return {"status": "error", "message": "Authentication failed: Invalid X-Auth-ID."}
+
+    all_profile_ids = list(profiles.keys())
+    # Exclude the user's own profile from potential matches
+    potential_matches_ids = [pid for pid in all_profile_ids if pid != requester_profile_id]
+
+    # Randomly select up to 3 matches
+    num_matches = min(len(potential_matches_ids), 3)
+    selected_match_ids = random.sample(potential_matches_ids, k=num_matches)
+
+    matches_list = []
+    for match_id in selected_match_ids:
+        match_profile = profiles.get(match_id)
+        if match_profile:
+            matches_list.append({
+                "profile_id": match_profile.get("profile_id"),
+                "name": match_profile.get("name"),
+                "profile_summary": match_profile.get("profile_summary"),
+                "profile_image_filename": match_profile.get("profile_image_filename")
+            })
+
+    return {
+        "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 (this may have a cost). You can also use `send_message` to send a message to a match's `profile_id`.",
+        "instructions_for_user": "Here are some potential matches! Your AI agent can get more details on them or help you send a message."
+    }
+
+def get_profile(profile_id_to_get: str, request: gr.Request):
     """
-    Print the headers of the request for debugging purposes.
+    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).
+    """
+    auth_id_header = request.headers.get("x-auth-id")
+    if not auth_id_header:
+        return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
+    print(f"get_profile with Auth ID: {auth_id_header}")
+
+    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:
+            requester_profile_id = pid
+            break
+            
+    if not requester_profile_id:
+        return {"status": "error", "message": "Authentication failed: Invalid X-Auth-ID."}
+
+    target_profile = profiles.get(profile_id_to_get)
+
+    if not target_profile:
+        return {"status": "error", "message": f"Profile with ID '{profile_id_to_get}' not found."}
+
+    # For security, never return the auth_id
+    target_profile.pop("auth_id", None)
+    
+    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}."
 
-    Args:
-        text (str): The text to print
-        request (gr.Request): The request object
+    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."
 
-    Returns:
-        str: The text to print
+    return {
+        "status": "success",
+        "profile": target_profile,
+        "cost_incurred_usd": cost_incurred,
+        "instructions_for_agent": instructions_for_agent,
+        "instructions_for_user": instructions_for_user
+    }
+
+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).
     """
+    auth_id_header = request.headers.get("x-auth-id")
+    if not auth_id_header:
+        return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
+    print(f"send_message with Auth ID: {auth_id_header}")
+
+    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:
+            sender_profile_id = pid
+            break
+            
+    if not sender_profile_id:
+        return {"status": "error", "message": "Authentication failed: Invalid X-Auth-ID."}
+
+    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
+    
+    messages = load_json_data(MESSAGES_FILE, default_data=[])
+    
+    new_message = {
+        "message_id": str(uuid.uuid4()),
+        "sender_profile_id": sender_profile_id,
+        "receiver_profile_id": receiver_profile_id,
+        "content": content,
+        "timestamp": datetime.now(timezone.utc).isoformat(),
+        "read_status": False # Messages are unread when sent
+    }
+    
+    messages.append(new_message)
+    
+    if not save_json_data(MESSAGES_FILE, messages):
+        return {"status": "error", "message": "Failed to save message."}
+
+    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`.",
+        "instructions_for_user": f"Your message has been sent to {receiver_profile_id}!"
+    }
+
+def get_messages(request: gr.Request):
+    """
+    Gets all messages for the authenticated user (sent and received).
+    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:
+        return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
+    print(f"get_messages with Auth ID: {auth_id_header}")
+
+    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:
+            user_profile_id = pid
+            break
+            
+    if not user_profile_id:
+        return {"status": "error", "message": "Authentication failed: Invalid X-Auth-ID."}
+
+    all_messages = load_json_data(MESSAGES_FILE, default_data=[])
+    
+    # Use a deep copy to avoid modifying the list while iterating
+    messages_for_user = copy.deepcopy([
+        msg for msg in all_messages 
+        if msg.get("sender_profile_id") == user_profile_id or msg.get("receiver_profile_id") == user_profile_id
+    ])
+
+    # Sort messages by timestamp descending (newest first)
+    messages_for_user.sort(key=lambda x: x.get("timestamp", ""), reverse=True)
+
+    # Mark received messages as read and track if an update is needed
+    needs_save = False
+    for i, original_msg in enumerate(all_messages):
+        # Check if this message is one of the user's messages and was received by them
+        if original_msg.get("receiver_profile_id") == user_profile_id and not original_msg.get("read_status"):
+            all_messages[i]["read_status"] = True
+            needs_save = True
+
+    # Save the updated message list back to the file if any message was marked as read
+    if needs_save:
+        if not save_json_data(MESSAGES_FILE, all_messages):
+            # If saving fails, we should still return the messages, but log the error.
+            print("Error: Could not update read_status for messages in the database.")
+            # Depending on desired behavior, we could return an error status here.
+            # For now, we will proceed to return the messages as requested.
+
+    return {
+        "status": "success",
+        "messages": messages_for_user,
+        "instructions_for_agent": "You have received all messages for the user. Messages they received in this batch have now been marked as 'read' on the server.",
+        "instructions_for_user": "Here are your messages."
+    }
+
+def print_headers(text, request: gr.Request):
+    """Print the headers of the request for debugging purposes."""
     print(f"Headers for print_headers request: {request.headers}")
     print(f"Text for print_headers: {text}")
-    return text
+    return f"Printed headers and text: {text}"
+
+# --- End of MCP Matchmaker Tools ---
+
+# --- Start of Gradio App Setup ---
 
 #  Gradio interface for printing headers
 headers_demo = gr.Interface(
@@ -222,10 +424,53 @@ update_profile_answers_demo = gr.Interface(
 )
 # --- End of Update Profile Answers Interface ---
 
+# --- Start of Get Matches Interface ---
+get_matches_demo = gr.Interface(
+    fn=get_matches,
+    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."
+)
+# --- End of Get Matches Interface ---
+
+# --- Start of Get Profile Interface ---
+get_profile_demo = gr.Interface(
+    fn=get_profile,
+    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."
+)
+# --- End of Get Profile Interface ---
+
+# --- Start of Send Message Interface ---
+send_message_demo = gr.Interface(
+    fn=send_message,
+    inputs=[
+        gr.Textbox(label="Receiver Profile ID"),
+        gr.Textbox(label="Message Content", lines=5)
+    ],
+    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)"
+)
+# --- End of Send Message Interface ---
+
+# --- Start of Get Messages Interface ---
+get_messages_demo = gr.Interface(
+    fn=get_messages,
+    inputs=None,
+    outputs=gr.JSON(label="Messages"),
+    title="Get Messages",
+    description="Gets all messages sent or received by the authenticated user. Requires X-Auth-ID header."
+)
+# --- End of Get Messages Interface ---
+
 # Adjusted TabbedInterface to include the new tool
 demo = gr.TabbedInterface(
-    [profile_questionnaire_demo, update_profile_answers_demo, headers_demo], 
-    ["Profile Questionnaire", "Update Profile Answers", "Headers Debug"]
+    [profile_questionnaire_demo, update_profile_answers_demo, get_matches_demo, get_profile_demo, send_message_demo, get_messages_demo, headers_demo], 
+    ["Profile Questionnaire", "Update Profile Answers", "Get Matches", "Get Profile", "Send Message", "Get Messages", "Headers Debug"]
 )
 
 if __name__ == "__main__":

data/messages.json

@@ -1 +1,27 @@
-[]
+[
+  {
+    "message_id": "7880956d-244d-4c2b-9b7c-bd6ecb371ed4",
+    "sender_profile_id": "user_y38k71ox",
+    "receiver_profile_id": "user_p9zqwxv8",
+    "content": "Hey there!",
+    "timestamp": "2025-06-06T04:56:31.804886+00:00",
+    "read_status": true
+  },
+  {
+    "message_id": "5809af1e-8a57-41d1-9319-233c7e66a26a",
+    "sender_profile_id": "user_p9zqwxv8",
+    "receiver_profile_id": "user_y38k71ox",
+    "content": "Hey you!",
+    "timestamp": "2025-06-06T05:12:46.611177+00:00",
+    "read_status": true
+  },
+  {
+    "message_id": "440f9c41-626f-4b45-89b5-1f59351d190f",
+    "sender_profile_id": "user_y38k71ox",
+    "receiver_profile_id": "user_p9zqwxv8",
+    "content": "How are you doing?",
+    "timestamp": "2025-06-06T06:44:02.018941+00:00",
+    "read_status": true,
+    "payment_confirmation_id": "not_provided"
+  }
+]

data/profiles.json

@@ -22,5 +22,47 @@
       "q_looking_for": "I'm looking for someone who is as cool as me.",
       "q_vibe": "My general vibe is... coding."
     }
+  },
+  "user_h7bafk3d": {
+    "profile_id": "user_h7bafk3d",
+    "auth_id": "a4f8b9e6-7d6a-4c3b-8e1f-9a0c1b2d3e4f",
+    "created_at": "2025-06-05T17:10:00.000000+00:00",
+    "updated_at": "2025-06-05T17:10:00.000000+00:00",
+    "name": "Alex",
+    "profile_image_filename": "placeholder.png",
+    "profile_summary": "Loves hiking and exploring new cafes.",
+    "answers": {
+      "q_hobby": "Hiking, photography, trying new recipes.",
+      "q_looking_for": "Someone adventurous and kind.",
+      "q_vibe": "Outdoorsy and creative."
+    }
+  },
+  "user_p9zqwxv8": {
+    "profile_id": "user_p9zqwxv8",
+    "auth_id": "b5c7d8f9-8e7b-4d2c-9f0a-1b2c3d4e5f6g",
+    "created_at": "2025-06-05T17:11:00.000000+00:00",
+    "updated_at": "2025-06-05T17:11:00.000000+00:00",
+    "name": "Bella",
+    "profile_image_filename": "placeholder.png",
+    "profile_summary": "Bookworm, artist, and enjoys quiet nights in.",
+    "answers": {
+      "q_hobby": "Reading, painting, watching classic movies.",
+      "q_looking_for": "A thoughtful person to share deep conversations with.",
+      "q_vibe": "Cozy and artistic."
+    }
+  },
+  "user_t2ysgmnp": {
+    "profile_id": "user_t2ysgmnp",
+    "auth_id": "c6d8e9g0-9f8c-5e1d-a01b-2c3d4e5f6g7h",
+    "created_at": "2025-06-05T17:12:00.000000+00:00",
+    "updated_at": "2025-06-05T17:12:00.000000+00:00",
+    "name": "Charlie",
+    "profile_image_filename": "placeholder.png",
+    "profile_summary": "Tech enthusiast and loves a good board game.",
+    "answers": {
+      "q_hobby": "Building PCs, board games, sci-fi novels.",
+      "q_looking_for": "A partner-in-crime for game nights and tech talks.",
+      "q_vibe": "Geeky and friendly."
+    }
   }
 }