Add life status endpoint

app/main.py

@@ -15,7 +15,7 @@ from openai import OpenAI
 import psycopg2
 from psycopg2 import sql
 import os
-from app.utils import get_api_key, get_user_info, print_log, update_user, upload_file_to_s3, get_user, upload_mementos_to_db, get_user_summary
+from app.utils import get_api_key, get_user_info, print_log, update_user, upload_file_to_s3, get_user, upload_mementos_to_db, get_user_summary, get_user_life_status
 from dotenv import load_dotenv
 import logging.config
 import time
@@ -574,6 +574,30 @@ def get_summary_by_id(user_id: str, api_key: str = Security(get_api_key)):
     except Exception as e:
         print_log("ERROR",f"Error getting user: {str(e)}", extra={"user_id": user_id, "endpoint": "/get_user_summary"}, exc_info=True)
         logger.error(f"Error getting user: {str(e)}", extra={"user_id": user_id, "endpoint": "/get_user_summary"}, exc_info=True)
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=str(e)
+        )
+    
+@app.get("/get_life_status")
+def get_life_status_by_id(user_id: str, api_key: str = Security(get_api_key)):
+    print_log("INFO", "Getting user's life status", extra={"user_id": user_id, "endpoint": "/get_life_status"})
+    logger.info("Getting user's life status", extra={"user_id": user_id, "endpoint": "/get_life_status"})
+    try:
+        life_status = get_user_life_status(user_id)
+        print_log("INFO", "Successfully generated life status", extra={"user_id": user_id, "endpoint": "/get_life_status"})
+        logger.info("Successfully generated life status", extra={"user_id": user_id, "endpoint": "/get_life_status"})
+        return life_status
+    except LookupError:
+        print_log("ERROR", "User not found", extra={"user_id": user_id, "endpoint": "/get_life_status"})
+        logger.error("User not found", extra={"user_id": user_id, "endpoint": "/get_life_status"})
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail=f"User with ID {user_id} not found"
+        )
+    except Exception as e:
+        print_log("ERROR",f"Error getting user: {str(e)}", extra={"user_id": user_id, "endpoint": "/get_life_status"}, exc_info=True)
+        logger.error(f"Error getting user: {str(e)}", extra={"user_id": user_id, "endpoint": "/get_life_status"}, exc_info=True)
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail=str(e)

app/utils.py

@@ -598,6 +598,247 @@ def get_user_summary(user_id):
     logger.info(f"User summary generated successfully for user {user_id}", extra={'user_id': user_id, 'endpoint': function_name})
     return reports
 
