diff --git a/builder/modal-builder/src/volume-builder/app.py b/builder/modal-builder/src/volume-builder/app.py index 3a7e779..3eee974 100644 --- a/builder/modal-builder/src/volume-builder/app.py +++ b/builder/modal-builder/src/volume-builder/app.py @@ -2,6 +2,7 @@ import modal from config import config import os import subprocess +from pprint import pprint stub = modal.Stub() @@ -51,6 +52,8 @@ def download_model(volume_name, download_config): status = {"status": "success"} requests.post(callback_url, json={**status, **callback_body}) + print(f"finished! sending to {callback_url}") + pprint({**status, **callback_body}) @stub.local_entrypoint() @@ -63,7 +66,11 @@ def simple_download(): except modal.exception.FunctionTimeoutError as e: status = {"status": "failed", "error_logs": f"{str(e)}", "timeout": timeout} requests.post(callback_url, json={**status, **callback_body}) + print(f"finished! sending to {callback_url}") + pprint({**status, **callback_body}) except Exception as e: status = {"status": "failed", "error_logs": str(e)} requests.post(callback_url, json={**status, **callback_body}) + print(f"finished! sending to {callback_url}") + pprint({**status, **callback_body}) diff --git a/web/src/app/(app)/api/volume-upload/route.ts b/web/src/app/(app)/api/volume-upload/route.ts index f7cb3cb..25eb2d4 100644 --- a/web/src/app/(app)/api/volume-upload/route.ts +++ b/web/src/app/(app)/api/volume-upload/route.ts @@ -19,6 +19,7 @@ export async function POST(request: Request) { if (!data || error) return error; const { checkpoint_id, error_log, status, folder_path } = data; + console.log( checkpoint_id, error_log, status, folder_path ) if (status === "success") { await db diff --git a/web/src/components/CheckpointList.tsx b/web/src/components/CheckpointList.tsx index 48a2c81..01afbf7 100644 --- a/web/src/components/CheckpointList.tsx +++ b/web/src/components/CheckpointList.tsx @@ -89,7 +89,7 @@ export const columns: ColumnDef[] = [ {checkpoint.is_public ? ( - Public + Public ) : ( Private )} @@ -112,7 +112,7 @@ export const columns: ColumnDef[] = [ }, cell: ({ row }) => { return ( - + {row.original.status} ); diff --git a/web/src/server/curdCheckpoint.ts b/web/src/server/curdCheckpoint.ts index 4019b7e..acbacc9 100644 --- a/web/src/server/curdCheckpoint.ts +++ b/web/src/server/curdCheckpoint.ts @@ -115,6 +115,7 @@ function getUrl(civitai_url: string) { export const addCivitaiCheckpoint = withServerPromise( async (data: z.infer) => { const { userId, orgId } = auth(); + console.log("START") console.log("1"); if (!data.civitai_url) return { error: "no civitai_url" }; @@ -221,7 +222,7 @@ async function uploadCheckpoint( volume_name: v.volume_name, volume_id: v.id, checkpoint_id: c.id, - callback_url: `${protocol}://${domain}/api/volume-updated`, + callback_url: `${protocol}://${domain}/api/volume-upload`, upload_type: "checkpoint" }), },