Spaces:
Running
Running
| import logging | |
| from collections import defaultdict | |
| from typing import Dict | |
| logger = logging.getLogger(__name__) | |
| class ResponseOptimizer: | |
| """🚀 Public Response Optimizer (Basic Filtering + Length Control)""" | |
| def __init__(self): | |
| self.response_cache = {} # Stores past responses to reduce redundant processing | |
| self.optim_rules = defaultdict(list) # Holds rule-based response refinements | |
| self.cache_limit = 100 # ✅ Limit cache size to avoid overflow | |
| async def optimize_response(self, response: str, context: Dict) -> str: | |
| """✅ Optimizes AI-generated responses (Fast + Secure).""" | |
| # ✅ Return cached response if available | |
| if response in self.response_cache: | |
| logger.info("✅ Returning cached optimized response.") | |
| return self.response_cache[response] | |
| # ✅ Apply context-based optimization (length, profanity) | |
| optimized_response = self.apply_optimizations(response, context) | |
| # ✅ Store in cache (Respect size limit) | |
| if len(self.response_cache) >= self.cache_limit: | |
| oldest_response = next(iter(self.response_cache)) | |
| del self.response_cache[oldest_response] | |
| self.response_cache[response] = optimized_response | |
| return optimized_response | |
| def apply_optimizations(self, response: str, context: Dict) -> str: | |
| """✅ Applies context-specific response optimization.""" | |
| if "filter_profanity" in context and context["filter_profanity"]: | |
| response = self.remove_profanity(response) | |
| if "trim_length" in context: | |
| response = response[:context["trim_length"]].strip() + "..." # Trims to desired length | |
| return response | |
| def remove_profanity(self, response: str) -> str: | |
| """🚫 Removes flagged words from AI-generated responses.""" | |
| banned_words = [ | |
| "badword1", "badword2", "badword3", "shit", "fuck", "damn", "bitch", "asshole" | |
| ] # ✅ Add or remove based on testing | |
| for word in banned_words: | |
| response = response.replace(word, "***") # ✅ Replace with censorship symbol | |
| return response | |