add desc

app.py

@@ -10,7 +10,7 @@ def extract_collection_id(input_text):
         return url_match[1]
 
     # Check if input is already in the correct format
-    return input_text if re.match(r"^[\w-]+/[\w-]+$", input_text) else None
+    return input_text if re.match(r"^[\w-]+/[\w-]+", input_text) else None
 
 
 def load_collection():
@@ -35,14 +35,12 @@ def load_collection():
 
 def display_dataset(dataset_ids, index):
     dataset_id = dataset_ids[index]
-    return gr.HTML(
-        f"""<iframe
+    return gr.HTML(f"""<iframe
     src="https://huggingface.co/datasets/{dataset_id}/embed/viewer"
     frameborder="0"
     width="100%"
     height="560px"
-></iframe>"""
-    )
+></iframe>""")
 
 
 def navigate_dataset(dataset_ids, index, direction):
@@ -53,11 +51,31 @@ def navigate_dataset(dataset_ids, index, direction):
     )
 
 
+def get_display_name(collection_id):
+    # Strip out the extra ID part of the name
+    return re.sub(r"-[a-f0-9]{32}$", "", collection_id)
+
+
 try:
     dataset_ids, collection_id = load_collection()
+    display_name = get_display_name(collection_id)
 
     with gr.Blocks() as demo:
-        gr.Markdown(f"<h1>Dataset Viewer for Collection: {collection_id}</h1>")
+        gr.Markdown(f"<h1>Dataset Viewer for Collection: {display_name}</h1>")
+        gr.Markdown(
+            f"[View full collection on Hugging Face](https://huggingface.co/collections/{collection_id})"
+        )
+
+        gr.Markdown("""
+        This app allows you to browse and view datasets from a specific Hugging Face collection. 
+        Use the 'Previous' and 'Next' buttons to navigate through the datasets in the collection.
+
+        **Note**: This space is currently set up to display datasets from a specific collection. 
+        If you'd like to use it for a different collection:
+        1. Duplicate this space
+        2. In your duplicated space, set the `COLLECTION_SLUG_OR_URL` environment variable to your desired collection ID or URL
+        3. Your new space will then display datasets from your chosen collection!
+        """)
 
         index_state = gr.State(value=0)