Deploy (see actual commits on https://github.com/mlcommons/croissant).

core/state.py

@@ -183,8 +183,11 @@ class Metadata:
     name: str = ""
     description: str | None = None
     citation: str | None = None
+    conforms_to: str | None = None
+    creators: list[mlc.PersonOrOrganization] = dataclasses.field(default_factory=list)
     data_biases: str | None = None
     data_collection: str | None = None
+    date_published: datetime.datetime | None = None
     license: str | None = ""
     personal_sensitive_information: str | None = None
     url: str = ""

events/metadata.py

@@ -1,9 +1,11 @@
+import datetime
 import enum
 
 import streamlit as st
 
 from core.names import find_unique_name
 from core.state import Metadata
+import mlcroissant as mlc
 
 # List from:
 LICENSES_URL = "https://huggingface.co/docs/hub/repositories-licenses"
@@ -90,7 +92,9 @@ class MetadataEvent(enum.Enum):
     """Event that triggers a metadata change."""
 
     NAME = "NAME"
+    CONFORMS_TO = "CONFORMS_TO"
     DESCRIPTION = "DESCRIPTION"
+    DATE_PUBLISHED = "DATE_PUBLISHED"
     URL = "URL"
     LICENSE = "LICENSE"
     CITATION = "CITATION"
@@ -98,11 +102,17 @@ class MetadataEvent(enum.Enum):
     DATA_BIASES = "DATA_BIASES"
     DATA_COLLECTION = "DATA_COLLECTION"
     PERSONAL_SENSITIVE_INFORMATION = "PERSONAL_SENSITIVE_INFORMATION"
+    CREATOR_ADD = "CREATOR_ADD"
+    CREATOR_NAME = "CREATOR_NAME"
+    CREATOR_URL = "CREATOR_URL"
+    CREATOR_REMOVE = "CREATOR_REMOVE"
 
 
 def handle_metadata_change(event: MetadataEvent, metadata: Metadata, key: str):
     if event == MetadataEvent.NAME:
         metadata.name = find_unique_name(set(), st.session_state[key])
+    if event == MetadataEvent.CONFORMS_TO:
+        metadata.conforms_to = st.session_state[key]
     elif event == MetadataEvent.DESCRIPTION:
         metadata.description = st.session_state[key]
     elif event == MetadataEvent.LICENSE:
@@ -119,3 +129,20 @@ def handle_metadata_change(event: MetadataEvent, metadata: Metadata, key: str):
         metadata.data_collection = st.session_state[key]
     elif event == MetadataEvent.PERSONAL_SENSITIVE_INFORMATION:
         metadata.personal_sensitive_information = st.session_state[key]
+    elif event == MetadataEvent.DATE_PUBLISHED:
+        date = st.session_state[key]
+        metadata.date_published = datetime.datetime(date.year, date.month, date.day)
+    elif event == MetadataEvent.CREATOR_ADD:
+        metadata.creators = [mlc.PersonOrOrganization()]
+    elif event == MetadataEvent.CREATOR_REMOVE:
+        metadata.creators = []
+    elif event == MetadataEvent.CREATOR_NAME:
+        if metadata.creators:
+            metadata.creators[0].name = st.session_state[key]
+        else:
+            metadata.creators = [mlc.PersonOrOrganization(name=st.session_state[key])]
+    elif event == MetadataEvent.CREATOR_URL:
+        if metadata.creators:
+            metadata.creators[0].url = st.session_state[key]
+        else:
+            metadata.creators = [mlc.PersonOrOrganization(url=st.session_state[key])]

views/metadata.py

@@ -106,3 +106,48 @@ def _render_generic_metadata(metadata: Metadata):
         on_change=handle_metadata_change,
         args=(MetadataEvent.CITATION, metadata, key),
     )
+    key = "metadata-date-published"
+    st.date_input(
+        label="Date of first broadcast/publication.",
+        key=key,
+        value=metadata.date_published,
+        on_change=handle_metadata_change,
+        args=(MetadataEvent.DATE_PUBLISHED, metadata, key),
+    )
+    if metadata.creators:
+        creator = metadata.creators[0]
+        col1, col2, col3 = st.columns([1, 1, 1])
+        key = "metadata-creator-name"
+        col1.text_input(
+            label="Creator name",
+            key=key,
+            value=creator.name,
+            on_change=handle_metadata_change,
+            placeholder="A person or an organization",
+            args=(MetadataEvent.CREATOR_NAME, metadata, key),
+        )
+        key = "metadata-creator-url"
+        col2.text_input(
+            label="Creator URL",
+            key=key,
+            value=creator.url,
+            placeholder="https://mlcommons.org",
+            on_change=handle_metadata_change,
+            args=(MetadataEvent.CREATOR_URL, metadata, key),
+        )
+        key = "metadata-creator-remove"
+        col3.button(
+            "✖️",
+            key=key,
+            help="Remove the creator",
+            on_click=handle_metadata_change,
+            args=(MetadataEvent.CREATOR_REMOVE, metadata, key),
+        )
+    else:
+        key = "metadata-add-creator"
+        st.button(
+            label="✚ Add a creator",
+            key=key,
+            on_click=handle_metadata_change,
+            args=(MetadataEvent.CREATOR_ADD, metadata, key),
+        )