from langchain.memory import ConversationBufferWindowMemory
from langchain_community.chat_models import ChatOpenAI
from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory
# from langchain_experimental.data_anonymizer import PresidioReversibleAnonymizer
from langchain.agents import AgentExecutor
from langchain.agents.format_scratchpad import format_to_openai_functions
from langchain.agents.output_parsers import OpenAIFunctionsAgentOutputParser
from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain.schema.runnable import RunnablePassthrough
from langchain_core.utils.function_calling import convert_to_openai_function

from database_functions import set_chat_bot_name,isFirstSession
from utils import deanonymizer, create_agent
import time


is_second = False
use_extra = False
def chat_conversations(query,user_id):
    is_first = False
    global is_second ,use_extra
    if is_second:
        if 'no' in query:
            query= """ "Okk no problem!
            What's up today? Need  ✨ Advice, ✨ a Mood Boost, ✨ a Chat, ✨ Resource Suggestions, ✨ App Features help?  How can I help?"
            Ask this question only without thinking anything.
            """
            is_second = False
        else:
            use_extra = True
    if query == "START":
       if isFirstSession(user_id):
           is_second = True
           query = """ return this message without changing it.:- 
           also don't thought about it. Hey! I'm your BMOXI AI bestie, ready to help you tackle the wild ride of teen life. Want to give me a name? Type it below, or just say 'no' if you're cool with 'AI Bestie'!"""
       else:
          is_first = True
          query = ""

    # anonymizer = PresidioReversibleAnonymizer(
    #     analyzed_fields=["PHONE_NUMBER",
    #                      "EMAIL_ADDRESS", "CREDIT_CARD"],
    #     faker_seed=42,
    # )
    # anonymized_input = anonymizer.anonymize(
    #     query
    # )
    start = time.time()
    agent = create_agent(user_id,is_first)
    print("time to create agent: ",time.time()-start)
    response = agent({"input": query})['output']
    print(response)
    print("time to generate response by agent",time.time()-start)
    
    # response = chat_balancer(query,response)
    if "Okay, from now my name will be " in response:
        set_chat_bot_name(response.split("Okay, from now my name will be ")[-1], user_id)
        if use_extra:
            use_extra = False
            is_second = False
            return response+ "\n What's up today? Need  ✨ Advice, ✨ a Mood Boost, ✨ a Chat, ✨ Resource Suggestions, ✨ App Features help?  How can I help?"
        return response

    # output = deanonymizer(response, anonymizer)
    return response