Update app.py

app.py

@@ -31,9 +31,9 @@ class ChatBot:
         # Initialize the model
         self.model = Together(
             model="meta-llama/Llama-3.3-70B-Instruct-Turbo",
-            temperature=0.7,
-            max_tokens=128,
-            top_k=50,
+            temperature=0.5,
+            max_tokens=150,
+            top_k=30,
             together_api_key=TOGETHER_API_KEY
         )
         
@@ -45,13 +45,13 @@ class ChatBot:
         )
         
         # Create the prompt template
-        self.template = """<s>[INST] Based on the following context and chat history, answer the question naturally:
+        self.template = """Quyidagi kontekst va suhbat tarixiga asoslanib, savolga o'zbek tilida tabiiy tarzda javob bering:
 
-Context: {context}
+Kontekst: {context}
 
-Chat History: {chat_history}
+Suhbat Tarixi: {chat_history}
 
-Question: {question} [/INST]"""
+Savol: {question}"""
         
         self.prompt = ChatPromptTemplate.from_template(self.template)
         
@@ -74,16 +74,26 @@ Question: {question} [/INST]"""
     
     def process_response(self, response: str) -> str:
         """Clean up the response"""
-        response = response.replace("[/INST]", "").replace("<s>", "").replace("</s>", "")
+        unwanted_tags = ["[INST]", "[/INST]", "<s>", "</s>"]
+        for tag in unwanted_tags:
+            response = response.replace(tag, "")
         return response.strip()
     
     def chat(self, message: str, history: List[Tuple[str, str]]) -> str:
         """Process a single chat message"""
-        self.memory.chat_memory.add_user_message(message)
-        response = self.chain.invoke(message)
-        clean_response = self.process_response(response)
-        self.memory.chat_memory.add_ai_message(clean_response)
-        return clean_response
+        try:
+            self.memory.chat_memory.add_user_message(message)
+            response = self.chain.invoke(message)
+            clean_response = self.process_response(response)
+            
+            # Agar javob to'liq bo'lmasa yoki noto'g'ri bo'lsa, qayta urinib ko'rish
+            if not clean_response or len(clean_response.split()) < 3:
+                clean_response = "Kechirasiz, savolingizni tushunolmadim. Iltimos, batafsilroq savol bering."
+            
+            self.memory.chat_memory.add_ai_message(clean_response)
+            return clean_response
+        except Exception as e:
+            return f"Xatolik yuz berdi: {str(e)}"
 
     def reset_chat(self) -> List[Tuple[str, str]]:
         """Reset the chat history"""
@@ -114,6 +124,11 @@ def create_demo() -> gr.Interface:
         clear = gr.Button("New Chat")
         
         def respond(message, chat_history):
+            # Foydalanuvchi xabarini tozalash
+            message = message.strip()
+            if not message:
+                return "", chat_history
+            
             bot_message = chatbot.chat(message, chat_history)
             chat_history.append((message, bot_message))
             return "", chat_history