o @Tj@stdZddlmZmZddlmZGdddeZGdddeZGdd d eZGd d d eZ Gd d d eZ dS)z&Abstract base class for LLM providers.)ABCabstractmethod) BaseModelc@sJeZdZUdZeed<dZedBed<dZedBed<dZe dBed<dS)GarmentAnalysisz(Single garment analysis result from LLM. category_mainN category_sub descriptiontags) __name__ __module__ __qualname____doc__str__annotations__rrr dictrr5/home/ubuntu/htdocs/ootd-api/app/services/llm/base.pyrs rc@s"eZdZUdZeed<eed<dS)StyleCategoryScorez.Score and comment for a single style category.scorecommentN)r r r r intrrrrrrrs  rc@sFeZdZUdZeed<eed<eed<eed<eed<eeed<dS) StyleAnalysisz.Overall outfit style analysis result from LLM. color_palette silhouettedetail overall_scoreoverall_comment style_tagsN) r r r r rrfloatrlistrrrrrs rc@s:eZdZUdZeed<eed<eed<eed<eed<dS)StyleAnalysisTempz4Context-aware outfit style analysis result from LLM. time_contextoccasion_contextweather_context color_harmonyimprovement_tipsN)r r r r rrrrrrr #s  r c@seZdZdZededeefddZe ddedede d e dBdedBf d d Z edede fd d Z edede fddZe ddedeedee d e dBdeedBe dBff ddZdS)BaseLLMProviderz0Interface that all LLM providers must implement.imagereturncdS)zAnalyze image and return detected garments with categories, tags, description. Args: image: Raw image bytes (JPEG/PNG). Returns: List of GarmentAnalysis for each detected garment. Nrselfr'rrranalyze_garment0 zBaseLLMProvider.analyze_garmentNitemmode image_modelcr))uGenerate ghost mannequin image for a garment. Args: image: Original image bytes. item: Analysis result for the target garment. mode: "edit" (remove person) or "generate" (new image from description). image_model: Optional image-generation model override. ``None`` → provider default. The Gemini provider ignores this. Returns: Generated image bytes, or None on failure. Nr)r+r'r.r/r0rrrgenerate_ghost_mannequin<sz(BaseLLMProvider.generate_ghost_mannequincr))zAnalyze overall outfit style from image. Args: image: Raw image bytes (JPEG/PNG). Returns: StyleAnalysis with scores, comments, and style tags. Nrr*rrr analyze_styleRr-zBaseLLMProvider.analyze_stylecr))zContext-aware outfit style analysis (time/occasion/weather). Args: image: Raw image bytes (JPEG/PNG). Returns: StyleAnalysisTemp with 5 contextual sections. Nrr*rrranalyze_style_temp^r-z"BaseLLMProvider.analyze_style_temp person_image ghost_images categoriescr))u0Composite ghost garments onto a person via multi-image generation. Args: person_image: Raw person photo bytes (JPEG). ghost_images: List of isolated garment (ghost mannequin) PNG bytes, ordered 1..N — the prompt references them as "#N". categories: Parallel list of ``category_main`` strings (``"top" | "bottom" | "outer" | "dress"``) used by the prompt to let the LLM decide which garments are bottoms (to be skipped when only the upper body is visible). image_model: Optional image-generation model override. ``None`` → provider default. The Gemini provider ignores this. Returns: Tuple ``(image_bytes, llm_note)``. - ``image_bytes`` — composited PNG bytes, or ``None`` on failure. - ``llm_note`` — a short English sentence the model returned alongside the image (e.g. ``"Skipped #2 because ..."``), or ``None`` if the model returned no text. Nr)r+r4r5r6r0rrr generate_vtonjszBaseLLMProvider.generate_vton)N)r r r r rbytesrrr,rr1rr2r r3tupler7rrrrr&-sD    r&N) r abcrrpydanticrrrrr r&rrrrs