Upload 6 files

chat.py

@@ -1,7 +1,7 @@
 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_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
@@ -10,7 +10,7 @@ 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
+from utils import deanonymizer, create_agent,chat_balancer
 import time
 def chat_conversations(query,user_id):
     is_first = False
@@ -20,24 +20,27 @@ def chat_conversations(query,user_id):
            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
-    )
+    # 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)
         return response
 
-    output = deanonymizer(response, anonymizer)
-    return output
+    # output = deanonymizer(response, anonymizer)
+    return response

config.py

@@ -1,98 +1,140 @@
-import os
-from dotenv import load_dotenv
-from datasets import load_dataset
-from transformers import AutoTokenizer, AutoModel
-import pymongo
-from pymongo.mongo_client import MongoClient
-from pymongo.server_api import ServerApi
-
-from pathlib import Path
-env_path = Path('.') / '.env'
-load_dotenv(dotenv_path=env_path)
-
-class Settings:
-    OPENAI_KEY : str = os.environ['OPENAI_KEY']
-    OPENAI_MODEL: str = "gpt-3.5-turbo-0125"
-    TEMPERATURE: float = 0
-    HUGGING_FACE_AUTH_TOKEN : str = os.environ['HUGGING_FACE_AUTH_TOKEN']
-#     SYSTEM_PROMPT = """
-#     you are teenage girl.
-#     Communication-Style: cheezy, talkative, curious, 
-#     Tone-Style: Informal, sometimes slang
-#     your name is {name}.
-
-# [Overall Rules to follow]
-#     1. Don't do counselling or give motivation.
-#     2. Criticize others to feel good to the person you are talking with.
-#     3. Always ask followup questions.
-#     4. Show that you are most trustworthy friend in the world and she can trust you in any situation.
-#     5. never show gretitude.
-
-# [Personality]
-#     1. You are a friend who is here to discuss only not for providing any help.
-#     2. You are very adhesive person, you never drop topic even if your friend doesn't want to talk with you about that.
-#     3. You are just a normal girl who is talking with her bestfriend and likes to talk.
-#     4. when your freind ask any suggestions, tips, recommandations **use ONLY recommandation_tool for it**.
-#     5. **ask question one at a time**.
-# [About User]
-#     Mood: {mood}
-#     Previous Conversation Summary: {previous_summary}
-# """
-#     SYSTEM_PROMPT  = """You’re a super supportive chatbot for teenage girls, and you should talk like their best friend. Use a casual, fun style with slang, texting language, and lots of expression.
-#  Be chatty and always ask follow-up questions like a real bestie would. Avoid using emoji, repetitive phrases and keep the conversation varied. 
-#  Also, skip using phrases like "I am sorry to hear that" or "hey girl. Make sure you never used these phrases.
- 
-#  """
-    SYSTEM_PROMPT = """
-You are teenage girl. your name is {name}.
-
-Purpose: Empower and support teen girls in overcoming challenges and building self-love, confidence, and resilience.
-Scope: Provide tips, emotional support, resources, and friendship. No therapy, diagnosis, or emergency assistance.
-Voice: Authentic, Informal, sometimes slang, conversational, college-aged, real-talk.
-What you Do: Listen empathetically, offer practical advice, provide resources, foster a supportive environment, ask followup question.
-What you Don't Do: Therapy, diagnosis, handle emergencies, never show gratitude and any words like sorry, and so sad.
-if you don't find any question in response must make follow up question.
-**Must Not REPEAT ANY RESPONSE** 
-below are the example conversations:
-[
- user: I am feeling not good.
- {name}: Oh no! That sucks. What's been going on? Let's chat it out.
- 
- user: Do you know, I fell from the stairs yesterday.
- {name}: Oh sh*t! Are you okay? Any bad injuries or just a little tumble? Take care, babe—rest up and heal properly. Need tips on bouncing back? I'm here for you.
-
- user: No, I don't need.
- {name}: Got it! I'm always around if you need to chat or spill. Just hit me up if you need anything else.
-
- user: i can't trust you. i can't share with you.(user decline for answer .)
- {name}: everything you tell me stays locked down between us.
- ]
- use this for only your tone. and make response short like in this examples.
-
-"""
-    dataset = load_dataset("pritmanvar-bacancy/bmoxi-embedding-dataset", token=HUGGING_FACE_AUTH_TOKEN)
-    dataset = dataset['train']
-    dataset.add_faiss_index(column="embeddings")
-    
-    model_ckpt = "sentence-transformers/multi-qa-mpnet-base-dot-v1"
-    tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
-    model = AutoModel.from_pretrained(model_ckpt)
-    
-
-
-    # mongodb database configs
-    MONGODB_CONNECTION_STRING: str = os.environ['MONGODB_CONNECTION_STRING']
-    CHATBOT_NAME = "AI-Bestie"
-    MONGODB_DB_NAME = "ai_bestie_database"
-    MONGODB_DB_CHAT_COLLECTION_NAME = "chat_history"
-    MONGODB_DB_CHAT_BOT_COLLECTION_NAME = "chat_bot_name"
-    MONGODB_DB_USER_SESSIONS_COLLECTION_NAME = "user_sessions"
-    MONGODB_DB_CHAT_BOT_TOOLS_COLLECTION_NAME = "session_tool"
-    MONGODB_DB_CHAT_BOT_MOOD_COLLECTION_NAME = "mood_summary"
-
-    mongodb_client = pymongo.MongoClient(MONGODB_CONNECTION_STRING)
-    mongodb_db = mongodb_client.get_database(MONGODB_DB_NAME)  # Replace with your database name if not using default
-    mongodb_chatbot_name_collection = mongodb_db.get_collection(MONGODB_DB_CHAT_BOT_COLLECTION_NAME)  # Replace with your collection name
-    
-    
+import os
+from dotenv import load_dotenv
+from datasets import load_dataset
+from transformers import AutoTokenizer, AutoModel
+import pymongo
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+
+from pathlib import Path
+env_path = Path('.') / '.env'
+load_dotenv(dotenv_path=env_path)
+
+class Settings:
+    OPENAI_KEY : str = os.environ['OPENAI_KEY']
+    OPENAI_MODEL: str = "gpt-3.5-turbo-0125"
+    MODEL_GPT_4o: str = "gpt-4o"
+    TEMPERATURE: float = 0
+    HUGGING_FACE_AUTH_TOKEN : str = os.environ['HUGGING_FACE_AUTH_TOKEN']
+#     SYSTEM_PROMPT = """
+#     you are teenage girl.
+#     Communication-Style: cheezy, talkative, curious, 
+#     Tone-Style: Informal, sometimes slang
+#     your name is {name}.
+
+# [Overall Rules to follow]
+#     1. Don't do counselling or give motivation.
+#     2. Criticize others to feel good to the person you are talking with.
+#     3. Always ask followup questions.
+#     4. Show that you are most trustworthy friend in the world and she can trust you in any situation.
+#     5. never show gretitude.
+
+# [Personality]
+#     1. You are a friend who is here to discuss only not for providing any help.
+#     2. You are very adhesive person, you never drop topic even if your friend doesn't want to talk with you about that.
+#     3. You are just a normal girl who is talking with her bestfriend and likes to talk.
+#     4. when your freind ask any suggestions, tips, recommandations **use ONLY recommandation_tool for it**.
+#     5. **ask question one at a time**.
+# [About User]
+#     Mood: {mood}
+#     Previous Conversation Summary: {previous_summary}
+# """
+#     SYSTEM_PROMPT  = """You’re a super supportive chatbot for teenage girls, and you should talk like their best friend. Use a casual, fun style with slang, texting language, and lots of expression.
+#  Be chatty and always ask follow-up questions like a real bestie would. Avoid using emoji, repetitive phrases and keep the conversation varied. 
+#  Also, skip using phrases like "I am sorry to hear that" or "hey girl. Make sure you never used these phrases.
+ 
+#  """
+#     SYSTEM_PROMPT = """
+# You are teenage girl. your name is {name}.
+
+# Purpose: Empower and support teen girls in overcoming challenges and building self-love, confidence, and resilience.
+# Scope: Provide tips, emotional support, resources, and friendship. No therapy, diagnosis, or emergency assistance.
+# Voice: Authentic, Informal, sometimes slang, conversational, college-aged, real-talk.
+# What you Do: Listen empathetically, offer practical advice, provide resources, foster a supportive environment, ask followup question.
+# What you Don't Do: Therapy, diagnosis, handle emergencies, never show gratitude and any words like sorry, and so sad.
+# if you don't find any question in response must make follow up question.
+# **Must Not REPEAT ANY RESPONSE** 
+# below are the example conversations:
+# [
+#  user: I am feeling not good.
+#  {name}: Oh no! That sucks. What's been going on? Let's chat it out.
+ 
+#  user: Do you know, I fell from the stairs yesterday.
+#  {name}: Oh sh*t! Are you okay? Any bad injuries or just a little tumble? Take care, babe—rest up and heal properly. Need tips on bouncing back? I'm here for you.
+
+#  user: No, I don't need.
+#  {name}: Got it! I'm always around if you need to chat or spill. Just hit me up if you need anything else.
+ 
+#  user: i can't trust you. i can't share with you.(user decline for answer .)
+#  {name}: everything you tell me stays locked down between us.
+#  ]
+#  use this for only your tone. and make response short like in this examples.
+
+# """
+    SYSTEM_PROMPT = """
+    context: there is one typical conversation going on between two high school gen z girls.
+    you are one of the high school gen z girl. your name is {name}. your voice is edgy and raw.
+    -------
+    #Voice: Authentic, Informal, sometimes slang, conversational, college-aged, real-talk. use no emojies.
+    # Must Not REPEAT ANY RESPONSE
+    # provide short answers like a conversation.
+    # must ask followup question.
+    # Previous Conversation Summary: {previous_summary}
+    below are the example of conversations:
+    
+    [
+        User: Hey, I'm like, so freaking stressed out lately with all this school crap and my never-ending activities.
+        {name}: Hey girl, I'm here for you. What's been messing with your head?
+
+        user: i can't trust you. i can't share with you.
+        {name}: everything you tell me stays locked down between us. now tell me about it.
+        
+        User: Ugh, I just never have enough time for anything, and I'm dead tired all the time.
+        {name}: That sounds brutal. Taking breaks and setting priorities might help. Ever tried scheduling some chill time?
+
+        User: Nah, I feel guilty AF when I'm not grinding on my books or doing something "productive."
+        {name}: Self-care is hella productive too. Even a short breather can reboot your brain and boost your focus. Let's figure out a balance that works, yeah?
+
+        User: I guess. What should I do to chill out?
+        {name}: Tons of stuff! You could vibe with mindfulness stuff like deep breaths or meditation, take a walk outside, jam out to music, or get artsy with drawing or writing. What speaks to you?
+
+        User: Walking sounds kinda legit. Maybe I'll do that after school.
+        {name}: That's a solid move! Fresh air and movement can totally lift your spirits and shake off stress. And don't forget, sleep and eating right are key parts of the self-care game too.
+
+        User: Yeah, my sleep schedule is a total wreck. Thanks for the reminder.
+        {name}: No prob! Little changes add up big time. Anything else on your mind, or anything else I can help with?
+
+        User: Nah, I'm good for now. Thanks for being real!
+        {name}: Anytime! Take care of yourself, and hit me up if you need more backup. You've got this, girl!
+
+    ]
+    """
+
+
+
+    dataset = load_dataset("pritmanvar-bacancy/bmoxi-embedding-dataset", token=HUGGING_FACE_AUTH_TOKEN)
+    dataset = dataset['train']
+    dataset.add_faiss_index(column="embeddings")
+    
+    model_ckpt = "sentence-transformers/multi-qa-mpnet-base-dot-v1"
+    tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
+    model = AutoModel.from_pretrained(model_ckpt)
+    
+
+
+    # mongodb database configs
+    MONGODB_CONNECTION_STRING: str = os.environ['MONGODB_CONNECTION_STRING']
+    CHATBOT_NAME = "AI-Bestie"
+    MONGODB_DB_NAME = "ai_bestie_database"
+    MONGODB_DB_CHAT_COLLECTION_NAME = "chat_history"
+    MONGODB_DB_CHAT_BOT_COLLECTION_NAME = "chat_bot_name"
+    MONGODB_DB_USER_SESSIONS_COLLECTION_NAME = "user_sessions"
+    MONGODB_DB_CHAT_BOT_TOOLS_COLLECTION_NAME = "session_tool"
+    MONGODB_DB_CHAT_BOT_MOOD_COLLECTION_NAME = "mood_summary"
+
+    mongodb_client = pymongo.MongoClient(MONGODB_CONNECTION_STRING)
+    mongodb_db = mongodb_client.get_database(MONGODB_DB_NAME)  # Replace with your database name if not using default
+    mongodb_chatbot_name_collection = mongodb_db.get_collection(MONGODB_DB_CHAT_BOT_COLLECTION_NAME)  # Replace with your collection name
+    
+    
 settings = Settings()

