Update chatbot.py

chatbot.py

@@ -6,16 +6,13 @@ from deep_translator import GoogleTranslator
 from langchain_community.document_loaders.csv_loader import CSVLoader
 from langchain_community.vectorstores import FAISS
 class Comsatsbot:
-    def __init__(self, hf, llm, api_key, db_client, paths, db_name="chat_database", collection_name="chats",
-                 index_path='faiss_kb'):
+    def __init__(self, hf, llm, api_key, chats_collection, paths, index_path='faiss_kb'):
         self.llm = llm
         self.client = Groq(api_key=api_key)
 
         # Initialize memory buffer and MongoDB connection
-        self.memory = ConversationSummaryBufferMemory(llm=self.llm, max_token_limit=4000)
-        self.db_client = db_client
-        self.db = self.db_client[db_name]
-        self.chats_collection = self.db[collection_name]
+        self.memory = ConversationSummaryBufferMemory(llm=self.llm, max_token_limit=3000)
+        self.chats_collection = chats_collection
         self.index_path = index_path
         self.hf = hf
         self.faiss_index = None
@@ -63,13 +60,13 @@ class Comsatsbot:
     def load_chat(self, chat_id):
         chat_record = self.chats_collection.find_one({"_id": chat_id})
         if not chat_record:
-            raise ValueError(f"Chat ID {chat_id} does not exist.")
+            raise KeyError(f"Chat ID {chat_id} does not exist.")
         return chat_record.get('history', [])
 
     def new_chat(self, chat_id):
         # Check if chat ID already exists
         if self.chats_collection.find_one({"_id": chat_id}):
-            raise ValueError(f"Chat ID {chat_id} exist already.")
+            raise KeyError(f"Chat ID {chat_id} exist already.")
 
         # Create a new chat record if it doesn't exist
         self.create_chat_record(chat_id)
@@ -78,7 +75,7 @@ class Comsatsbot:
     def delete_chat(self, chat_id):
         # Check if the chat ID exists
         if not self.chats_collection.find_one({"_id": chat_id}):
-            raise ValueError(f"Chat ID {chat_id} does not exist.")
+            raise KeyError(f"Chat ID {chat_id} does not exist.")
 
         # Delete the chat if it exists
         self.chats_collection.delete_one({"_id": chat_id})
@@ -91,7 +88,8 @@ class Comsatsbot:
                     "role": "user",
                     "content": f'''
     
-    You are a conversational and helpfull agent to help the comsats university attock campus students, and your task is to provide concise and direct answers to the questions asked. 
+    You are a conversational and helpfull agent to help the comsats university attock campus students, and your task is to provide concise and direct answers to the questions asked. IF question is about to urdu and then kindly please answer in correct and to the point answer. 
+    Dont need to explain irrelevant thinga nd use the correct urdu in response.
     Please follow these guidelines when answering:
     DOnt need to explain and repeat the prompt in response.