app.py Beta 2 (push 6)

app.py

@@ -57,7 +57,7 @@ class CombinedResponse(BaseModel):
 
 # Helper Functions
 def extract_keywords(text: str) -> List[str]:
-    doc = nlp(text)  # Fixed syntax error here
+    doc = nlp(text)
     keywords = [token.text for token in doc if token.pos_ in ["NOUN", "PROPN"]]
     return list(set(keywords))
 
@@ -83,13 +83,15 @@ 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]])
+    # 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]])
     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()}!"
+        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()}!"
     elif intent == "company_info":
         return f"{response} At Hutter Products GmbH, we specialize in sustainable product design and production!"
     return response