app.py Beta 2 (Push 7)

app.py

@@ -76,22 +76,29 @@ def search_products_by_keywords(keywords: List[str]) -> List[Dict[str, Any]]:
         logger.info("No keywords provided, returning empty product list.")
         return []
     query = {"$or": [{"name": {"$regex": keyword, "$options": "i"}} for keyword in keywords]}
-    matched_products = [dict(p, id=str(p["_id"])) for p in products_collection.find(query)]
+    # Convert ObjectId to string and sanitize document
+    matched_products = [
+        {
+            "id": str(p["_id"]),
+            "name": p.get("name", "Unknown"),
+            "skuNumber": p.get("skuNumber", "N/A"),
+            "description": p.get("description", "No description available")
+        }
+        for p in products_collection.find(query)
+    ]
     return matched_products
 
 def get_product_context(products: List[Dict]) -> str:
     if not products:
         return ""
     product_str = "Here are some products: "
-    # Use .get() to handle missing 'skuNumber' with a fallback
-    product_str += ", ".join([f"'{p['name']}' (SKU: {p.get('skuNumber', 'N/A')}) - {p['description']}" for p in products[:2]])
+    product_str += ", ".join([f"'{p['name']}' (SKU: {p['skuNumber']}) - {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]
-        sku = product.get('skuNumber', 'N/A')  # Handle missing skuNumber
-        return f"{response} For example, check out our '{product['name']}' (SKU: {sku})—it’s {product['description'].lower()}!"
+        return f"{response} For example, check out our '{product['name']}' (SKU: {product['skuNumber']})—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