J7ih ddlmZmZmZmZddlmZddlmZddl m Z ddl m Z ddlZddlZddlmZddlZddlmZddlZddlZddlZddlmZddlZdd lmZddlZeej:ej<d ej>e Z!ed d dZ"e"jGedgddgdggZ$ejJdZ&ejJdZ'ejJdZ(ejJdZ)ejJdZ*e&e'e(e)e*fD]Z+e+se$jYe+e$se!j[de.dda/e!jade1e$ddZ2dZ3e1e$Z4gZ5GddeZ6Gd d!eZ7d"Z8d_d#e jd$e9d%e jfd&Z:d'e;d(e;d)e9d%e6fd*Zd%e6f d-Z?e"jd.d/ZAe"jd0d1ZBe"jd2e73ed4fd5eefd6ZDe"jd7d8ZEe"jd9d:ZFe"jd;dk(rddlIZIeJd?eJd@eJdAeJdBeJdCe1e$eJdDe2dEeJdFe3dGeJdHe1e$dIeJdJeJdKeJeJdLeJdMeJdNeJdOeJdPeJdQeJdReJdSeJdTeJeJdUe2dVeJdWe1e$dXzdYe1e$dZeJd[eIje"d\d]^yy)a)FastAPIFile UploadFile HTTPException)CORSMiddleware)StreamingResponseN)Image)List) BaseModel) load_dotenv)datetimez4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatz"Tree Species and DBH Detection APIz*Upload 1-10 tree images for batch analysis2.0.0)title descriptionversion*T) allow_originsallow_credentials allow_methods allow_headersGEMINI_API_KEYGEMINI_API_KEY_2GEMINI_API_KEY_3GEMINI_API_KEY_4GEMINI_API_KEY_5u❌ No GEMINI_API_KEY found!z5At least one GEMINI_API_KEY must be set in .env file!u ✅ Loaded z API key(s) for rotation g@ceZdZUdZdZeed<eed<eed<dZ eed<dZ eed <dZ eed <dZ eed <d Z eed <dZeed<dZeed<dZeed<dZeed<dZeed<gZeeed<dZeed<dZeed<dZeed<dZeed<dZeed<y) TreeAnalysiszSingle tree analysis resultTsuccessfilenameimage_idUnknownspeciesN species_hindispecies_marathiscientific_nameLow confidenceheightcanopygirth dbh_estimate conditioncharacteristicsadditional_inforecommendationsremarks timestamperror)__name__ __module__ __qualname____doc__r!bool__annotations__strintr%r&r'r(r*r+r,r-r.r/r0r r1r2r3r4r5+/var/www/html/treecensus/species_ai/main.pyr r Ks%GTMMGSM3OSOSJFCFCE3L#Is!#OT#Y#OSOSGSIsE3r?r cTeZdZUdZeed<eed<eed<eed<eed<eeed<y) BatchAnalysisResponsez Response for batch tree analysis total_images successfulfailedunhealthy_treesprocessing_timeresultsN) r6r7r8r9r=r;floatr r r>r?r@rBrBbs,*O K , r?rBcRtt}tdzttza|S)zGet next API key in rotation)GEMINI_API_KEYScurrent_key_indexlen)keys r@get_next_api_keyrPls) + ,C*Q.#o2FF Jr?imagemax_sizereturnct|j|kDrN|j||ftjj t jd|j|S)zResize image if too largez Resized to )maxsize thumbnailr ResamplingLANCZOSloggerinfo)rQrRs r@ resize_imager\tsM 5::! 8,e.>.>.F.FG nUZZL12 Lr? response_textr"r#cd||ddgtjjd}|jd}d}|D]}|j }|j dr)|j ddj }|sH||d <N|j d r<|j d dj }|s|jd vs||d <|j d r<|j d dj } | s| jd vs| |d<|j dr+|j ddj } | s| |d<$|j dr+|j ddj } | sY| |d<`|j dr?|j ddj } | sd| jvs| |d<|j dr?|j ddj } | sd| jvs| |d<|j dr?|j ddj }|s5d|jvsI||d<P|j dr?|j ddj }|sd|jvs||d<|j dr+|j ddj }|r||d<d}|j drd}|j d rd!}|j d"rd#}|j d$rd%}0|j d&s|j d'sT|d(dj }|sk|dk(r|dj||d!k(r"d!|vs|d!s||d!<|d!xxd)|zz cc<|d#k(r"d#|vs|d#s||d#<|d#xxd)|zz cc<|d%k(sd%|vs|d%s||d%<|d%xxd)|zz cc<td*i|S)+z2Parse Gemini response into structured TreeAnalysisTr$r))r!r"r#r%r*r0r4 Nz ENGLISH NAME:r%z HINDI NAME:)z not availablenazn/ar&z MARATHI NAME:r'zSCIENTIFIC NAME:r(z CONFIDENCE:r*zHEIGHT:notr+zCANOPY:r,zGIRTH:r-zDBH:r.z CONDITION:r/zCHARACTERISTICS:r0zADDITIONAL INFO:r1zRECOMMENDATIONS:r2zREMARKS:r3z- u•  r>) r now isoformatsplitstrip startswithreplacelowerappendr )r]r"r#datalinescurrent_sectionliner%hindimarathisciconfr+r,r-dbhconditems r@parse_gemini_responserx|s\\^--/ D    %EOX2zz| ??? +ll?B7==?G")Y __] +LL399;E.LL(-_% ___ -ll?B7==?G7==?2PP*1&' __/ 0,,126<<>C*-&' __] +<< r288:D%)\" __Y '\\)R0668F%v||~5!'X __Y '\\)R0668F%v||~5!'X __X &LL2.446Eekkm3 %W __V $,,vr*002CuCIIK/'*^$ __\ *<< b1779D$([!)O __/ 0/O __/ 0/O __/ 0/O __Z ('O __T "doof&=8>>#D"33&'..t4 $55$D0=N8O.2D*+*+sTz9+ $55$D0=N8O.2D*+*+sTz9+ I-D(Y&*DOOsTz1OqX2t  $ r? image_data add_delayc K|rAtjdtd|dtjtd{tjd|d| t j tj|}tjd|jd|j|jd k(rHt jd |jd }|j||jd  |}n |jd k7r|jd }t|}tj}|j!|dd|j#}d} t9t:D]j} t=} t?j@| t?jBd}tjdtDtGtHzdzdtGtHd| dzdt:d |jK| dtMjN|jQdd g}tS|jTjW||}tjd!|d"|jXd#|jZd$tjd%|j\d&|j^t2j5|j7|cSy7#t$$r}dt'|} tj)d|d| t+d||| t-j.j1} t2j5| j7| cYd}~Sd}~wwxYw#t$$r}t'|} d'| vs%d(| javsd)| javrtjcd*|d+| dzdt:| t:dz kr9tjd,tjd-d{7Yd}~tj)d|d.tGtHd/t+d||d0tGtHd1t-j.j1} t2j5| j7| cYd}~cStj)d|d2|d3| t+d||| t-j.j1} t2j5| j7| cYd}~cSd}~wwxYww)4zIAnalyze single tree image with AI (with retry logic and API key rotation)u⏳ Rate limit delay: z s before z...Nu📸 [z] Processing: z Image size: z, Mode: RGBARGB)r~r~)maskJPEGU)rqualityzImage processing error: u❌ [] F)r!r"r#r5r4uAnalyze this tree image thoroughly and provide detailed information in the EXACT format below: ENGLISH NAME: [Common English name of the tree species] HINDI NAME: [Hindi name if known, otherwise "Not available"] MARATHI NAME: [Marathi name if known, otherwise "Not available"] SCIENTIFIC NAME: [Full scientific name - Genus species] CONFIDENCE: [High/Medium/Low - based on visible features and image clarity] HEIGHT: [Estimated height in meters, e.g., "12-15 meters" or "Not clearly visible"] CANOPY: [Canopy diameter/spread in meters, e.g., "8-10 meters" or "Not clearly visible"] GIRTH: [Tree circumference at breast height (1.3m), e.g., "2.5 meters" or "Not measurable from image"] DBH: [Diameter at Breast Height calculated from girth, e.g., "79.6 cm" or "Not calculable"] CONDITION: [Overall health - Healthy/Good/Fair/Poor/Declining with detailed explanation of visible signs] CHARACTERISTICS: - [Bark texture, color, and pattern details] - [Leaf shape, size, arrangement, and color] - [Branch structure and growth pattern] - [Any flowers, fruits, or seeds visible] - [Unique identifying features] - [Additional observable characteristics] ADDITIONAL INFO: [Ecological information - native habitat, typical growth conditions, cultural significance, common uses, interesting facts about this species] RECOMMENDATIONS: [Specific care recommendations for this species - watering needs, pruning guidelines, fertilization, pest management, best growing conditions, seasonal care tips] REMARKS: [Specific observations about THIS tree - any visible issues, notable features, age estimation if possible, overall assessment] IMPORTANT: 1. Be as specific and detailed as possible with all measurements and descriptions 2. For DBH calculation: DBH = Girth / π (if girth is measurable) 3. Provide botanical details to justify species identification 4. Include both common and scientific nomenclature 5. Be honest about confidence level - explain if uncertain 6. Base all observations on what's actually visible in the image)api_keyzmodels/gemini-2.5-flashu 🔑 Using API key #rK/z (attempt z)...z image/jpegzutf-8) mime_typermu✅ [z ] SUCCESS: z ()z DBH: z , Condition: 429quotaexceededu⚠️ [z] Quota exceeded on attempt u🔄 Trying next API key...rcz] All z API keys exhaustedzAll zQ API keys have exceeded their quota. Please try again later or add more API keys.z ] FAILED:  - )2rZr[RATE_LIMIT_DELAYasynciosleepr openioBytesIOrVmodenewpastergconvertr\savegetvalue Exceptionr<r5r r rerfanalysis_historyrl model_dumprange MAX_RETRIESrPgenai configureGenerativeModelrMrNrLgenerate_contentbase64 b64encodedecoderxtextrhr%r(r.r/rkwarning)ryr"r#rzrQ background img_buffer img_bytese error_msg error_resultpromptretryrmodelresponseresults r@analyze_tree_imagers ,-=,>izQTUVmm,--- KK& . ;< 2::j12 oejj\%**FG :: 5%**oFJ   Uq)9  :E ZZ5 MM%(EU#ZZ\  :fb 9'') DF@{#:$9 $&(G OOG ,))*CDE KK13Ds?G[3[_`2`1aabcfgvcwbxyCDILMDMCNNOP[O\\`a b--*F4D4DY4O4V4VW^4_`/H +8==+>+>+@(HUF KK%zV^^4DBvG]G]F^^_` a KK(6#6#6"7}VEUEUDVW X  # #F$5$5$7 8M+:$O .2  .s1vh7  uXJb 45#lln..0    7 7 9: J# $AI !W 0A%AZS\SbSbSdEd(3OPUXYPY{Z[\g[hij;?*KK"=?!--***LL5 &_9M8NNa!bc#/ %!)!) $S%9$:;LM"*,,.":":"< $L%++L,C,C,EF'' uXJj #i[QR+!%%#&lln668   !'' (?(?(AB##G# $s?U K3U "DK63U  E%N .U 6 N?A=N<N=U NU  UBU&P)'U,U 2BUUU A2U:U;U UU rc Kdddttdttdtddj tdd d d d d Sw)zAPI Informationu%🌳 Tree Species & DBH Detection APIrzFlexible Batch Processing seconds between imagesz Upload 1-z tree images for analysisz/POST /api/analyze-multiple (upload 1-{} images)zGET /api/healthzGET /api/statszGET /api/export/csvz GET /docs)analyzehealthstatsexportdocs)namerr max_imagesrate_limit_delayapi_keys_availablemessage endpoints) MAX_IMAGESrrNrLrr>r?r@rootrrs_8+ /00GH!/2zl*CDHOOPZ['%+   sAAz /api/healthc KdttdkDttdttttdSw)z Health CheckhealthyrzFLEXIBLE BATCH PROCESSING)statusapi_configuredapi_keys_countrrrtotal_analyses)rNrLrrrr>r?r@ health_checkrs<o.2o.+ ,./ s=?z/api/analyze-multiple)response_model.filesc Ktj}tjdtjdtjdt|tjdtdtjdtt tjdt|dkr"tj dtd d t|tkDrPtj d t|d ttd dtdt|dtd t|D]i\}}|jr|jjdr.tj d|jtd d|jd  tjdg}t|D]t\}}|jd{}|j||j|dzftjd|dzd|jdt|ddvtjdtdt|tz}tjd|d d!g}t|D]6\}\}} } |d"kD} t|| | | #d{} |j| 8t!d$|D} t!d%|D}gd&t!fd'|D}tj|z }tjdtjd(tjd)t|d*| d+|d,|tjd-|d.d!tjdt#t|| ||t%|d/|0S77#t$rt&$rf}tj d1t)|tj t+j,td2d3t)| d}~wwxYww)4u  🌳 FLEXIBLE TREE ANALYSIS - Upload 1-10 tree images Features: - Upload anywhere from 1 to 10 images (botanist's choice) - Rate limited (5.5s delay between images) - Automatic API key rotation on quota exceeded - Comprehensive analysis: Species, DBH, Height, Canopy, Girth, Health - Regional names (Hindi & Marathi) - Detailed characteristics and recommendations Returns: - Complete analysis for all trees - Success/failure statistics - Unhealthy tree count - Processing time P================================================================================u🌳 BATCH ANALYSIS REQUESTu📊 Images received: u📋 Allowed range: 1-z imagesu🔑 API keys available: rKu❌ No images uploadedzPlease upload at least 1 image. status_codedetailu❌ Too many images: z > zMaximum z( images allowed per batch. You uploaded z images. Please select up to z images only.zimage/u❌ Invalid file type: zFile 'z=' is not an image. All files must be images (JPG, PNG, WEBP).u📂 Reading uploaded files...Nu ✓ [rr,z bytesu)🔄 Starting sequential processing with zs rate limiting...u⏱️ Estimated time: ~z.0fz secondsr)rzc3:K|]}|jsdywrKNr!.0rs r@ z)analyze_multiple_trees..s9qqyy9c3:K|]}|jrdywrrrs r@rz)analyze_multiple_trees..s91qyyQ9r)poor declining unhealthydiseasedstresseddamagedweakc3K|]4jsjstfdDrd6yw)c3VK|] }|jjv"yw)N)r/rk)rwordrs r@rz3analyze_multiple_trees...s#V441;;#4#4#66Vs&)rKN)r!r/any)rrunhealthy_keywordss @r@rz)analyze_multiple_trees..s5XaaiiAKKVCUVVXs > >>u✅ BATCH ANALYSIS COMPLETEDu 📊 Total: z | Success: z | Failed: z | Unhealthy: u⏱️ Processing time: .1frc)rCrDrErFrGrHu❌ CRITICAL ERROR: izServer error during analysis: )timerZr[rNrrLr5r enumerate content_typerir"readrlrrsumrBroundrr< traceback format_exc)r start_timeidxfile files_datacontentsestimated_timerHrmrimg_idrzrrDrErrGrrs @r@analyze_multiple_treesrs&J KK KK-/ KK(U 56 KK( G<= KK+C,@+ABC KK 5zA~ -/4   5zJ ,SZLJ<HIj\)QRUV[R\Q]^**4]D  u% T  (9(9(D(DX(N LL24==/B C .kl 7  45 "5) WIC!YY[(H   xa@ A KK(3q5'DMM?#c(mA=NfU V W  ?@P?QQcdeU&66 /s/C8LM)2:)> # %C%$fqI-dD&IVVF NN6 "  #9G99 999kX7XX ))+ 2 F 24 l3u:,l:,kRXQYYghqgrst //DHMN F$U!%!/15   C)W:    +CF845 Y))+,3CF8<   sWFQ'>Q';O,?O&CO,O)DO,%Q'&O,)O,,Q$>A!QQ$$Q'z /api/statsc KtsddiStt}tdtD}||z }i}tD]C}|jds|jds'|d}|j|ddz||<E||||dkDr ||z dzd d nd t t |j d d dddSw)zGet analysis statisticsrz9No analysis history available yet. Upload trees to begin!c3FK|]}|jddsdyw)r!FrKN)get)ras r@rz!get_statistics.. sL1AEE)U4KQLs!!r!r%rrKdr%z0%c |dS)NrKr>)xs r@z get_statistics..s QqTr?T)rOreverseNr)rrDrE success_ratespecies_frequency)rrNrrdictsorteditems)totalrDrE species_freqrr%s r@get_statisticsr s VWW  !EL 0LLJ Z FL E 55 i 0 lG$0$4$4Wa$@1$DL !E   [_"`adbd"ef  sACC&ACz/api/export/csvc JKts tddtj}gd}t j ||}|j tD]}|jid|jddd|jddd |jd dd |jd dd |jd dd |jd dd |jd dd|jddd|jddd|jddd|jddd|jddd|jddddj|jdgd|jddd|jddd|jdd|jdd|jddd|jdtt|jgddd i!Sw)"z"Export all analysis history as CSVizNo analysis history availabler)r4r#r"r%r(r&r'r*r+r,r-r.r/r0r1r2r3r!r5 fieldnamesr4r`r#r"r%r(r&r'r*r+r,r-r.r/r0; r1r2r3r!Fr5)r!r5rtext/csvContent-Dispositionz.attachment; filename=tree_analysis_history.csv media_typeheaders)rrrStringIOcsv DictWriter writeheaderwriterowrjoinseekriterr)outputr writerrecords r@export_all_historyrsU 4STT [[]FJ ^^Fz :F "  K4  :r2   :r2  vzz)R0  vzz*;R@  VZZ<   vzz*;R@  &**\26  fjj2.  fjj2.  VZZ,  FJJ~r:  K4  tyy4Er)JK  vzz*;R@ vzz*;R@! " vzz)R0# $zz)U3ZZ,'   . KKN  foo !&*XZ sH!H#z/api/export/csv/resultsrHc bK|s tddtj}gd}tj||}|j |D]z}|j id|jddd|jddd |jd dd |jd dd |jd dd |jd dd |jd dd|jddd|jddd|jddd|jddd|jddd|jddddj|jdgd|jddd|jddd|jddd|jddi}|jdtt|jgdddtjjdd i!Sw)"zExport provided results as CSVrzNo results to exportr)r4r#r"r%r(r&r'r*r+r,r-r.r/r0r1r2r3r!r r4r`r#r"r%r(r&r'r*r+r,r-r.r/r0r r1r2r3r!Frrrz#attachment; filename=tree_analysis_z %Y%m%d_%H%M%Sz.csvr)rrrrrrrrrrrrrr restrftime)rHrr rrs r@export_current_resultsr!Nsf 4JKK [[]FJ ^^Fz :F   K4  :r2   :r2  vzz)R0  vzz*;R@  VZZ<   vzz*;R@  &**\26  fjj2.  fjj2.  VZZ,  FJJ~r:  K4  tyy4Er)JK  vzz*;R@ vzz*;R@! " vzz)R0# $ vzz)U3%   , KKN  foo !&*MhllnNeNefuNvMww{(|} sH-H/__main__zQ ================================================================================u%🌳 TREE SPECIES & DBH DETECTION APIru4🔧 AI System: Advanced Vision Analysisu🔑 API Keys Loaded: u(📊 Upload Mode: FLEXIBLE (1-z images)u⏱️ Rate Limit: ru+🔄 Auto Retry: Enabled (up to z keys)u1🌐 Server: http://localhost:8000u6📚 API Documentation: http://localhost:8000/docsu📋 Analysis Includes:u7 ✓ Species identification (English, Hindi, Marathi)u1 ✓ DBH (Diameter at Breast Height) estimationu% ✓ Height and Canopy measurementsu ✓ Girth measurementu" ✓ Health condition assessmentu ✓ Detailed characteristicsu ✓ Care recommendationsu ✓ CSV export functionalityu4💡 Botanist-Friendly: Upload any number from 1 to z trees!u🎯 Daily Capacity: ~z requests with z API keyszQ================================================================================ z0.0.0.0i@)hostport)i)F)Lfastapirrrrfastapi.middleware.corsrfastapi.responsesrgoogle.generativeai generativeairPILr rostypingr loggingpydanticr rrrdotenvr rr r basicConfigINFO getLoggerr6rZappadd_middlewarerLgetenv api_key_1 api_key_2 api_key_3 api_key_4 api_key_5rOrlr5 ValueErrorrMr[rNrrrrr rBrPr=r\r<rxbytesr:rrrrpostrr rrr!uvicornprintrunr>r?r@rBsZ<<2/#      ,, A   8 $ .< %%%  BII& ' BII( ) BII( ) BII( ) BII( )  y)Y B$C s#$ LL/0 L MM k#o.//G HI /" 9. I  sekki i i si |i XG$G$#G$G$Y]G$jvG$T(   !2GH;?9l Z(8l Il ^2 ++\ #$*$t**%*Z z - 12 &M @B (_)=(> ?@ 4ZL IJ +,<+==T UV 7O8L7MV TU =? BD #J #% CE =? 13 $& .0 +- ') +- #J @ G TU "3#7"#<"=_SQ`MaLbbk lm -GKK)$/5r?