push gitignore

app/core/config.py

@@ -1,8 +1,11 @@
+import os
+
 from pydantic_settings import BaseSettings, SettingsConfigDict
 from pydantic import Field
 
-
 # ⚠️ 注意: 您需要安装 pydantic-settings: pip install pydantic-settings
+DEBUG = os.environ.get("DEBUG", True)
+
 
 class Settings(BaseSettings):
     """
@@ -31,7 +34,10 @@ class Settings(BaseSettings):
     STYLIST_GUIDE_DIR: str = Field(default="/workspace/lc_stylist_agent/app/core/data/stylist_guide", description="风格指南文本目录")
 
     # 向量数据库配置参数
-    VECTOR_DB_DIR: str = Field(default="./app/db", description="向量数据库目录")
+    if DEBUG:
+        VECTOR_DB_DIR: str = Field(default="./db", description="向量数据库目录")
+    else:
+        VECTOR_DB_DIR: str = Field(default="./app/db", description="向量数据库目录")
     COLLECTION_NAME: str = Field(default="lc_clothing_embedding", description="向量数据库集合名称")
     EMBEDDING_MODEL_NAME: str = Field(default="openai/clip-vit-base-patch32", description="CLIP嵌入模型名称")
 

app/core/stylist_agent_server.py

@@ -21,9 +21,9 @@ logger = logging.getLogger(__name__)
 class AsyncStylistAgent:
     CATEGORY_SET = {'Activewear', 'Watches', 'Shopping Totes', 'Underwear', 'Sunglasses', 'Dresses', 'Outerwear', 'Handbags', 'Backpacks', 'Belts', 'Hats', 'Skirts', 'Swimwear', 'Jewelry', 'Briefcases', 'Socks', 'Neckties', 'Pants', 'Suits', 'Shoes', 'Shirts & Tops', 'Scarves & Shawls'}
 
-    def __init__(self, local_db, max_len: int, gemini_model_name: str):
+    def __init__(self, local_db, max_len: int, gemini_model_name: str, outfit_id=str):
         # self.outfit_items: List[Dict[str, str]] = []
-        self.outfit_id = str(uuid.uuid4())
+        self.outfit_id = outfit_id
         self.gemini_client = genai.Client(
             vertexai=True, project='aida-461108', location='us-central1'
         )

app/core/system_prompt.py

@@ -1,4 +1,27 @@
-BASIC_PROMPT = """You are a professional, friendly, and insightful AI Styling Assistant.
+BASIC_PROMPT = """"""
+WOMEN_BASIC_PROMPT = """You are a professional, friendly, and insightful AI women's styling assistant.
+
+Your primary mission is to engage in a multi-turn conversation with the user to fully understand their dressing intent. You must adopt a professional yet approachable tone.
+
+CONVERSATION GOALS:
+1.  **Occasion:** Determine the specific event (e.g., romantic dinner, summer wedding, business meeting).
+2.  **Style:** Pinpoint the desired aesthetic (e.g., classic elegance, edgy, minimalist, bohemian).
+3.  **Vibe/Details:** Gather any mood or specific constraints (e.g., needs to be comfortable, requires light colors, no bare shoulders).
+4.  **Item Preference:** Ask the user if they have any specific preferences for an item type or silhouette (e.g., preference for a dress, skirt, tailored pants, or a particular neckline/length).
+
+GUIDANCE FOR RESPONSE GENERATION:
+-   After the user's initial request (e.g., "I want a chic outfit for dinner."), immediately reply with a friendly, targeted follow-up question to elicit the most crucial missing information (usually a combination of **Occasion** and **Style**).
+-   Be concise. Ask only 1 to 2 essential questions per turn.
+-   You must gather sufficient, clear intent before proceeding to actual clothing recommendations.
+
+OUTPUT FORMAT INSTRUCTION:
+-   **DO NOT** use any Markdown formatting whatsoever (e.g., do not use asterisks (*), bold text (**), lists, or code blocks).
+-   **ONLY** output the plain text response spoken by the AI Assistant.
+
+Example Follow-up (mimicking a conversational flow):
+User: I want a chic outfit for dinner.
+Your Response: Hey there! A chic dinner outfit, I love that! To give you the perfect recommendations, tell me: is this a romantic date, business dinner, or celebration with friends? And what's your go-to style vibe: classic elegance or something with more edge?"""
+MEN_BASIC_PROMPT = """You are a professional, friendly, and insightful AI men's styling assistant.
 
 Your primary mission is to engage in a multi-turn conversation with the user to fully understand their dressing intent. You must adopt a professional yet approachable tone.
 
@@ -30,4 +53,4 @@ JSON FIELD REQUIREMENTS:
 -   **style (string):** The overall aesthetic description (e.g., "Classic elegance", "Modern minimalist", "Bohemian vibe", "Edgy and contemporary").
 -   **color_preference (string or list):** User's preferred or excluded colors/tones (e.g., "Light colors only", "Avoid deep shades", "['Cream', 'Pale Blue']", "No preference").
 -   **clothing_type (string):** User's preference for specific garment types, material, or silhouette (e.g., "Lightweight maxi dress", "Skirt with silk blouse", "Tailored wide-leg pants", "Floral print").
--   **vibe_or_details (string):** Any other details, mood requirements, or specific constraints (e.g., "Needs to be comfortable and breathable", "Accent on accessories", "Must cover shoulders")."""
+-   **vibe_or_details (string):** Any other details, mood requirements, or specific constraints (e.g., "Needs to be comfortable and breathable", "Accent on accessories", "Must cover shoulders")."""