o @Tj1@sdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z mZmZmZmZeeZdZdZdZdZddhZd ed ed efd d ZGdddeZdS)uHOpenAI LLM provider — GPT-5.4 Vision + Responses API image_generation.N)settings)BaseLLMProviderGarmentAnalysis StyleAnalysisStyleAnalysisTemp)GARMENT_ANALYSIS_PROMPTGHOST_MANNEQUIN_EDIT_PROMPTSTYLE_ANALYSIS_PROMPTSTYLE_ANALYSIS_TEMP VTON_PROMPTzhttps://api.openai.com/v1z gpt-5.4-minizgpt-5.4model temperaturereturncCs|tvrd|iSiS)uSampling params for ``model`` — include ``temperature`` only if supported. Keeps a low, deterministic ``temperature`` for models that honor it while transparently dropping the param for models that don't, instead of hard-failing the request. r )TEMPERATURE_CAPABLE_MODELS)r r r7/home/ubuntu/htdocs/ootd-api/app/services/llm/openai.py_sampling_params1src@seZdZdZdddZdeeeffddZdeeeffdd Zd edefd d Z d e de e fddZ dd e de dededBde dBf ddZ dde de e de ededBdee dBedBff ddZd e defddZd e defddZdS) OpenAIProviderz3OpenAI API provider using httpx for async requests.rNcCs tj|_|jstddSdS)Nu4OPENAI_API_KEY is not set — OpenAI calls will fail)rOPENAI_API_KEYapi_keyloggerwarningselfrrr__init__@szOpenAIProvider.__init__cCsdd|jiS)N AuthorizationBearer rrrrr _auth_headerEszOpenAIProvider._auth_headercCsd|jddS)Nrzapplication/json)rz Content-Typerrrrr _json_headersHs zOpenAIProvider._json_headerspayloadc std}tjdd4IdH0}|j|||dIdH}|js2td|d|j |j | | WdIdHS1IdHsFwYdS)abPOST to /chat/completions, surfacing OpenAI's error body on failure. ``raise_for_status()`` alone discards the response body, but that body carries the actionable detail (unknown model, unsupported parameter value, etc.). Logging it mirrors what ``generate_ghost_mannequin`` already does for the images endpoint. z/chat/completionsgN@timeoutN)jsonheadersu9OpenAI chat/completions failed (model=%s): HTTP %d — %sr ) OPENAI_BASE_URLhttpx AsyncClientpostr is_successrerrorget status_codetextraise_for_statusr#)rr urlclientresprrr_chat_completionNs 0zOpenAIProvider._chat_completionimagecst|}tddtddd|dddgd gd ttd d d did}||IdH}|dddd}t|}t |t rW| D] }t |t rV|}nqKt |t sft dt|gSdd|DS)z/Analyze garments in image using GPT-5.4 Vision.userr-typer- image_urldata:image/jpeg;base64,highr/detailr6r7rolecontentr messagesg?r6 json_objectmax_completion_tokensresponse_formatNchoicesrmessager?z"Unexpected GPT response format: %scSsg|] }tdi|qS)r)r).0itemrrr sz2OpenAIProvider.analyze_garment..)base64 b64encodedecodeOPENAI_VISION_MODELrrr2r#loads isinstancedictvalueslistrr*r6)rr3 b64_imager datar-parsedvrrranalyze_garmentcs@     zOpenAIProvider.analyze_garmentrJmode image_modelc s4|ptj}tj|j|jp|jd}dd|dffg}||ddddd }|d vr+d |d <td } tjdd4IdH/} | j | ||| dIdH} | j s[| j } t d| j| | | } WdIdHn 1IdHsowY| dpzg}|st ddS|ddd}|st ddSt|S)a Generate ghost mannequin via /v1/images/edits. Skips the Responses-API agent reasoning overhead by calling the direct edits endpoint; ``input_fidelity=high`` preserves logos, stitching, and color patterns more faithfully than the previous setup. The ``mode`` parameter is accepted for signature compatibility with Gemini but is currently ignored (single "edit" path). ``image_model`` selects the image-generation model; ``None`` falls back to ``OPENAI_IMAGE_MODEL``. ) category_main category_subimage[]z source.jpg image/jpeg1 1024x1024lowpngr promptnsizequality output_format)z gpt-image-1z gpt-image-1.5r9input_fidelity /images/editsgf@r!NrVfilesr$u-OpenAI ghost mannequin failed: HTTP %d — %srVz%OpenAI images/edits returned no imagerb64_jsonz+OpenAI images/edits returned empty b64_json)rOPENAI_IMAGE_MODELrformatr\r]r%r&r'r(rr)r-rr*r,r.r#r+rrL b64decode)rr3rJrZr[r rermrVr/r0r1 error_bodybodyitemsb64rrrgenerate_ghost_mannequinsV      (   z'OpenAIProvider.generate_ghost_mannequin person_image ghost_images categoriesc st|ptj}dddt|D}tjt||d}dd|dffg}t|D]\} } |dd| d | d ffq'||d d d dd} td} t j dd4IdH/} | j | | || dIdH}|j ss|j}td|j|||}WdIdHn 1IdHswY|dpg}|stddS|ddd}|rt|nd}|durtd|dfS)uComposite ghost garments onto a person via /v1/images/edits. Sends the person photo and each ghost image as multipart ``image[]`` parts. The direct edits endpoint skips the Responses-API agent reasoning, so it is notably faster but does not return any free-text remark — the second tuple slot is always None. ``image_model`` selects the image-generation model; ``None`` falls back to ``OPENAI_IMAGE_MODEL``. z, css(|]\}}d|dd|VqdS)#=Nr)rIicatrrr s&z/OpenAIProvider.generate_vton..) garment_countcatalogr^z person.jpgr_garment_z.pngz image/pngr` 1024x1536rbrcrdrkgn@r!Nrlu"OpenAI VTON failed: HTTP %d — %srVz'OpenAI VTON response contained no image)NNrrnro)rrpjoin enumerater rqlenappendr%r&r'r(rr)r-rr*r,r.r#r+rrLrr)rrxryrzr[r rrermr~grVr/r0r1rsrtrurv image_bytesrrr generate_vtonsV     (  zOpenAIProvider.generate_vtonct|}tddtddd|dddgd gd ttd d d did}||IdH}|dddd}t|}t di|S)z2Analyze overall outfit style using GPT-5.4 Vision.r4r-r5r7r8r9r:r<r=r@333333?rBr6rCrDNrGrrHr?r) rLrMrNrOr rr2r#rPrrr3rUr rVr-rWrrr analyze_style. zOpenAIProvider.analyze_stylecr)z7Context-aware outfit style analysis using GPT-5.4 Mini.r4r-r5r7r8r9r:r<r=r@rrBr6rCrDNrGrrHr?r) rLrMrNOPENAI_MINI_MODELr rr2r#rPrrrrranalyze_style_temp7rz!OpenAIProvider.analyze_style_temp)rN)N)__name__ __module__ __qualname____doc__rrRstrrrr2bytesrTrrYrwtuplerrrrrrrrrr=sB 1 L @!r)rrLr#loggingr& app.configrapp.services.llm.baserrrrapp.services.llm.promptsrrr r r getLoggerrrr%rOOPENAI_VISION_MODEL_HIGHrrrfloatrRrrrrrrs