infer space

viewer/src/routes/bridge.ts

@@ -145,6 +145,7 @@ export const inferFields = async (row: Row): Promise<Row | null> => {
     const date = row.Date || (await inferDate(row));
     const model = row.Model || (await inferModel(row));
     const dataset = row.Dataset || (await inferDataset(row));
+    const space = row.Space || (await inferSpace(row));
     const license = row.License || (await inferLicense(row));
     if (paper) {
         row.Paper = paper;
@@ -167,6 +168,9 @@ export const inferFields = async (row: Row): Promise<Row | null> => {
     if (dataset) {
         row.Dataset = dataset;
     }
+    if (space) {
+        row.Space = space;
+    }
     if (license) {
         row.License = license;
     }
@@ -466,6 +470,39 @@ const inferDataset = async (row: Row) => {
     return data.dataset;
 };
 
+const inferSpace = async (row: Row) => {
+    statusMessage.append(`<br><span>Inferring space...</span>`);
+    let data;
+    try {
+        const response = await fetch(baseURL + "/infer-space", {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: "Bearer " + import.meta.env.VITE_HF_TOKEN,
+            },
+            body: JSON.stringify(row),
+        });
+        data = await response.json();
+    } catch (e) {
+        statusMessage.remove("<br><span>Inferring space...</span>");
+        statusMessage.append(`<br><span style="color: red;">Error: Failed to infer space</span>`);
+        return null;
+    }
+    statusMessage.remove("<br><span>Inferring space...</span>");
+    if (data.error || !data.space) {
+        if (data.error) {
+            statusMessage.append(`<br><span style="color: red;">Error: ${data.error}</span>`);
+        } else if (data.message) {
+            statusMessage.append(`<br><span>${data.message}</span>`);
+        } else {
+            statusMessage.append(`<br><span style="color: red;">Error: Failed to infer space</span>`);
+        }
+        return null;
+    }
+    statusMessage.append(`<br><span style="color: green;">Inferred space: ${data.space}</span>`);
+    return data.space;
+};
+
 const inferLicense = async (row: Row) => {
     statusMessage.append(`<br><span>Inferring license...</span>`);
     let data;