Updated with Content msg, product recommendation and history information5

app.py

@@ -8,7 +8,7 @@ import os
 import logging
 import re
 import torch
-import random  # For response variety
+import random
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -74,11 +74,11 @@ def detect_intent(text: str) -> str:
         return "recommend_product"
     elif any(token.text in ["company", "who", "do"] for token in doc):
         return "company_info"
-    elif "name" in text_lower or "yourself" in text_lower or "you" in doc and "about" in doc:
+    elif "name" in text_lower or "yourself" in text_lower or ("you" in doc and "about" in doc):
         return "ask_name"
     elif re.search(r"\d+\s*[\+\-\*/]\s*\d+", text_lower):
         return "math_query"
-    return "chat"  # New fallback for general conversation
+    return "chat"
 
 def search_products_by_keywords(keywords: List[str]) -> List[Dict[str, Any]]:
     if not keywords:
@@ -102,39 +102,44 @@ def get_product_context(products: List[Dict]) -> str:
     return product_str
 
 def format_response(response: str, products: List[Dict], intent: str, input_text: str, history: List[str]) -> str:
-    no_product_prompts = [
-        "I’d love to recommend something! What are you looking for in our sustainable catalog?",
-        "Our sustainable catalog has lots to offer—what catches your interest?",
-        "Tell me what you’re after, and I’ll find something great from our eco-friendly range!"
-    ]
-    
+    # Base response is BlenderBot’s output; adjust based on intent
+    base_response = response if response else "I’m here to help—what’s on your mind?"
+
     if intent == "recommend_product":
-        if not products:
-            return random.choice(no_product_prompts)
-        product = products[0]
-        return f"Check out our '{product['name']}'—it’s {product['description'].lower()}. Want more options?"
+        if products:
+            product = products[0]
+            return f"{base_response} Speaking of sustainable products, check out our '{product['name']}'—it’s {product['description'].lower()}."
+        prompts = [
+            f"{base_response} What sustainable items are you looking for today?",
+            f"{base_response} Any specific eco-friendly products you’re curious about?",
+        ]
+        return random.choice(prompts)
+    
     elif intent == "company_info":
-        return "Hutter Products GmbH offers sustainable products like recycled textiles and ocean plastic goods."
+        return f"{base_response} I’m with Hutter Products GmbH—we focus on sustainable items like recycled textiles and ocean plastic goods."
+    
     elif intent == "ask_name":
-        return "I’m Hutter, your shopping guide for Hutter Products GmbH. I’m here to help you find eco-friendly products—how can I assist?"
+        return f"{base_response} I’m Hutter, your shopping guide for Hutter Products GmbH, here to assist with sustainable products."
+    
     elif intent == "math_query":
         match = re.search(r"(\d+)\s*([\+\-\*/])\s*(\d+)", input_text.lower())
         if match:
             num1, op, num2 = int(match.group(1)), match.group(2), int(match.group(3))
-            if op == "+": return f"{num1} + {num2} = {num1 + num2}. Need shopping help?"
-            elif op == "-": return f"{num1} - {num2} = {num1 - num2}. Anything else?"
-            elif op == "*": return f"{num1} * {num2} = {num1 * num2}. Explore our products?"
-            elif op == "/": return f"{num1} / {num2} = {num1 / num2}." if num2 != 0 else "Can’t divide by zero! Try our products?"
-        return "I can do math—try '2 + 2'. What else can I help with?"
+            if op == "+": return f"{base_response} By the way, {num1} + {num2} = {num1 + num2}."
+            elif op == "-": return f"{base_response} Also, {num1} - {num2} = {num1 - num2}."
+            elif op == "*": return f"{base_response} Oh, and {num1} * {num2} = {num1 * num2}."
+            elif op == "/": return f"{base_response} Plus, {num1} / {num2} = {num1 / num2}." if num2 != 0 else f"{base_response} Can’t divide by zero, though!"
+        return f"{base_response} I can help with math—try something like '2 + 2'."
+    
     elif intent == "chat":
-        # Use BlenderBot’s response if appropriate, else nudge toward shopping
-        if "yes" in input_text.lower() and history and "hat" in history[-1].lower():
-            return "Great! Besides the Bucket Hat, we have other sustainable items—want to hear more?"
-        return f"{response} How can I assist with our sustainable products today?" if response else "I’m here to help—anything on your mind?"
-    if products:
-        product = products[0]
-        return f"{response} Also, check out '{product['name']}'—it’s {product['description'].lower()}."
-    return response if response else "How can I assist with our sustainable products?"
+        if "yes" in input_text.lower() and history and any(word in history[-1].lower() for word in ["hat", "product", "store"]):
+            if products:
+                product = products[0]
+                return f"{base_response} Great! How about our '{product['name']}'? It’s {product['description'].lower()}."
+            return f"{base_response} Want me to suggest some sustainable items?"
+        return base_response  # Let BlenderBot shine for casual chat
+    
+    return base_response  # Fallback
 
 # Endpoints
 @app.get("/")
@@ -170,8 +175,8 @@ async def process_prompt(request: PromptRequest):
                 **inputs,
                 max_new_tokens=30,
                 do_sample=True,
-                top_p=0.95,  # Slightly higher for more variety
-                temperature=0.8,  # Slightly higher for creativity
+                top_p=0.95,
+                temperature=0.8,
                 no_repeat_ngram_size=2
             )
         logger.info("Model generation complete.")