add license

viewer/src/routes/+page.svelte

@@ -195,6 +195,24 @@
             },
             parseHTML: true,
         },
+        {
+            key: "License",
+            title: "License",
+            value: (v: Row) => v.License || "",
+            sortable: true,
+            searchValue: (v: Row) => v.License || "",
+            renderValue: (v: Row) => {
+                if (!v.License) return "";
+
+                let displayLicense = v.License;
+                if (v.License.length > 24) {
+                    displayLicense = v.License.slice(0, 20) + "...";
+                }
+
+                return `<span title="${v.License}">${displayLicense}</span>`;
+            },
+            parseHTML: true,
+        },
         {
             key: "Space",
             title: "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 license = row.License || (await inferLicense(row));
     if (paper) {
         row.Paper = paper;
     }
@@ -166,6 +167,9 @@ export const inferFields = async (row: Row): Promise<Row | null> => {
     if (dataset) {
         row.Dataset = dataset;
     }
+    if (license) {
+        row.License = license;
+    }
     statusMessage.append(`<br><span style="color: green;">Finished inferring fields for <b>${row.Name}</b></span>`);
     return row;
 };
@@ -461,3 +465,36 @@ const inferDataset = async (row: Row) => {
     statusMessage.append(`<br><span style="color: green;">Inferred dataset: ${data.dataset}</span>`);
     return data.dataset;
 };
+
+const inferLicense = async (row: Row) => {
+    statusMessage.append(`<br><span>Inferring license...</span>`);
+    let data;
+    try {
+        const response = await fetch(baseURL + "/infer-license", {
+            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 license...</span>");
+        statusMessage.append(`<br><span style="color: red;">Error: Failed to infer license</span>`);
+        return null;
+    }
+    statusMessage.remove("<br><span>Inferring license...</span>");
+    if (data.error || !data.license) {
+        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 license</span>`);
+        }
+        return null;
+    }
+    statusMessage.append(`<br><span style="color: green;">Inferred license: ${data.license}</span>`);
+    return data.license;
+};