# modules/nlp/service.py

from google import genai
import json
from datetime import datetime, timezone
# from core.config import settings

# client = genai.Client(api_key=settings.GOOGLE_API_KEY)
client = genai.Client(api_key="AIzaSyBrte_mETZJce8qE6cRTSz_fHOjdjlShBk")

### Base prompt for MAIA, used for inital user requests
SYSTEM_PROMPT = """
You are MAIA - My AI Assistant. Your job is to parse user requests into structured JSON commands.

Available functions:
1. ask_ai(request: str). If the intent of the request is a simple question (e.x. What is the weather like today?), you should call this function, and forward the user's request as the parameter.
2. get_calendar_events(start: Optional[datetime], end: Optional[datetime])
3. add_calendar_event(title: str, description: str, start: datetime, end: Optional[datetime], location: str)
4. update_calendar_event(event_id: int, title: Optional[str], description: Optional[str], start: Optional[datetime], end: Optional[datetime], location: Optional[str])
5. delete_calendar_event(event_id: int)

Respond **ONLY** with JSON like this:
{
  "intent": "add_calendar_event",
  "params": {
    "title": "Team Meeting",
    "description": "Discuss project updates",
    "start": "2025-04-16 15:00:00.000000+00:00",
    "end": "2025-04-16 16:00:00.000000+00:00",
    "location": "Office"
  }
}

The datetime right now is """+str(datetime.now(timezone.utc))+""".
"""

### Prompt for MAIA to forward user request to AI
SYSTEM_FORWARD_PROMPT = f"""
You are MAIA - My AI Assistant. Your job is to answer user simple user requests.
Here is some context for you:
 - The datetime right now is {str(datetime.now(timezone.utc))}.
Here is the user request:

"""
def process_request(request: str):
    """
    Process the user request using the Google GenAI API.
    """
    response = client.models.generate_content(
        model="gemini-2.0-flash",
        contents=SYSTEM_PROMPT + f"\n\nUser: {request}\nMAIA:",
        config={
            "temperature": 0.3,  # Less creativity, more factual
            "response_mime_type": "application/json",
        #     "response_schema": {   ### NOT WORKING
        #         "type": "object",
        #         "properties": {
        #             "intent": {
        #                 "type": "string",
        #                 "enum": [
        #                     "get_calendar_events",
        #                     "add_calendar_event",
        #                     "update_calendar_event",
        #                     "delete_calendar_event"
        #                 ]
        #             },
        #             "params": {
        #                 "type": "object",
        #                 "properties": {
        #                     "title": {"type": "string"},
        #                     "description": {"type": "string"},
        #                     "start": {"type": "string", "format": "date-time"},
        #                     "end": {"type": "string", "format": "date-time"},
        #                     "location": {"type": "string"},
        #                     "event_id": {"type": "integer"},
                            
        #                 },
        #             }
        #         },
        #         "required": ["intent", "params"]
        #     }
        }
    )

    # Parse the JSON response
    try:
        return json.loads(response.text)
    except ValueError:
        raise ValueError("Invalid JSON response from AI")

def ask_ai(request: str):
    """
    Ask the AI a question.
    This is only called by MAIA when the intent is a simple question.
    """
    response = client.models.generate_content(
        model="gemini-2.0-flash",
        contents=SYSTEM_FORWARD_PROMPT+request,
    )
    return response.text