对齐预发环境

src/BowongModalFunctions/api.py

@@ -78,8 +78,7 @@ async def scalar():
 web_app.include_router(ffmpeg.router)
 web_app.include_router(cache.router)
 
-# todo: prod 展示去掉Comfy相关API接口
-# web_app.include_router(comfyui.router)
+web_app.include_router(comfyui.router)
 
 web_app.include_router(google.router)
 web_app.include_router(task.router)

src/BowongModalFunctions/router/google.py

@@ -33,7 +33,7 @@ async def upload_file_multipart(file: UploadFile,
                             'video/webm', 'video/wmv', 'video/3gpp']:
         raise HTTPException(status_code=status.HTTP_415_UNSUPPORTED_MEDIA_TYPE)
     logger.info(f"Uploading name = {file.filename}, size = {content_length}, type = {content_type} to google file")
-    with httpx.Client() as client:
+    with httpx.Client(timeout=1800) as client:
         pre_upload_response = client.post(
             url=f"https://generativelanguage.googleapis.com/upload/v1beta/files?key={google_api_key}",
             headers={

src/cluster/app.py

@@ -16,8 +16,5 @@ app = modal.App(config.modal_app_name,
 app.include(media_app)
 app.include(ffmpeg_app)
 app.include(web_app)
-
-# todo: prod 暂时去掉Comfy相关接口
-
-# app.include(comfyui_v1_app)
-# app.include(comfyui_v2_app)
+app.include(comfyui_v1_app)
+app.include(comfyui_v2_app)