o ^ h @s8ddlmZddlmZddlmZddlZddlZddlZddl m Z eddZ e j edgd dgdgd d d Z e Zed eejdd dejdd de jdedddde jdeddddddlmZddlmZe jejddgde jejddgdedkrddlZeje ddd dSdS)!)FastAPI)CORSMiddleware) StaticFilesN)Pathz/backend) root_path*T) allow_originsallow_credentials allow_methods allow_headersc Cstjtjt}tj|dd}tj|odtjtj|dodtjtj|dodtjtj|dodtjtj|dodtjtj|dodtjtj|dodtjtj|d }|std |d zUtj |zd d l m }|stdt dWn9t ytj|d}t j tj|gdddd}|jd krtdtd|jtd|jtdYnwWnty}ztdt|tdWYd}~nd}~ww|tjd<|S)Nmodelszgrounding-dinoz config.jsonzmodel.safetensorszpreprocessor_config.jsonzspecial_tokens_map.jsonztokenizer_config.jsonztokenizer.jsonz vocab.txtzModel not found at z. Downloading...r)download_modelz-Failed to download model using import method.zDownload failedzdownload_model.pyTF)capture_outputtextcheckzError downloading model:zSTDOUT: zSTDERR: z@Failed to download model. Please check the error messages above.zERROR: zJThe application will continue, but image comparison features may not work. MODEL_DIR)ospathdirnameabspath__file__joinexistsprintsysappendr ImportError subprocessrun executable returncodestdoutstderr RuntimeError Exceptionstrenviron) current_dir model_dirmodel_files_existr download_scriptresulter-6/var/www/html/construction_image-detection-poc/main.pyensure_model_existssb        r/zUsing model directory: Input_Images)exist_ok Output_Imagesz /Input_Images) directory input_images)namez/Output_Images output_images)auth)uploadz/apir7)prefixtagsr8__main__z 127.0.0.1i)hostport)fastapirfastapi.middleware.corsrfastapi.staticfilesrrrrpathlibrappadd_middlewarer/r(rmakedirsmount src.routersr7r8include_routerrouter__name__uvicornrr-r-r-r.s<     2