Add wardrobe_description support for better AI understanding

models.py

@@ -13,6 +13,7 @@ class ChatRequest(BaseModel):
     message: str
     session_id: Optional[str] = "default"
     wardrobe: Optional[List[WardrobeItem]] = None
+    wardrobe_description: Optional[str] = None
     images: Optional[List[str]] = None
 
 class ChatResponse(BaseModel):

routes.py

@@ -219,7 +219,9 @@ def setup_routes(app):
             
             if request.wardrobe and len(request.wardrobe) > 0:
                 print(f"[WARDROBE CHAT] ===== WARDROBE REQUEST DETECTED =====")
-                return await handle_wardrobe_chat(message, request.wardrobe, session_id, images=request.images)
+                if request.wardrobe_description:
+                    print(f"[WARDROBE CHAT] Using provided wardrobe description ({len(request.wardrobe_description)} chars)")
+                return await handle_wardrobe_chat(message, request.wardrobe, session_id, images=request.images, wardrobe_description=request.wardrobe_description)
             
             conv_context = get_conversation_context(session_id)
             
@@ -385,6 +387,7 @@ def setup_routes(app):
         message: str = Form(...),
         session_id: str = Form(default="default"),
         wardrobe: Optional[str] = Form(default=None),
+        wardrobe_description: Optional[str] = Form(default=None),
         images: List[UploadFile] = File(default=[])
     ):
         try:
@@ -411,6 +414,7 @@ def setup_routes(app):
                 message=message,
                 session_id=session_id,
                 wardrobe=wardrobe_items if wardrobe_items else None,
+                wardrobe_description=wardrobe_description if wardrobe_description and wardrobe_description.strip() else None,
                 images=image_data_urls if image_data_urls else None
             )
             
@@ -428,6 +432,7 @@ def setup_routes(app):
         message: str = Form(...),
         session_id: str = Form(default="default"),
         wardrobe: Optional[str] = Form(default=None),
+        wardrobe_description: Optional[str] = Form(default=None),
         images: List[UploadFile] = File(default=[])
     ):
         image_data_urls = []
@@ -440,9 +445,36 @@ def setup_routes(app):
                 image_data_urls.append(data_url)
                 print(f"[STREAM UPLOAD] Processed image: {img_file.filename} ({len(content)} bytes)")
         
+        wardrobe_items = []
+        if wardrobe and wardrobe.strip() and wardrobe.strip() not in ["[]", "", "string"]:
+            try:
+                wardrobe_data = json.loads(wardrobe)
+                if isinstance(wardrobe_data, list):
+                    wardrobe_items = [WardrobeItem(**item) for item in wardrobe_data]
+            except json.JSONDecodeError:
+                print(f"[STREAM UPLOAD] Ignoring invalid wardrobe value: {wardrobe[:50]}")
+        
         rag_chunks = retrieve_relevant_context(message, top_k=3)
         rag_context = format_rag_context(rag_chunks)
         
+        wardrobe_context = ""
+        if wardrobe_description and wardrobe_description.strip():
+            wardrobe_context = wardrobe_description
+            print(f"[STREAM UPLOAD] Using provided wardrobe description ({len(wardrobe_context)} chars)")
+        elif wardrobe_items:
+            from wardrobe import format_wardrobe_for_prompt
+            wardrobe_context = format_wardrobe_for_prompt(wardrobe_items)
+            print(f"[STREAM UPLOAD] Generated wardrobe context ({len(wardrobe_context)} chars)")
+        
+        if wardrobe_context:
+            prompt = f"""{wardrobe_context}
+
+User request: {message}
+
+Suggest a complete outfit using ONLY the items listed above. Reference items by their exact names. Include accessories if available. Be friendly and conversational."""
+        else:
+            prompt = message
+        
         print(f"[STREAM UPLOAD] Starting streaming response for: {message[:50]}...")
         
         async def generate():
