Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import os
 from langchain_community.vectorstores import FAISS
@@ -14,13 +13,11 @@ TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
 
 class ChatBot:
     def __init__(self):
-        
         self.embeddings = TogetherEmbeddings(
             model="togethercomputer/m2-bert-80M-32k-retrieval",
             together_api_key=TOGETHER_API_KEY
         )
         
-        
         self.vectorstore = FAISS.load_local(
             ".",
             embeddings=self.embeddings,
@@ -28,33 +25,38 @@ class ChatBot:
         )
         self.retriever = self.vectorstore.as_retriever()
         
-        
         self.model = Together(
             model="meta-llama/Llama-3.3-70B-Instruct-Turbo",
-            temperature=0.2,
+            temperature=0.1,  
             max_tokens=128,
             top_k=30,
             together_api_key=TOGETHER_API_KEY
         )
         
-        
         self.memory = ConversationBufferMemory(
             return_messages=True,
             memory_key="chat_history",
             output_key="answer"
         )
         
-        
-        self.template = """Quyidagi kontekst va suhbat tarixiga asoslanib, savolga faqat matn ko'rinishida, kod yoki ortiqcha belgilarsiz, faqat o'zbek tilida tabiiy tarzda javob bering:
+        # Updated prompt template with more specific instructions
+        self.template = """Quyidagi ko'rsatmalarga qat'iy rioya qiling:
+
+1. Faqat o'zbek tilida javob bering
+2. Faqat berilgan ma'lumotlar asosida javob bering
+3. Agar savol tushunarsiz bo'lsa yoki ma'lumot bo'lmasa, "Kechirasiz, bu haqida ma'lumotga ega emasman" deb javob bering
+4. O'zingizdan savol bermang
+5. Javobni takrorlamang
+6. Salomlashish uchun "Assalomu alaykum" yoki "Vaalaykum assalom" dan foydalaning
+
 Kontekst: {context}
 Suhbat Tarixi: {chat_history}
 Savol: {question}
-Javobni faqat matn shaklida bering, kod yoki ortiqcha belgilar kiritmang."""
 
-        
+Javob:"""
+
         self.prompt = ChatPromptTemplate.from_template(self.template)
         
-        
         self.chain = (
             {
                 "context": self.retriever,
@@ -65,35 +67,28 @@ Javobni faqat matn shaklida bering, kod yoki ortiqcha belgilar kiritmang."""
             | self.model
             | StrOutputParser()
         )
-    
+
     def get_chat_history(self) -> str:
-        """Format chat history for the prompt"""
         messages = self.memory.load_memory_variables({})["chat_history"]
         return "\n".join([f"{m.type}: {m.content}" for m in messages])
     
-    import re
-
     def process_response(self, response: str) -> str:
-        """Clean up the response"""
         unwanted_tags = ["[INST]", "[/INST]", "<s>", "</s>"]
         for tag in unwanted_tags:
             response = response.replace(tag, "")
         
-        
         response = re.sub(r"```.*?```", "", response, flags=re.DOTALL)
         response = re.sub(r"print\(.*?\)", "", response)
-    
+        response = re.sub(r'\s+', ' ', response)  # Remove multiple spaces
+        
         return response.strip()
 
-    
     def chat(self, message: str, history: List[Tuple[str, str]]) -> str:
-        """Process a single chat message"""
         try:
             self.memory.chat_memory.add_user_message(message)
             response = self.chain.invoke(message)
             clean_response = self.process_response(response)
             
-            
             if not clean_response or len(clean_response.split()) < 3:
                 clean_response = "Kechirasiz, savolingizni tushunolmadim. Iltimos, batafsilroq savol bering."
             
@@ -103,11 +98,9 @@ Javobni faqat matn shaklida bering, kod yoki ortiqcha belgilar kiritmang."""
             return f"Xatolik yuz berdi: {str(e)}"
 
     def reset_chat(self) -> List[Tuple[str, str]]:
-        """Reset the chat history"""
         self.memory.clear()
         return []
 
-
 def create_demo() -> gr.Interface:
     chatbot = ChatBot()
     
@@ -118,6 +111,7 @@ def create_demo() -> gr.Interface:
         chatbot_interface = gr.Chatbot(
             height=600,
             show_copy_button=True,
+            value=[("", "Assalomu alaykum. Sizga qanday yordam bera olaman?")]  # Initial greeting
         )
         
         with gr.Row():
@@ -131,7 +125,6 @@ def create_demo() -> gr.Interface:
         clear = gr.Button("Yangi suhbat")
         
         def respond(message, chat_history):
-            
             message = message.strip()
             if not message:
                 return "", chat_history
@@ -140,9 +133,13 @@ def create_demo() -> gr.Interface:
             chat_history.append((message, bot_message))
             return "", chat_history
         
+        def reset_chat():
+            chatbot.reset_chat()
+            return [("", "Assalomu alaykum. Sizga qanday yordam bera olaman?")]
+        
         submit.click(respond, [msg, chatbot_interface], [msg, chatbot_interface])
         msg.submit(respond, [msg, chatbot_interface], [msg, chatbot_interface])
-        clear.click(lambda: chatbot.reset_chat(), None, chatbot_interface)
+        clear.click(reset_chat, None, chatbot_interface)
     
     return demo