Update setup_environment.py

setup_environment.py

@@ -113,7 +113,29 @@ def add_to_ld_library_path(library_dir):
         print("유효한 라이브러리 경로가 없습니다.")
 
 
-# 전체 환경 설정 프로세스 실행 함수
+def download_files_from_url():
+    os.makedirs('/tmp/stf',exist_ok=True)
+    os.makedirs('/tmp/stf/TEMP',exist_ok=True)
+    
+    file_id=os.getenv('DOWNLOAD_1')
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, '/tmp/stf-api-alternative.zip', quiet=False)
+    zipfile.ZipFile('/tmp/stf-api-alternative.zip', 'r').extractall('/tmp/stf/')
+
+    file_id=os.getenv('DOWNLOAD_2')
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, '/tmp/stf/089.pth', quiet=False)
+    
+
+    file_id=os.getenv('DOWNLOAD_3')
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, '/tmp/stf/TEMP/0157.pth', quiet=False)
+
+
+
+
+
+
 def initialize_environment():
     """환경 설정을 위한 전체 프로세스 실행"""
     print('aaaaaaaaaaaa')
@@ -125,6 +147,9 @@ def initialize_environment():
     print('11111111')
     check_stf_alternative_installed()
 
+    # 여기서 파일 다운로드
+    download_files_from_url()
+
     # stf-api-alternative에서 pip install . 실행
     run_pip_install("/tmp/stf/stf-api-alternative")