o jT,@sddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZmZdd l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m,Z,eeee-dddZ.eddZ/e.e/j0_.e/1eee/j2edgddgdgdeddZ3e3j4dddeddZ5e5j4ddde*e3Z6e+e5Z7de8d e8fd!d"Z9d#e8d e8fd$d%Z:dgd'e;d(e8d)e8d e;fd*d+Zd/e'd0e.?d1ed2ed2ed2edededed3ed3ed3ed4eddd5d6f d,ed7e8de8d8e8d9ee@d:eeAd;eeAde8d?ee8d@eeAfdAdBZBe/jCdCdDdEge!dFe.?dGed2ed2ed2edededed3ed3ed3edededededededededed4fd,ed7e8de8d8e8d9ee@d:eeAd;eeAde8dHee8dIee@dJeeAdKeeAdLeeAdMee@dNee@dOee@dPeDd?ee8f(dQdRZEe/j>dSe#d0e.?dTed2ed2ed2ed2ed2ed2ed3ed3ed3edf d,ed7e8de8d8e8dUe8d#e8dVe@de8dPeDfdWdXZFe/jGdYe%d0e.?dTd,edZe$fd[d\ZHe/jGd]e&d0e.?dGed2ed2ed2ed3ed3ed3e de d2fd,ed7e8de8d8e8de8d=e8dHee8d^e fd_d`ZIe/>dad,efdbdcZJe/>ddd,efdedfZKdS)hN)Optional)FastAPIQueryRequest HTTPException UploadFileFileForm) JSONResponse)CORSMiddleware) load_dotenv)Image)datetime)Limiter_rate_limit_exceeded_handler)get_remote_address)RateLimitExceeded)MealGenVectorStoreScannerVectorStore)MealPlanResponseSwapMealRequestSwapMealResponseMealImageRequestMealImageResponseFoodScanResponseDailyNutritionResponseMacronutrients)EnhancedMealPlanGenerator FoodScanner)translation_service REDIS_URLzredis://localhost:6379)key_func storage_uriz+BloodFIT Meal Plan Generator - Scalable API)title*T) allow_originsallow_credentials allow_methods allow_headers faiss_mealgen) index_pathz'diet/meal_generation/mealgen_foods.jsonF) force_rebuild faiss_scannerz$diet/meal_scanner/scanner_foods.json blood_typereturncCs$|}|ddddS)N+r-)stripuppersplit)r-r4L/home/esfera/Desktop/Rishu/fitness/Fitness-lenito-AI-main/fitness_app/app.pynormalize_blood_type7s r6 sub_categorycCs:|}ddddddd}|||ddddS)Nprotein_packed light_freshhealthy_comforting)zprotein-packedzprotein packedz light & freshzlight and freshzhealthy & comfortingzhealthy and comforting _r0)lowerr1getreplace)r7mappingr4r4r5normalize_sub_category;s rA meal_plan response_datalanguage response_typecsndksts |S|dkrt|S|dkrt|S|dkr5fdd|dgD}||d<|S|S)NenrB food_scan swap_mealcsg|]}t|qSr4)rtranslate_meal).0mealrDr4r5 Os z&translate_response.. alternatives)ris_translation_enabledtranslate_meal_plan_responsetranslate_food_scan_responser>)rCrDrEtranslated_alternativesr4rLr5translate_responseGs    rSrequestcCs|jddS)NrDrF) query_paramsr>)rTr4r4r5get_language_from_requestWsrVz/calculate-daily-nutrition)response_modelz 20/minute.zStay Fiti,)gtleuser_id diet_typeageweightheightcountry food_dislikes allergies main_goaldesired_weightc  st|} t|}|}gd} gd}|| vr'd}| dkr!d}td|d||vr9d}| dkr3d }td|d|d kr?d }||||||||| | | d }t|IdH}|S) NOABABclassicvegan pescatarian vegetarian+Invalid blood type. Must be O, A, B, or AB.koE잘못된 혈액형입니다. O, A, B, AB 중 하나여야 합니다. status_codedetailGInvalid diet type. Must be classic, vegan, pescatarian, or vegetarian.y잘못된 식단 유형입니다. 클래식, 비건, 페스카테리안, 카니보어, 비건 중 하나여야 합니다.rkstandard) r[r-r\r]r^r_r`rarbrcrd)rVr6r=rmeal_generatorcalculate_daily_nutrition)rTr[r-r\r]r^r_r`rarbrcrdrDvalid_blood_typesvalid_diet_types error_msg user_datanutrition_responser4r4r5rz^s@  rzz/generate-meal-planGETPOST)methodsrWz 10/minuteblood_group_rulestotal_daily_caloriescarbsproteinfatbreakfast_calorieslunch_caloriesdinner_caloriesgenerate_imagesc st|}t|}|}| s.|jdkr.z|IdH}|dp"d} Wn ty-Ynwgd}gd}||vrHd}|dkrBd}td|d ||vrZd }|dkrTd }td|d |d kr`d }||||||||| | |d }d}| r| r| r|ri}|r||d<|r||d<|r||d<t| t | | |d|||t |r|ndd}zt j |||dIdH}|}t||d}t|dWSty}zdt|}|dkrdt|}td|d d}~ww)Nrrrerjrorprqrrrsrvrwrkrx) r[r-r\r]r^r_r`rarbrrc breakfastlunchdinner) carbohydratesrr)rtotal_daily_macronutrientsr[r-r\calculation_timestampmeal_calorie_distributionrrBcontentzMeal plan generation failed: u식단 계획 생성 실패: )rVr6r=methodformr> Exceptionrrrrnowrygenerate_meal_plandictrSr str) rTr[r-r\r]r^r_r`rarbrrrrrrrrrrcrDrr{r|r}r~daily_nutritionmeal_distributionmeal_plan_response response_dicttranslated_responseer4r4r5rs         rz /swap-mealz 15/minutecategorycurrent_caloriesc  s&t|} t|}|}|}t|}gd} gd} || vr/d}| dkr)d}td|d|| vrAd}| dkr;d }td|d|d krGd }t||||||||| d }ztj|| d IdH}|}t || d}t |dWSt y}zdt |}| dkrdt |}td|dd}~ww)Nrerjrorprqrrrsrvrwrkrx) r[r-r\rr7rr`rarbrrHrzSwap meal generation failed: u대체 식사 생성 실패: r) rVr6r=rArrrygenerate_swap_mealsrrSr rr)rTr[r-r\rr7rr`rarbrrDr{r|r} swap_request swap_responserrrr4r4r5rHsV      rHz/generate-meal-images meal_requestc szt|}dd|jD}tj|j|j|}t||jdWSt yG}zdt |}t|}|dkr=dt |}t d|dd}~ww) NcSsg|] }|j|jp ddqS)somenamequantityr)rJingr4r4r5rMNsz(generate_meal_images..)meal_image_base64idzImage generation failed: rpu이미지 생성 실패: rrs) rV ingredientsrymeal_image_generatorgenerate_meal_image meal_name descriptionrrrrr)rTrrDingredients_listrrr}r4r4r5generate_meal_imagesFs. rz /scan-foodimagec  s\t|} t|}|}|s|jdpd}gd} gd} || vr1d} | dkr+d} td| d|| vrCd } | dkr=d } td| d|d krId }|jd s]d} | dkrWd} td| dz[|IdH} zt t | }| Wntyd} | dkrd} td| dwt t | }|jdks|jdkrd} | dkrd} td| d|jdkr|d}|Wn+tyty}zdt|} | dkrdt|} td| dd}~ww|||||||d}zt| |IdH}|}t|| d}t|dWSty-}zdt|} | dkr#dt|} td| dd}~ww)Nrrerjrorprqrrrsrvrwrkrxzimage/zFile must be an imageu#파일은 이미지여야 합니다zInvalid image fileu#잘못된 이미지 파일입니다dz1Image too small. Minimum 100x100 pixels required.uK이미지가 너무 작습니다. 최소 100x100 픽셀이 필요합니다.RGBzError processing image: u이미지 처리 오류: )r[r-r\r`rarbrrGrzFood scanning failed: u음식 스캔 실패: r)rVr6r=rUr>r content_type startswithreadr openioBytesIOverifyrwidthr_modeconvertcloser food_scannerscan_food_imagerrSr )rTr[r-r\r`rbrarrrDr{r|r} image_dataimgrr~ scan_responserrr4r4r5 scan_foodgs                   rz/healthcsvt|}tjdu}|rttjjnd}|rdnd}|dkr$|r"dnd}|d|||d gd d d d ddddtd S)u6Health check endpoint – now uses vector store stats.Nrhealthydegradedrpu정상u 저하됨BloodFIT Meal Plan APIA9 meals per category (3 per sub‑category × 3 sub‑categories)zProtein-Packedz Light & FreshzHealthy & Comfortingz0Calculate daily nutrition with meal distributionu*Generate 27‑meal plan with new structureu&Swap meals with sub‑category supportzScan food imagesGenerate meal images)z/meal/calculate-daily-nutritionz/meal/generate-meal-planz/meal/swap-mealz/meal/scan-foodz/meal/generate-meal-images) statusservicerDvector_store_loadedtotal_document_chunks new_structuresub_categoriestotal_meals_per_plan endpointstranslation_enabled)rV meal_store vectorstorelenindex_to_docstore_idrrO)rTrD index_loaded doc_countstatus_messager4r4r5 health_checks.   r/c sbt|}|dkrdddgddddd d d d gd dSdddgdddddddd gddS)zAPI DocumentationrpuBloodFIT 식단 계획 APIz3.0 (Vector Store)u`각 카테고리당 9개 식사 옵션 (하위 카테고리당 3개 × 3개 하위 카테고리))u단백질 풍부u가볍고 신선함u건강하고 편안함ru일일 영양소 목표 계산u(새로운 구조로 식단 계획 생성u8동일한 하위 카테고리에서 대체 식사 생성u음식 이미지 스캔u식사 이미지 생성)z#GET /meal/calculate-daily-nutritionzGET /meal/generate-meal-planzGET /meal/swap-mealzPOST /meal/scan-foodzPOST /meal/generate-meal-images)uN1. /meal/calculate-daily-nutrition 호출하여 일일 영양소 목표 계산uF2. 결과를 /meal/generate-meal-plan에 쿼리 매개변수로 전달uD3. 또는 /meal/generate-meal-plan 직접 호출하여 자동 계산)messageversionu구조u하위_카테고리u총_식사_옵션u엔드포인트u 사용법rrrz!Calculate daily nutrition targetsz%Generate meal plan with new structureu5Generate alternative meals within same sub‑categoryz'Scan food image for harmful ingredientsr)zF1. Call /meal/calculate-daily-nutrition to get daily nutrition targetszF2. Pass the results to /meal/generate-meal-plan using query parameterszF3. Or call /meal/generate-meal-plan directly for automatic calculation)rr structurertotal_meal_optionsr usage_flow)rV)rTrDr4r4r5homes:r)rB)Losruvicorntypingrfastapirrrrrrr fastapi.responsesr fastapi.middleware.corsr dotenvr PILr rslowapirr slowapi.utilrslowapi.errorsrfitness_app.vector_storerrfitness_app.data_managerrrrrrrrrfitness_app.ai_servicesrrrgetenvlimiterappstateadd_exception_handleradd_middlewarerload_or_create_index scanner_storeryrrr6rArrSrVr>limitintfloatrz api_routeboolrrHpostrrrrr4r4r4r5s $       (            6      m    ?  c