+def get_user_life_status(user_id):
+    function_name = get_user_life_status.__name__
+    logger.info(f"Generating user life status for user {user_id}", extra={'user_id': user_id, 'endpoint': function_name})
+
+    # Step 1: Call get_user to get user's info
+    try:
+        user = get_user(user_id)
+        user_info = user.user_info
+        user_messages = user.get_messages()
+    except LookupError as e:
+        logger.error(f"Error fetching user data: {e}", extra={'user_id': user_id, 'endpoint': function_name})
+        raise e
+
+    # Step 2: Construct the Prompt
+    chat_history = "\n".join(
+        [f"{message['role'].capitalize()}: {message['content']}" for message in user_messages]
+    )
+
+    # Build the system prompt according to the provided instructions
+    system_prompt = """
+    You are an AI assistant that generates a personalized life status report for users based on their profile and chat history. Your task is to analyze the provided user data and produce a JSON output following the specified schema.
+
+    **Instructions:**
+
+    1. **Life Score:**
+    - Calculate numerical scores for each of the following areas based on the user's profile and chat history, especially their responses to quantitative questions (e.g., "on a scale of 1 to 5, how do you feel about ..."):
+        - `personal_growth`
+        - `career_growth`
+        - `mental_well_being`
+        - `health_and_wellness`
+        - `relationship`
+    - Each score should be a float between 0 and 1. If an area hasn't been covered in the chat history, assign it a score of 0.
+    - Compute the `overall` score as the average of the non-zero area scores.
+
+    2. **Mantra of the Week:**
+    - Create a very short encouragement quote that encapsulates the user's journey toward achieving their goals.
+    - The mantra should be a single sentence with fewer than 10 words.
+
+    3. **This Week's Focus:**
+    - Identify the top three most important areas for the user from the five key areas:
+        - `personal_growth`
+        - `career_growth`
+        - `mental_well_being`
+        - `health_and_wellness`
+        - `relationship`
+    - For each selected area, provide a concise statement outlining the user's focus in that area.
+
+    4. **Suggested Action Items:**
+    - For the same three areas identified above, suggest concrete and actionable items that the user can undertake to make progress.
+
+    **Output Format:**
+
+    Produce your response in JSON format adhering to the following schema:
+
+    ```json
+    {
+        "life_score": {
+            "overall": float,
+            "personal_growth": float,
+            "health_and_wellness": float,
+            "mental_well_being": float,
+            "career_growth": float,
+            "relationship": float
+        },
+        "mantra_of_the_week": str,
+        "this_week_focus": [
+            {"area": str, "focus": str},
+            {"area": str, "focus": str},
+            {"area": str, "focus": str}
+        ],
+        "action_items": [
+            {"area": str, "focus": str},
+            {"area": str, "focus": str},
+            {"area": str, "focus": str}
+        ]
+    }
+    ```
+
+    **Guidelines:**
+
+    - Ensure all numerical values are floats (e.g., 3.0 instead of 3).
+    - The `mantra_of_the_week` should be personalized, positive, and encouraging.
+    - Focus statements and action items should be clear, specific, and tailored to the user's context.
+    - Do not include any additional text or commentary outside of the JSON structure.
+    """
+
+    # Combine user information and chat history for context
+    user_context = f"""
+    Based on the following user profile and chat history, generate the life status!
+
+    ### USER PROFILE ###
+    {user_info}
+
+    ### CHAT HISTORY ###
+    {chat_history}
+    """
+
+    # Step 3: Call the OpenAI API using the specified function
+    client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "system",
+                    "content": [
+                        {
+                            "type": "text",
+                            "text": system_prompt
+                        }
+                    ]
+                },
+                {
+                    "role": "user",
+                    "content": [
+                        {
+                            "type": "text",
+                            "text": user_context
+                        }
+                    ]
+                }
+            ],
+            response_format={
+                "type": "json_schema",
+                "json_schema": {
+                    "name": "life_status_report",
+                    "strict": True,
+                    "schema": {
+                        "type": "object",
+                        "properties": {
+                            "life_score": {
+                                "type": "object",
+                                "description": "Numerical life scores across different areas.",
+                                "properties": {
+                                    "overall": {
+                                        "type": "number",
+                                        "description": "Overall life score, average of all area scores."
+                                    },
+                                    "personal_growth": {
+                                        "type": "number",
+                                        "description": "Life score for personal growth."
+                                    },
+                                    "health_and_wellness": {
+                                        "type": "number",
+                                        "description": "Life score for health and wellness."
+                                    },
+                                    "mental_well_being": {
+                                        "type": "number",
+                                        "description": "Life score for mental well-being."
+                                    },
+                                    "career_growth": {
+                                        "type": "number",
+                                        "description": "Life score for career growth."
+                                    },
+                                    "relationship": {
+                                        "type": "number",
+                                        "description": "Life score for relationships."
+                                    }
+                                },
+                                "required": [
+                                    "overall",
+                                    "personal_growth",
+                                    "health_and_wellness",
+                                    "mental_well_being",
+                                    "career_growth",
+                                    "relationship"
+                                ],
+                                "additionalProperties": False
+                            },
+                            "mantra_of_the_week": {
+                                "type": "string",
+                                "description": "A very short encouragement quote that encapsulates the user's journey to achieve their goals."
+                            },
+                            "this_week_focus": {
+                                "type": "array",
+                                "description": "List of the user's top 3 most important areas with focus.",
+                                "items": {
+                                    "type": "object",
+                                    "properties": {
+                                        "area": {
+                                            "type": "string",
+                                            "description": "The area of focus (e.g., 'personal_growth')."
+                                        },
+                                        "focus": {
+                                            "type": "string",
+                                            "description": "The focus within that area."
+                                        }
+                                    },
+                                    "required": ["area", "focus"],
+                                    "additionalProperties": False
+                                }
+                            },
+                            "action_items": {
+                                "type": "array",
+                                "description": "List of suggested concrete action items for the user.",
+                                "items": {
+                                    "type": "object",
+                                    "properties": {
+                                        "area": {
+                                            "type": "string",
+                                            "description": "The area of action (e.g., 'personal_growth')."
+                                        },
+                                        "focus": {
+                                            "type": "string",
+                                            "description": "Concrete action items in that area."
+                                        }
+                                    },
+                                    "required": ["area", "focus"],
+                                    "additionalProperties": False
+                                }
+                            }
+                        },
+                        "required": [
+                            "life_score",
+                            "mantra_of_the_week",
+                            "this_week_focus",
+                            "action_items"
+                        ],
+                        "additionalProperties": False
+                    }
+                }
+            }
+            ,
+            temperature=0.5,
+            max_tokens=3000,
+            top_p=1,
+            frequency_penalty=0,
+            presence_penalty=0
+        )
+
+        # Get response and convert into dictionary
+        reports = json.loads(response.choices[0].message.content)
+
+    except Exception as e:
+        logger.error(f"OpenAI API call failed: {e}", extra={'user_id': user_id, 'endpoint': function_name})
+        raise e
+
+    # Step 4: Return the JSON reports
+    logger.info(f"User life status generated successfully for user {user_id}", extra={'user_id': user_id, 'endpoint': function_name})
+    return reports
+
 def get_api_key(api_key_header: str = Security(api_key_header)) -> str:
     if (api_key_header in api_keys):
         return api_key_header