Viewing File: /home/ubuntu/combine_ai/bg_upscale/src/utils.py
import json
import os
import uuid
import numpy as np
current_directory = os.getcwd()
src_directory = os.path.join(current_directory, "src")
# Image Scale Directory
image_scale_directory = os.path.join(current_directory, "imscale")
scale_data_directory = os.path.join(image_scale_directory, "data")
scale_user_directory = os.path.join(scale_data_directory, "user_data")
scale_app_directory = os.path.join(scale_data_directory, "app_data")
scale_input_directory = os.path.join(scale_user_directory, "input_data")
scale_output_directory = os.path.join(scale_user_directory, "output_data")
# Remove Background Directory
remove_background_directory = os.path.join(current_directory, "removebg")
rmg_data_directory = os.path.join(remove_background_directory, "data")
rmg_user_directory = os.path.join(rmg_data_directory, "user_data")
rmg_app_directory = os.path.join(rmg_data_directory, "app_data")
rmg_input_directory = os.path.join(rmg_user_directory, "input_data")
rmg_output_directory = os.path.join(rmg_user_directory, "output_data")
# Environment Variables
os.environ['OUTPUT_MEDIA_URL'] = "https://combine-ai.botfingers.com:8013/static"
os.environ['CURRENT_DIRECTORY'] = current_directory
os.environ['SRC_DIRECTORY'] = src_directory
os.environ['IMAGE_SCALE_DIRECTORY'] = image_scale_directory
os.environ['SCALE_DATA_DIRECTORY'] = scale_data_directory
os.environ['SCALE_INPUT_DIRECTORY'] = scale_input_directory
os.environ['SCALE_OUTPUT_DIRECTORY'] = scale_output_directory
os.environ['SCALE_USER_DIRECTORY'] = scale_user_directory
os.environ['SCALE_APP_DIRECTORY'] = scale_app_directory
os.environ['REMOVE_BACKGROUND_DIRECTORY'] = remove_background_directory
os.environ['RMG_DATA_DIRECTORY'] = rmg_data_directory
os.environ['RMG_INPUT_DIRECTORY'] = rmg_input_directory
os.environ['RMG_OUTPUT_DIRECTORY'] = rmg_output_directory
os.environ['RMG_USER_DIRECTORY'] = rmg_user_directory
os.environ['RMG_APP_DIRECTORY'] = rmg_app_directory
AWS_ACCESS_KEY_ID="AKIAXWYZ72NY4I63JG76"
AWS_SECRET_ACCESS_KEY="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"
AWS_DEFAULT_REGION="us-east-1"
os.environ['AWS_ACCESS_KEY_ID'] = AWS_ACCESS_KEY_ID
os.environ['AWS_SECRET_ACCESS_KEY'] = AWS_SECRET_ACCESS_KEY
os.environ['AWS_DEFAULT_REGION'] = AWS_DEFAULT_REGION
Back to Directory
File Manager