Update app.py

app.py

@@ -225,17 +225,31 @@ Javobni faqat matn shaklida bering, kod yoki ortiqcha belgilar kiritmang."""
     def process_response(self, response: str) -> str:
         """Clean up the response"""
         try:
+            # Remove common model instruction tags
             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)
-        
+            # Remove code blocks with language specifications
+            response = re.sub(r"```\w*\n.*?```", "", response, flags=re.DOTALL)
+            
+            # Remove single line code blocks
+            response = re.sub(r"`.*?`", "", response)
+            
+            # Remove any remaining code-like artifacts
+            response = re.sub(r"//.*?$", "", response, flags=re.MULTILINE)  # Remove single line comments
+            response = re.sub(r"/\*.*?\*/", "", response, flags=re.DOTALL)  # Remove multi-line comments
+            response = re.sub(r"[{}<>]", "", response)  # Remove brackets
+            response = re.sub(r"\b(java|python|class|public|private|void)\b", "", response, flags=re.IGNORECASE)  # Remove programming keywords
+            
+            # Clean up multiple spaces and newlines
+            response = re.sub(r'\s+', ' ', response)
+            
+            # Final cleanup
             return response.strip()
         except Exception as e:
             logging.error(f"Error processing response: {str(e)}")
-            return response
+            return response.strip()
 
     def chat(self, message: str, history: List[Tuple[str, str]]) -> str:
         """Process a single chat message"""