@@ -450,7 +482,7 @@ def setup_routes(app):
             
             full_response = ""
             async for chunk in generate_chat_response_streaming(
-                prompt=message,
+                prompt=prompt,
                 max_length=512,
                 temperature=0.7,
                 rag_context=rag_context,
@@ -461,7 +493,6 @@ def setup_routes(app):
             
             yield f"data: {json.dumps({'type': 'end', 'full_response': full_response, 'session_id': session_id})}\n\n"
             print(f"[STREAM UPLOAD] Streaming complete: {len(full_response)} chars")
-            print(f"[STREAM RESPONSE] {full_response}")
         
         return StreamingResponse(
             generate(),

test_api.sh

@@ -0,0 +1,85 @@
+#!/bin/bash
+
+if [ -z "$1" ]; then
+    BASE_URL="https://nexusbert-style.hf.space"
+else
+    BASE_URL="$1"
+fi
+
+echo "=== Testing Style GPT API ==="
+echo "Base URL: $BASE_URL"
+echo "Usage: $0 [base_url]"
+echo "Example: $0 http://localhost:7860"
+echo ""
+
+echo "1. Testing GET / (Root endpoint)"
+curl -X GET "$BASE_URL/" \
+  -H "Content-Type: application/json" \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "2. Testing GET /health"
+curl -X GET "$BASE_URL/health" \
+  -H "Content-Type: application/json" \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "3. Testing POST /text (Text-only chat)"
+curl -X POST "$BASE_URL/text" \
+  -H "Content-Type: application/json" \
+  -d '{
+    "message": "Hello, what colors go well with blue?",
+    "session_id": "test-session-1"
+  }' \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "4. Testing POST /chat (Chat with optional images - text only)"
+curl -X POST "$BASE_URL/chat" \
+  -H "Content-Type: application/json" \
+  -d '{
+    "message": "What should I wear with a black jacket?",
+    "session_id": "test-session-2",
+    "images": null
+  }' \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "5. Testing POST /chat (Chat with wardrobe)"
+curl -X POST "$BASE_URL/chat" \
+  -H "Content-Type: application/json" \
+  -d '{
+    "message": "Suggest an outfit for a casual meeting",
+    "session_id": "test-session-3",
+    "wardrobe": [
+      {
+        "category": "shirt",
+        "style": "casual",
+        "color": "white",
+        "brand": "Zara",
+        "name": "White casual shirt"
+      },
+      {
+        "category": "pants",
+        "style": "formal",
+        "color": "navy",
+        "brand": "H&M",
+        "name": "Navy trousers"
+      }
+    ]
+  }' \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "6. Testing POST /chat/upload (File upload - text only)"
+curl -X POST "$BASE_URL/chat/upload" \
+  -F "message=What colors match with red?" \
+  -F "session_id=test-session-4" \
+  -F "wardrobe=[]" \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "7. Testing POST /chat/upload/stream (Streaming - text only)"
+curl -X POST "$BASE_URL/chat/upload/stream" \
+  -F "message=Tell me about fashion trends" \
+  -F "session_id=test-session-5" \
+  -F "wardrobe=[]" \
+  --no-buffer \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "=== All tests completed ==="
+

test_local.sh

@@ -0,0 +1,80 @@
+#!/bin/bash
+
+BASE_URL="http://localhost:7860"
+
+echo "=== Testing Style GPT API (Local) ==="
+echo "Base URL: $BASE_URL"
+echo "Make sure the server is running: uvicorn app:app --host 0.0.0.0 --port 7860"
+echo ""
+
+echo "1. Testing GET / (Root endpoint)"
+curl -X GET "$BASE_URL/" \
+  -H "Content-Type: application/json" \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "2. Testing GET /health"
+curl -X GET "$BASE_URL/health" \
+  -H "Content-Type: application/json" \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "3. Testing POST /text (Text-only chat)"
+curl -X POST "$BASE_URL/text" \
+  -H "Content-Type: application/json" \
+  -d '{
+    "message": "Hello, what colors go well with blue?",
+    "session_id": "test-session-1"
+  }' \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "4. Testing POST /chat (Chat with optional images - text only)"
+curl -X POST "$BASE_URL/chat" \
+  -H "Content-Type: application/json" \
+  -d '{
+    "message": "What should I wear with a black jacket?",
+    "session_id": "test-session-2",
+    "images": null
+  }' \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "5. Testing POST /chat (Chat with wardrobe)"
+curl -X POST "$BASE_URL/chat" \
+  -H "Content-Type: application/json" \
+  -d '{
+    "message": "Suggest an outfit for a casual meeting",
+    "session_id": "test-session-3",
+    "wardrobe": [
+      {
+        "category": "shirt",
+        "style": "casual",
+        "color": "white",
+        "brand": "Zara",
+        "name": "White casual shirt"
+      },
+      {
+        "category": "pants",
+        "style": "formal",
+        "color": "navy",
+        "brand": "H&M",
+        "name": "Navy trousers"
+      }
+    ]
+  }' \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "6. Testing POST /chat/upload (File upload - text only)"
+curl -X POST "$BASE_URL/chat/upload" \
+  -F "message=What colors match with red?" \
+  -F "session_id=test-session-4" \
+  -F "wardrobe=[]" \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "7. Testing POST /chat/upload/stream (Streaming - text only)"
+curl -X POST "$BASE_URL/chat/upload/stream" \
+  -F "message=Tell me about fashion trends" \
+  -F "session_id=test-session-5" \
+  -F "wardrobe=[]" \
+  --no-buffer \
+  -w "\nHTTP Status: %{http_code}\n\n"
+
+echo "=== All tests completed ==="
+

wardrobe.py

@@ -62,11 +62,11 @@ def format_wardrobe_for_prompt(wardrobe: List[WardrobeItem]) -> str:
 
 Categories: {categories_list}"""
 
-async def handle_wardrobe_chat(message: str, wardrobe: List[WardrobeItem], session_id: str, images: Optional[List[str]] = None) -> ChatResponse:
+async def handle_wardrobe_chat(message: str, wardrobe: List[WardrobeItem], session_id: str, images: Optional[List[str]] = None, wardrobe_description: Optional[str] = None) -> ChatResponse:
     conv_context = get_conversation_context(session_id)
     enhanced_message = enhance_message_with_context(message, conv_context["context"])
     
-    wardrobe_context = format_wardrobe_for_prompt(wardrobe)
+    wardrobe_context = wardrobe_description if wardrobe_description else format_wardrobe_for_prompt(wardrobe)
     
     wardrobe_by_category = {}
     for item in wardrobe: