app.py Beta 2 (Push 8)

app.py

@@ -63,9 +63,10 @@ def extract_keywords(text: str) -> List[str]:
 
 def detect_intent(text: str) -> str:
     doc = nlp(text.lower())
-    if any(token.text in ["shirt", "shirts"] for token in doc):
+    # Stricter matching for "shirt" vs "short"
+    if "shirt" in [token.text for token in doc]:  # Exact match for "shirt"
         return "recommend_shirt"
-    elif any(token.text in ["short", "shorts"] for token in doc):
+    elif "short" in [token.text for token in doc]:  # Exact match for "short"
         return "recommend_shorts"
     elif any(token.text in ["what", "who", "company", "do", "products"] for token in doc):
         return "company_info"
@@ -75,8 +76,8 @@ def search_products_by_keywords(keywords: List[str]) -> List[Dict[str, Any]]:
     if not keywords:
         logger.info("No keywords provided, returning empty product list.")
         return []
-    query = {"$or": [{"name": {"$regex": keyword, "$options": "i"}} for keyword in keywords]}
-    # Convert ObjectId to string and sanitize document
+    # Use stricter matching: only return products with exact keyword in name
+    query = {"$or": [{"name": {"$regex": f"\\b{keyword}\\b", "$options": "i"}} for keyword in keywords]}
     matched_products = [
         {
             "id": str(p["_id"]),
@@ -92,13 +93,13 @@ def get_product_context(products: List[Dict]) -> str:
     if not products:
         return ""
     product_str = "Here are some products: "
-    product_str += ", ".join([f"'{p['name']}' (SKU: {p['skuNumber']}) - {p['description']}" for p in products[:2]])
+    product_str += ", ".join([f"'{p['name']}' - {p['description']}" for p in products[:2]])
     return product_str
 
 def format_response(response: str, products: List[Dict], intent: str) -> str:
     if intent in ["recommend_shirt", "recommend_shorts"] and products:
         product = products[0]
-        return f"{response} For example, check out our '{product['name']}' (SKU: {product['skuNumber']})—it’s {product['description'].lower()}!"
+        return f"{response} For example, check out our '{product['name']}'—it’s {product['description'].lower()}!"
     elif intent == "company_info":
         return f"{response} At Hutter Products GmbH, we specialize in sustainable product design and production!"
     return response