update_answers

app.py

@@ -3,7 +3,7 @@ import random
 import json
 import uuid
 import os # Added for path joining
-from datetime import datetime # Added for timestamps
+from datetime import datetime, timezone # Added timezone for UTC consistency
 
 # --- Start of JSON I/O Helper Functions ---
 
@@ -63,7 +63,7 @@ def save_json_data(filepath, data):
 
 # --- Start of MCP Matchmaker Tools ---
 
-def profile_questionnaire_func(request: gr.Request):
+def profile_questionnaire(request: gr.Request):
     """
     Generates and returns a profile questionnaire, a new public profile_id, 
     and a new private auth_id. Also creates an initial profile stub.
@@ -80,7 +80,8 @@ def profile_questionnaire_func(request: gr.Request):
 
     # 3. Create new profile entry in profiles.json
     profiles = load_json_data(PROFILES_FILE, default_data={})
-    timestamp = datetime.utcnow().isoformat() + "Z" # ISO 8601 format
+    # Ensure timestamp is timezone-aware (UTC)
+    timestamp = datetime.now(timezone.utc).isoformat()
     
     profiles[profile_id] = {
         "profile_id": profile_id,
@@ -97,7 +98,7 @@ def profile_questionnaire_func(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). 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)."
+    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)."
     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."
     
     return {
@@ -108,6 +109,72 @@ def profile_questionnaire_func(request: gr.Request):
         "instructions_for_user": instructions_for_user
     }
 
+def update_profile_answers(answers_payload_str: str, request: gr.Request):
+    """
+    Updates a user's profile based on questionnaire answers.
+    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:
+        return {"status": "error", "message": "Authentication failed: X-Auth-ID header is missing."}
+
+    try:
+        answers_payload = json.loads(answers_payload_str)
+        if not isinstance(answers_payload, dict):
+            raise json.JSONDecodeError("Payload is not a dictionary.", answers_payload_str, 0)
+    except json.JSONDecodeError as e:
+        return {"status": "error", "message": f"Invalid JSON format in answers_payload: {e}"}
+
+    profiles = load_json_data(PROFILES_FILE, default_data={})
+    user_profile = None
+    target_profile_id = None
+
+    for pid, profile_data in profiles.items():
+        if profile_data.get("auth_id") == auth_id_header:
+            user_profile = profile_data
+            target_profile_id = pid
+            break
+    
+    if not user_profile:
+        return {"status": "error", "message": "Authentication failed: Invalid X-Auth-ID."}
+
+    questionnaire = load_json_data(QUESTIONNAIRE_FILE, default_data={"questions": []})
+    questions_map = {q["id"]: q for q in questionnaire.get("questions", [])}
+
+    updated_fields = False
+    for question_id, answer_value in answers_payload.items():
+        question_details = questions_map.get(question_id)
+        if not question_details:
+            print(f"Warning: Received answer for unknown question_id '{question_id}'. Skipping.")
+            continue
+
+        if question_details.get("purpose") == "metadata":
+            field_to_map = question_details.get("maps_to_field")
+            if field_to_map and field_to_map in user_profile:
+                user_profile[field_to_map] = answer_value
+                updated_fields = True
+            else:
+                print(f"Warning: Metadata question '{question_id}' has no valid 'maps_to_field' or field not in profile. Skipping.")
+        elif question_details.get("purpose") == "matchmaking":
+            user_profile["answers"][question_id] = answer_value
+            updated_fields = True
+        else:
+            print(f"Warning: Question '{question_id}' has unknown purpose '{question_details.get('purpose')}'. Skipping.")
+
+    if updated_fields:
+        user_profile["updated_at"] = datetime.now(timezone.utc).isoformat()
+        profiles[target_profile_id] = user_profile # Update the profile in the main dictionary
+        if not save_json_data(PROFILES_FILE, profiles):
+            return {"status": "error", "message": "Failed to save profile updates."}
+    
+    return {
+        "status": "success", 
+        "message": "Profile updated successfully.",
+        "instructions_for_agent": "Profile updated. Next recommended steps: call `get_matches` to find potential matches, or `get_profile` (using the user's `profile_id`) to view the updated profile.",
+        "instructions_for_user": "Your profile has been updated! You can now look for matches or view your profile."
+    }
+
 # --- End of MCP Matchmaker Tools ---
 
 # Original example functions from the user's app.py for context
@@ -137,18 +204,28 @@ headers_demo = gr.Interface(
 
 # --- Start of New Profile Questionnaire Interface ---
 profile_questionnaire_demo = gr.Interface(
-    fn=profile_questionnaire_func,
+    fn=profile_questionnaire,
     inputs=None, # No direct input from user for this one, it's triggered by the agent
     outputs=gr.JSON(label="Questionnaire, IDs, and Instructions"), 
-    title="MCP - Profile Questionnaire",
-    description="Generates a new user profile questionnaire, Profile ID, and Auth ID. (Agent-triggered MCP tool)"
+    title="Profile Questionnaire",
+    description="Generates a new user profile and returns a questionnaire, Profile ID, and Auth ID. (Agent-triggered MCP tool)"
 )
 # --- End of New Profile Questionnaire Interface ---
 
+# --- Start of Update Profile Answers Interface ---
+update_profile_answers_demo = gr.Interface(
+    fn=update_profile_answers,
+    inputs=[gr.Textbox(label="Answers Payload (JSON string)", lines=5)], # Agent provides this as a JSON string
+    outputs=gr.JSON(label="Update Status"),
+    title="Update Profile Answers",
+    description="Updates a user's profile with answers to the questionnaire. Requires X-Auth-ID header."
+)
+# --- End of Update Profile Answers Interface ---
+
 # Adjusted TabbedInterface to include the new tool
 demo = gr.TabbedInterface(
-    [profile_questionnaire_demo, headers_demo], 
-    ["MCP - Profile Questionnaire", "Headers Debug"]
+    [profile_questionnaire_demo, update_profile_answers_demo, headers_demo], 
+    ["Profile Questionnaire", "Update Profile Answers", "Headers Debug"]
 )
 
 if __name__ == "__main__":

data/profiles.json

@@ -8,5 +8,19 @@
     "profile_image_filename": null,
     "profile_summary": "",
     "answers": {}
+  },
+  "user_y38k71ox": {
+    "profile_id": "user_y38k71ox",
+    "auth_id": "cc175293-a9eb-4db0-ad2f-f3ba7a0dab25",
+    "created_at": "2025-06-05T16:49:53.871722+00:00",
+    "updated_at": "2025-06-05T17:00:46.839293+00:00",
+    "name": "User",
+    "profile_image_filename": "placeholder.png",
+    "profile_summary": "I am very cool, please date me.",
+    "answers": {
+      "q_hobby": "My main hobby is coding.",
+      "q_looking_for": "I'm looking for someone who is as cool as me.",
+      "q_vibe": "My general vibe is... coding."
+    }
   }
 }