5i- ddlmZmZddlmZddlmZddlZddlZddl Z ddl Z ddl m Z e edZ e jdZes ed d#d ed ed edefdZd$dedededefdZd%defdZe ddZe dd&d ed ed efdZe dd&d ed ed efdZedkrddlZeje d d!"dSdS)')FastAPI HTTPException)StreamingResponse)ImageN) load_dotenvAI Renovation Inference Service)titleSTABILITY_API_KEYuG⚠️ Warning: STABILITY_API_KEY not set. Will use placeholder images. living_roomstyle layout_id room_typereturncdddddd}|||d}idd |d d d |d d d |ddd |ddd|ddd|ddd|ddd|ddd|ddd|ddd|ddd|dd d!|d"d#d!|d"d$d%|d&d'd%|d&d(d%|d&}|||d}|d)|d*}d+}||fS),uJ根据风格和房间类型生成 Stability AI 提示词和负面提示词z+living room, sofa, coffee table, TV cabinetz"bedroom, bed, nightstand, wardrobez$kitchen, cabinets, countertops, sinkz!bathroom, bathtub, shower, vanityz!dining room, dining table, chairs)r bedroomkitchenbathroom dining_roomr 现代简约zultra modern minimalist z, floor-to-ceiling windows, pure white walls, no decorations, sleek glass coffee table, hidden LED strip lighting, open concept, IKEA-style furniture, concrete floor, high ceiling, 8K realistic photozmodern minimalistu 北欧风zauthentic Scandinavian z, light birch wood floor, sheepskin rug, hygge atmosphere, pastel color palette (mint, blush, sky blue), pendant lamp with fabric shade, minimalist bookshelf, Monstera plant, white walls with wood accents, cozy textiles, 8K realistic photo Scandinavianu 中式风格ztraditional Chinese a, dark rosewood furniture with intricate carvings, red lacquer cabinets, golden dragon motifs, silk embroidered cushions, paper lanterns hanging from ceiling, moon gate entrance, bonsai tree, calligraphy wall art, terra cotta floor tiles, 8K realistic photozChinese traditionalu 美式风格zclassic American z, warm beige walls, overstuffed sectional sofa, brick fireplace with wooden mantel, oak hardwood floor, vintage rug, family photos on wall, ceiling fan with lights, bay window with curtains, 8K realistic photozAmerican classicu 工业风zauthentic industrial loft z, exposed red brick walls, polished concrete floor, black metal frame furniture, Edison bulb pendant lights, steel beams on ceiling, leather sofa with metal legs, vintage factory cart coffee table, pipe shelving, 8K realistic photo industrialu 日式风格z zen Japanese z, tatami mat floor, low wooden table, shoji sliding paper doors, bonsai tree, neutral earth tones (beige, brown, gray), minimalist futon, bamboo accents, rock garden view through window, 8K realistic photoz Japanese zenu 轻奢风zluxury modern z, Italian marble floor, gold-plated furniture legs, crystal chandelier, velvet sofa in emerald green, mirrored accent wall, abstract art painting, smart home automation visible, floor-to-ceiling curtains, 8K realistic photoz luxury modernu地中海风格zauthentic Mediterranean a*, white stucco walls with blue accents, arched doorways and windows, terracotta floor tiles, mosaic tile borders in blue and white, exposed wooden ceiling beams, wicker furniture with blue cushions, large clay pots with olive trees, sea view through window, wrought iron details, 8K realistic photo Mediterranean mediterranean, zS, professional architectural photography, canon EOS R5, 24mm lens, f/8, high detailzcartoon, drawing, sketch, low quality, blurry, distorted, deformed, ugly, bad anatomy, watermark, text, signature, other styles mixed, messy, cluttered)get) r r r room_prompts room_desc style_prompts base_promptpromptnegative_prompts ;/home/ubuntu/projects/home-renovation-app/ai-service/app.pygenerate_promptr#s E797: L  L,GHHIF9FFFK KKK jyjjj  m)mmm  |y|||  Ci C C C IIIII M MMM e)eee f9fff A AAA!" A AAA#& QiQQQ'( U)UUU), lilll-. jIjjj/0 jIjjj1M4 ##E=+HIIK } }y } } }FpO ? "" 1024x1024r r!sizec KtstS tt|d\}}t j4d{V}|ddtddd|dd|d dgd d ||dd d dd{V}|jdkr7td|j tcdddd{VS| }|ddd}tj |} | cdddd{VS#1d{VswxYwYdS#t$r*} td| tcYd} ~ Sd} ~ wwxYw)u$调用 Stability AI API 生成图片xNzRhttps://api.stability.ai/v1/generation/stable-diffusion-xl-1024-v1-0/text-to-imagezBearer zapplication/json) Authorizationz Content-TypeAccept)textweightNONE) text_prompts cfg_scaleclip_guidance_presetheightwidthsamplesstepsgN@)headersjsontimeoutzStability AI API Error: artifactsrbase64z Error calling Stability AI API: )r generate_placeholder_imagemapintsplithttpx AsyncClientpost status_codeprintr-r;r? b64decode Exception) r r!r'r7r6clientresponsedata image_base64 image_byteses r"generate_with_stabilityrQFs ,)+++',CC11 v$&&       &#[[d%B/@%B%B$60"(155!0B??%"#,2$"   ')H,#s**@@@AAA1333              6==??D ,Q/9L *<88K?                              B ,,, 444555)++++++++,sOAD>A+D+D>=D+D>+ D55D>8D59D>> E2E-'E2-E2ctjd||fd}tj}||d|d|S)u8生成占位图（当 API 不可用时的后备方案）RGB)rUrUPNG)formatr)rnewioBytesIOsaveseekgetvalue)r7r6img img_byte_arrs r"r@r@us` )EE6?O < rsN********//////   g5666BI122U E STTT1#1#31#31#31#SX1#1#1#1#f-,-,#-,-,PS-,fk-,-,-,-,^######,  3 C SV    > RRsRRWZRRRR6 zNNNGK)$//////r$