| 30 | from tasks import create_comfyui_project |
| 31 | |
| 32 | def celery_init_app(app: Flask) -> Celery: |
| 33 | class FlaskTask(Task): |
| 34 | def __call__(self, *args: object, **kwargs: object) -> object: |
| 35 | with app.app_context(): |
| 36 | return self.run(*args, **kwargs) |
| 37 | |
| 38 | celery_app = Celery(app.name, task_cls=FlaskTask) |
| 39 | celery_app.config_from_object(app.config["CELERY"]) |
| 40 | celery_app.set_default() |
| 41 | app.extensions["celery"] = celery_app |
| 42 | return celery_app |
| 43 | |
| 44 | CW_ENDPOINT = os.environ.get("CW_ENDPOINT", "https://comfyworkflows.com") |
| 45 | |