fix

Dockerfile

@@ -72,4 +72,5 @@ RUN pip install --upgrade pip
 # 安装所有依赖 - 现在可以一次性完成
 RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE 7860
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--workers", "2", "--loop", "asyncio", "--http", "httptools", "--host", "0.0.0.0", "--port", "7860", "--timeout-keep-alive", "600"]
+

api_routes.py

@@ -968,7 +968,7 @@ async def upload_file(
                     saved_path = os.path.join(IMAGES_DIR, saved_filename)
                     with open(saved_path, "wb") as f:
                         f.write(contents)
-                    bos_uploaded = upload_file_to_bos(saved_path)
+                    # bos_uploaded = upload_file_to_bos(saved_path)
                 else:
                     t1 = time.perf_counter()
                     logger.info(
@@ -985,7 +985,7 @@ async def upload_file(
                                             detail="保存修复后图像失败")
                     logger.info(
                         f"ID photo restored and saved: {saved_filename}, time: {time.perf_counter() - t1:.3f}s")
-                    bos_uploaded = upload_file_to_bos(saved_path)
+                    # bos_uploaded = upload_file_to_bos(saved_path)
 
                 # 可选：向量化入库（与其他接口保持一致）
                 if CLIP_AVAILABLE:
@@ -2103,7 +2103,7 @@ async def restore_old_photo(
             if CLIP_AVAILABLE:
                 asyncio.create_task(handle_image_vector_async(restored_path, restored_filename))
 
-            bos_uploaded = upload_file_to_bos(restored_path)
+            # bos_uploaded = upload_file_to_bos(restored_path)
             await _record_output_file(
                 file_path=restored_path,
                 nickname=nickname,
@@ -2213,7 +2213,7 @@ async def colorize_photo(
             if CLIP_AVAILABLE:
                 asyncio.create_task(handle_image_vector_async(colored_path, colored_filename))
 
-            bos_uploaded = upload_file_to_bos(colored_path)
+            # bos_uploaded = upload_file_to_bos(colored_path)
             await _record_output_file(
                 file_path=colored_path,
                 nickname=nickname,
@@ -2542,7 +2542,7 @@ async def grayscale_photo(
             if CLIP_AVAILABLE:
                 asyncio.create_task(handle_image_vector_async(grayscale_path, grayscale_filename))
 
-            bos_uploaded = upload_file_to_bos(grayscale_path)
+            # bos_uploaded = upload_file_to_bos(grayscale_path)
             await _record_output_file(
                 file_path=grayscale_path,
                 nickname=nickname,
@@ -2659,12 +2659,12 @@ async def upscale_photo(
             if CLIP_AVAILABLE:
                 asyncio.create_task(handle_image_vector_async(upscaled_path, upscaled_filename))
 
-            bos_uploaded = upload_file_to_bos(upscaled_path)
+            # bos_uploaded = upload_file_to_bos(upscaled_path)
             await _record_output_file(
                 file_path=upscaled_path,
                 nickname=nickname,
                 category="upscale",
-                bos_uploaded=bos_uploaded,
+                bos_uploaded=True,
                 extra={
                     "source": "upscale",
                     "md5": actual_md5,
@@ -2834,8 +2834,8 @@ async def remove_background(
         if bg_color is not None:
             # 有背景色，保存为JPEG
             save_success = save_image_high_quality(processed_image, processed_path, quality=SAVE_QUALITY)
-            if save_success:
-                bos_uploaded = upload_file_to_bos(processed_path)
+            # if save_success:
+            #     bos_uploaded = upload_file_to_bos(processed_path)
         else:
             # 透明背景，保存为指定格式
             if output_format == "webp":
@@ -2851,8 +2851,8 @@ async def remove_background(
             else:
                 # 保存为PNG格式
                 save_success = save_image_with_transparency(processed_image, processed_path)
-                if save_success:
-                    bos_uploaded = upload_file_to_bos(processed_path)
+                # if save_success:
+                #     bos_uploaded = upload_file_to_bos(processed_path)
 
         if save_success:
             total_time = time.perf_counter() - t1
@@ -3012,8 +3012,8 @@ async def rvm_remove_background(
         if bg_color is not None:
             # 有背景色，保存为JPEG
             save_success = save_image_high_quality(processed_image, processed_path, quality=SAVE_QUALITY)
-            if save_success:
-                bos_uploaded = upload_file_to_bos(processed_path)
+            # if save_success:
+            #     bos_uploaded = upload_file_to_bos(processed_path)
         else:
             # 透明背景，保存为指定格式
             if output_format == "webp":
@@ -3029,8 +3029,8 @@ async def rvm_remove_background(
             else:
                 # 保存为PNG格式
                 save_success = save_image_with_transparency(processed_image, processed_path)
-                if save_success:
-                    bos_uploaded = upload_file_to_bos(processed_path)
+                # if save_success:
+                #     bos_uploaded = upload_file_to_bos(processed_path)
 
         if save_success:
             total_time = time.perf_counter() - t1

start_local.sh

@@ -1,15 +1,54 @@
 #!/bin/bash
 export TZ=Asia/Shanghai
-export HOSTNAME=HG
-export AUTO_INIT_GFPGAN=true
+export HOSTNAME=ME
+
+export OUTPUT_DIR=/opt/data/output
+export IMAGES_DIR=/opt/data/images
+export MODELS_PATH=/opt/data/models
+export DEEPFACE_HOME=/opt/data/models
+export FAISS_INDEX_DIR=/opt/data/faiss
+export CELEBRITY_SOURCE_DIR=/opt/data/chinese_celeb_dataset
+
+export GENDER_CONFIDENCE=1
+export UPSCALE_SIZE=2
+export AGE_CONFIDENCE=0.9999
+export DRAW_SCORE=true
+export FACE_CONFIDENCE=0.7
+
+export ENABLE_DDCOLOR=true
+export ENABLE_GFPGAN=true
+export ENABLE_REALESRGAN=true
+export ENABLE_ANIME_STYLE=true
+export ENABLE_RVM=true
+export ENABLE_REMBG=true
+export ENABLE_CLIP=false
+
+export CLEANUP_INTERVAL_HOURS=1
+export CLEANUP_AGE_HOURS=1
+
+export BEAUTY_ADJUST_GAMMA=0.8
+export BEAUTY_ADJUST_MIN=1.0
+export BEAUTY_ADJUST_MAX=9.0
+export ENABLE_ANIME_PRELOAD=true
+export ENABLE_LOGGING=true
+export BEAUTY_ADJUST_ENABLED=true
+
+export RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting
+export RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth
+export RVM_MODEL=resnet50
+
+export AUTO_INIT_GFPGAN=false
 export AUTO_INIT_DDCOLOR=false
 export AUTO_INIT_REALESRGAN=false
 export AUTO_INIT_ANIME_STYLE=false
 export AUTO_INIT_CLIP=false
-export AUTO_INIT_RVM=true
-export RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting
-export RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth
-export RVM_MODEL=resnet50
+export AUTO_INIT_RVM=false
 export AUTO_INIT_REMBG=false
+
+export ENABLE_WARMUP=true
+export REALESRGAN_MODEL=realesr-general-x4v3
+export CELEBRITY_FIND_THRESHOLD=0.87
+export FEMALE_AGE_ADJUSTMENT=4
+
 uvicorn app:app --workers 1 --loop asyncio --http httptools --host 0.0.0.0 --port 7860 --timeout-keep-alive 600