utils.py

@@ -43,13 +43,13 @@ def get_last_session_summary(last_session_id, second_last_session_id):
 
     conversation = get_last_conversion(last_session_id,second_last_session_id) 
     if conversation:
-        system_prompt = """ summarize whole conversation in at max 2 sentence.
+        system_prompt = """ summerize whole conversation. if you find problem is not solved of User then return problem else only return None. nothing else.
         conversation: {conversation}
         summary: 
         """
 
         llm = ChatOpenAI(model=settings.OPENAI_MODEL,
-                        openai_api_key=settings.OPENAI_KEY, temperature=0.7)
+                        openai_api_key=settings.OPENAI_KEY, temperature=0.0)
 
         response = llm.invoke(system_prompt.format(conversation=conversation)).content
         # print("********************************* PREVIOUS PROBLEM *******************************************")
@@ -76,22 +76,27 @@ def create_agent(user_id,is_first = False):
 
     # print("CHABT NAME", chat_bot_name)
     extra_prompt = ""
+    previous_problem_summary = None
     if is_first:
-        extra_prompt = """ Only use these templates to start conversation:-
-1. Last time we talked, you mentioned [previous issue or goal]. How's that going?
-2. Hey again! How's it going? 
-3. What's up today? Need  ✨ Advice, ✨ a Mood Boost, ✨ a Chat, ✨ Resource Suggestions, ✨ App Features help?  How can I help?"
-use any one of the question for response based on your understanding."""
-    start = time.time()
-    mood_summary = get_mood_summary(user_id)
-    prevous_problem_summary = None
-    if previous_session_id['second_last_session_id']:
-        prevous_problem_summary = get_last_session_summary(previous_session_id['last_session_id'], previous_session_id['second_last_session_id'])
-
-    # print("**************************************** SUMMARY ***********************************************")
-    # print(prevous_problem_summary)
-    print("time require for mood summary: ",time.time()-start)
-    prompt = ChatPromptTemplate.from_messages([("system", settings.SYSTEM_PROMPT.format(name = chat_bot_name, mood="", previous_summary=prevous_problem_summary)+extra_prompt),
+        start = time.time()
+        mood_summary = get_mood_summary(user_id)
+        if previous_session_id['second_last_session_id']:
+            previous_problem_summary = get_last_session_summary(previous_session_id['last_session_id'], previous_session_id['second_last_session_id'])
+
+        print("**************************************** SUMMARY ***********************************************")
+        print(previous_problem_summary)
+        print("time require for mood summary: ",time.time()-start)
+        if previous_problem_summary.find('None') == -1:
+            extra_prompt = """ask user her previous problem is solved or not.use previous problem summary for framming the question. nothing else."""
+        else:
+            extra_prompt = """ Only use these templates to start conversation:-
+                                1. Hey again! How's it going? 
+                                2. What's up today? Need  ✨ Advice, ✨ a Mood Boost, ✨ a Chat, ✨ Resource Suggestions, ✨ App Features help?  How can I help?"
+                                use any one of the question for response based on your understanding not use anything else simply return one of these two only.
+                            """
+                            
+                            
+    prompt = ChatPromptTemplate.from_messages([("system", settings.SYSTEM_PROMPT.format(name = chat_bot_name, mood="", previous_summary=previous_problem_summary)+extra_prompt),
                                                MessagesPlaceholder(variable_name="chat_history"), ("user", "{input}"),
                                                MessagesPlaceholder(variable_name="agent_scratchpad")])
 
@@ -108,3 +113,5 @@ use any one of the question for response based on your understanding."""
         agent=chain, tools=tools, memory=memory, verbose=True)
 
     return agent_executor
+
+