Spaces:
Running
Running
| 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'] | |
| 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" | |
| 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. | |
| # """ | |
| 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_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() |