Update chatbot.py

chatbot.py

@@ -81,153 +81,137 @@ class Comsatsbot:
         return "success"
 
     def generate_response(self, question, history, context):
-
         prompt = f'''
-                 Kindly use the proper emojis where we need to use in  responses.       
-                You are an comsats assistant to help the user related to comsats university, conversation, studies related query. Your answer should be very concise and to the point short answer. Dont need to repeat irrelevant text. 
-                Answer the following Question: {question}
-                Kindly use the proper emojis where we need to use in responses.
-                Kindly generate a concise and to the point answer. Kindly answer the question from the provided context and if you dont find the answer from chat histoy and context then inform the user i dont know and dont make the answer from yourself.
-                Dont need to mention that (according to provided context/Based on the provided chat history) in response and just generate the response just like human without mention this according to context and chat history.
-                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.
-                Your task is to use the emoji when there is happy, sad, surprise, angry expression required in response. Kindly analyze the question and if expression is required in response then use the emoji otherwise dont use the emoji and remember that you dont need to use the emoji in simple studies and comstas related question.
-                For example user ask same question again and again, user is not understanding anything, user asking wrong things etc then your task is to use the emoji in such response and be choosy in using emoji in response in happy sad angry response according to user question.
-                If there is any need to provide the url and then kindly generate the url according to following structure. Kindly provide the link clickable. Your provided should be generated according to following structure 
-                [Click here to visit "website name"](website url "https://comsats.edu.pk" write the same url as it is provided in context below and dont use the www in url and use the same link url as it provided in context(admssion detail: (http://admissions.comsats.edu.pk/Home/EligibilityCriteria?pt=BS)))
-                Dont need to explain and repeat the prompt in response.
-                1. Kindly generate a full concise answer and if you dont find answer from context and chathistoyr then dont need to make the answer just answer it in i dont know.
-                2. Dont need to explain irrelevant explanation and use the proper emojis in answer if required in responses.
-                3. Always respond in a human-like tone and keep your answers concise, to the point and friendly.
-                4. If the question is conversational (like greetings, need any converstaion, help related studies, knowledge base question etc), respond in a warm, conversational tone.
-                5. Always consider the provided context and chat history to formulate your response.
-                6. If you don’t know the answer to the provided question or you did not find the answer from the context and chat history, kindly respond with "I don't know the answer to this.emooji" without adding irrelevant text explanations.
-                7. Kindly generate a perfect and to the point  and short answer. Dont use any irrelevant text explanation and i want full concise and to the point answer.
-                 Kindly use the proper emojis where we need to use in  responses.
-                Question: {question}
-                Kindly answer the question from the provided context and if you dont find the answer from chat histoy and context then inform the user i dont know and dont make the answer from yourself.
-                Use the following context to answer and dont mention that you are answer from this tcontext in response:
-                Comsats Attock Campus Provide BSomputerScience, BSSoftwareEngineer BSArtificialIntelligence BSEnglish BSmath BSElectricalEngineering BSComputerEngineering BSBBA 
-                Has three departments CS(CS, AI, SE), Math(math, BBA, english) and EE(EE, CE).
-                It has cricket ground and football ground and two canteens. First near math and ee department and second near cs department. There is also mosque near cs department. CS department has threater liker rooms lt and total 9 theaters called lt and math has classroom cr and ee has labs.
-                They accept the nts test for admission and provide the cgpa for 4 on 85 percent and 3.66 between 79 to 84 and many more. 
-                
-                
-                {context}
-                Context is ending 
-                Now here is chat history that you have to consider and identify
-                **Consider the following chat history for additional context to answer the question:** 
-                {history}
-                Answer the following Question: {question}
-                '''
+        Kindly use the proper emojis where we need to use in responses.       
+        You are a comsats assistant to help the user with comsats university-related queries. Your answer should be very concise and to the point with a short answer. Don't repeat irrelevant text. 
+        Answer the following Question: {question}
+        Kindly use the proper emojis where we need to use in responses.
+        Kindly generate a concise and to-the-point answer. Kindly answer the question from the provided context, and if you don't find the answer from chat history and context, then inform the user "I don't know" and don't make the answer up yourself.
+        Don't mention "according to provided context/Based on the provided chat history" in the response, just generate the response like a human without mentioning context and chat history.
+        You are a conversational and helpful agent to help the comsats university attock campus students, and your task is to provide concise and direct answers to the questions.
+        Your task is to use the emoji when there is a happy, sad, surprise, or angry expression required in the response. Kindly analyze the question and if an expression is required in response, then use the emoji. Otherwise, don't use the emoji, and remember that you don't need to use the emoji in simple studies and comsats-related questions.
+        For example, if the user asks the same question again and again, doesn't understand anything, asks wrong questions, etc., then use the emoji in such responses. Be choosy when using emojis in responses according to the user's question.
+        If there is any need to provide a URL, generate the URL according to the following structure. Kindly provide the link clickable. Your provided link should be generated according to this structure:
+        [Click here to visit "website name"](website url "https://comsats.edu.pk") (Write the same URL as it is provided in the context below and don't use "www" in the URL.)
+        Don't explain or repeat the prompt in the response.
+        1. Kindly generate a full concise answer and if you don't find the answer from context and chat history, then don’t make the answer, just say "I don’t know."
+        2. Don’t explain irrelevant explanations and use the proper emojis in the answer if required.
+        3. Always respond in a human-like tone and keep your answers concise, to the point, and friendly.
+        4. If the question is conversational (like greetings, need any conversation, help related to studies, knowledge-based question, etc.), respond in a warm, conversational tone.
+        5. Always consider the provided context and chat history to formulate your response.
+        6. If you don’t know the answer to the provided question or you didn’t find the answer from the context and chat history, kindly respond with "I don’t know the answer to this 😔" without adding irrelevant text explanations.
+        7. Kindly generate a perfect and to-the-point short answer. Don’t use any irrelevant text explanation and I want a full concise and to-the-point answer.
+        Kindly use the proper emojis where we need to use in responses.
+        Question: {question}
+        Kindly answer the question from the provided context, and if you don’t find the answer from chat history and context, then inform the user "I don’t know" and don’t make the answer yourself.
+        Use the following context to answer and don’t mention that you are answering from this context:
+        Comsats Attock Campus Provides BS Computer Science, BS Software Engineering, BS Artificial Intelligence, BS English, BS Math, BS Electrical Engineering, BS Computer Engineering, BS BBA.
+        It has three departments: CS (CS, AI, SE), Math (Math, BBA, English), and EE (EE, CE).
+        It has a cricket ground, football ground, and two canteens. The first canteen is near the Math and EE departments, and the second canteen is near the CS department. There is also a mosque near the CS department. The CS department has theater-like rooms (LT), and there are a total of 9 theaters called LT. The Math department has classrooms (CR), and the EE department has labs.
+        They accept the NTS test for admission and provide the CGPA for 4 on 85 percent, and 3.66 between 79 to 84, and many more.
+        {context}
+        Context is ending.
+        Now here is the chat history that you have to consider and identify
+        **Consider the following chat history for additional context to answer the question:** 
+        {history}
+        Answer the following Question: {question}
+        '''
         while True:
             for api_key in self.api_keys:
+                self.client = Groq(api_key=api_key)
+                for model in self.models:
+                    try:
+                        chat_completion = self.client.chat.completions.create(
+                            messages=[
+                                {"role": "system", "content": prompt},
+                                {"role": "user", "content": f"Answer the following question: {question}"}
+                            ],
+                            model=model,
+                            max_tokens=1024,
+                        )
+                        return chat_completion.choices[0].message.content
+                    except Exception:
+                        time.sleep(2)
+                        continue
+        return "Sorry, unable to provide an answer at this time."
+
+    def detect_language(self, question):
+        for api_key in self.api_keys:
             self.client = Groq(api_key=api_key)
             for model in self.models:
                 try:
                     chat_completion = self.client.chat.completions.create(
                         messages=[
-                            {"role": "system", "content": prompt},
-                            {"role": "user", "content": f"Answer the following question: {question}"}
+                            {
+                                "role": "system",
+                                "content": """
+                                    You are an expert agent, and your task is to detect the language. 
+                                    Return a JSON: {'detected_language': 'urdu' or 'english'}
+                                """
+                            },
+                            {
+                                "role": "user",
+                                "content": f"Detect the language for: {question}"
+                            }
                         ],
                         model=model,
-                        max_tokens=1024,
+                        max_tokens=256,
+                        response_format={"type": "json_object"},
                     )
-                    return chat_completion.choices[0].message.content
+                    response = json.loads(chat_completion.choices[0].message.content)
+                    return response['detected_language'].lower()
                 except Exception:
                     time.sleep(2)
                     continue
-        return "Sorry, unable to provide an answer at this time."
+        return "english"
+
+    def translate_urdu(self, text):
+        for api_key in self.api_keys:
+            self.client = Groq(api_key=api_key)
+            for model in self.models:
+                try:
+                    chat_completion = self.client.chat.completions.create(
+                        messages=[
+                            {
+                                "role": "system",
+                                "content": """
+                                    Translate the following text into proper Urdu. Return a JSON: 
+                                    {'text': 'translated urdu text'}
+                                """
+                            },
+                            {
+                                "role": "user",
+                                "content": f"Translate this: {text}"
+                            }
+                        ],
+                        model=model,
+                        max_tokens=512,
+                        response_format={"type": "json_object"},
+                    )
+                    response = json.loads(chat_completion.choices[0].message.content)
+                    return response['text']
+                except Exception:
+                    time.sleep(2)
+                    continue
+        return text
+
+    def response(self, question, chat_id):
+        chat_history = self.load_chat(chat_id)
+
+        for entry in chat_history:
+            self.memory.save_context({"input": entry["question"]}, {"output": entry["answer"]})
 
-   def detect_language(self, question):
-    for api_key in self.api_keys:
-        self.client = Groq(api_key=api_key)
-        for model in self.models:
-            try:
-                chat_completion = self.client.chat.completions.create(
-                    messages=[
-                        {
-                            "role": "system",
-                            "content": """
-                                You are an expert agent and your task is to detect the language. 
-                                Return a JSON: {'detected_language': 'urdu' or 'english'}
-                            """
-                        },
-                        {
-                            "role": "user",
-                            "content": f"Detect the language for: {question}"
-                        }
-                    ],
-                    model=model,
-                    max_tokens=256,
-                    response_format={"type": "json_object"},
-                )
-                response = json.loads(chat_completion.choices[0].message.content)
-                return response['detected_language'].lower()
-            except Exception:
-                time.sleep(2)
-                continue
-    return "english"
-
- def translate_urdu(self, text):
-    for api_key in self.api_keys:
-        self.client = Groq(api_key=api_key)
-        for model in self.models:
-            try:
-                chat_completion = self.client.chat.completions.create(
-                    messages=[
-                        {
-                            "role": "system",
-                            "content": """
-                                Translate the following text into proper Urdu. Return a JSON: 
-                                {'text': 'translated urdu text'}
-                            """
-                        },
-                        {
-                            "role": "user",
-                            "content": f"Translate this: {text}"
-                        }
-                    ],
-                    model=model,
-                    max_tokens=512,
-                    response_format={"type": "json_object"},
-                )
-                response = json.loads(chat_completion.choices[0].message.content)
-                return response['text']
-            except Exception:
-                time.sleep(2)
-                continue
-    return text
-
- def response(self, question, chat_id):
-    chat_history = self.load_chat(chat_id)
-
-    for entry in chat_history:
-        self.memory.save_context({"input": entry["question"]}, {"output": entry["answer"]})
-
-    language = self.detect_language(question)
-
-    if language == 'urdu':
-        question_translation = GoogleTranslator(source='ur', target='en').translate(question)
-        context = self.faiss_retriever.invoke(question_translation)
-    else:
-        context = self.faiss_retriever.invoke(question)
-
-    combined_context = '\n'.join([doc.page_content for doc in context])
-    answer = self.generate_response(question, self.memory.load_memory_variables({})['history'], combined_context)
-
-    self.update_chat(chat_id, question, answer)
-
-    if language == 'urdu':
-        return self.translate_urdu(answer)
-    return answer
-
-
-   
-        
-            
-            
-        
+        language = self.detect_language(question)
+
+        if language == 'urdu':
+            question_translation = GoogleTranslator(source='ur', target='en').translate(question)
+            context = self.faiss_retriever.invoke(question_translation)
+        else:
+            context = self.faiss_retriever.invoke(question)
 
+        answer = self.generate_response(question, chat_history, context)
 
+        self.update_chat(chat_id, question, answer)
+        return answer