chore: Refactor app.py for improved readability and maintainability

app.py

@@ -1,81 +1,90 @@
 import os
+import re
+
 import gradio as gr
 from huggingface_hub import get_collection
-import re
+
 
 def extract_collection_id(input_text):
-    # Check if input is a full URL
-    url_match = re.match(r'https://huggingface\.co/collections/(.+)$', input_text)
-    if url_match:
-        return url_match.group(1)
-    
+    if url_match := re.match(r"https://huggingface\.co/collections/(.+)$", input_text):
+        return url_match[1]
+
     # Check if input is already in the correct format
-    if re.match(r'^[\w-]+/[\w-]+$', input_text):
-        return input_text
-    
-    return None
+    return input_text if re.match(r"^[\w-]+/[\w-]+$", input_text) else None
+
 
 def load_collection():
-    collection_input = os.getenv('COLLECTION_SLUG_OR_URL')
+    collection_input = os.getenv("COLLECTION_SLUG_OR_URL")
     if not collection_input:
         raise ValueError("COLLECTION_SLUG_OR_URL environment variable is not set.")
-    
+
     collection_id = extract_collection_id(collection_input)
     if not collection_id:
-        raise ValueError("Invalid collection ID or URL in COLLECTION_SLUG_OR_URL environment variable.")
-    
+        raise ValueError(
+            "Invalid collection ID or URL in COLLECTION_SLUG_OR_URL environment variable."
+        )
+
     collection = get_collection(collection_id)
-    dataset_ids = [item.item_id for item in collection.items if item.item_type == 'dataset']
-    if not dataset_ids:
+    if dataset_ids := [
+        item.item_id for item in collection.items if item.item_type == "dataset"
+    ]:
+        return dataset_ids, collection_id
+    else:
         raise ValueError("No datasets found in this collection.")
-    
-    return dataset_ids, collection_id
+
 
 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):
     new_index = (index + direction) % len(dataset_ids)
-    return new_index, f"Dataset {new_index + 1} of {len(dataset_ids)}: {dataset_ids[new_index]}"
+    return (
+        new_index,
+        f"Dataset {new_index + 1} of {len(dataset_ids)}: {dataset_ids[new_index]}",
+    )
+
 
 try:
     dataset_ids, collection_id = load_collection()
-    
+
     with gr.Blocks() as demo:
         gr.Markdown(f"<h1>Dataset Viewer for Collection: {collection_id}</h1>")
-        
+
         index_state = gr.State(value=0)
-        
+
         with gr.Row():
             left_btn = gr.Button("Previous")
             right_btn = gr.Button("Next")
-        
+
         dataset_info = gr.Markdown(f"Dataset 1 of {len(dataset_ids)}: {dataset_ids[0]}")
         iframe_output = gr.HTML()
-        
+
         left_btn.click(
             navigate_dataset,
             inputs=[gr.State(dataset_ids), index_state, gr.Number(-1, visible=False)],
-            outputs=[index_state, dataset_info]
+            outputs=[index_state, dataset_info],
         )
         right_btn.click(
             navigate_dataset,
             inputs=[gr.State(dataset_ids), index_state, gr.Number(1, visible=False)],
-            outputs=[index_state, dataset_info]
+            outputs=[index_state, dataset_info],
         )
-        
+
         index_state.change(
             display_dataset,
             inputs=[gr.State(dataset_ids), index_state],
-            outputs=[iframe_output]
+            outputs=[iframe_output],
         )
-        
+
         # Initialize the display with the first dataset
         demo.load(
             fn=lambda: display_dataset(dataset_ids, 0),
@@ -88,4 +97,6 @@ try:
 
 except Exception as e:
     print(f"Error: {str(e)}")
-    print("Please set the COLLECTION_SLUG_OR_URL environment variable with a valid collection ID or URL.")
+    print(
+        "Please set the COLLECTION_SLUG_OR_URL environment variable with a valid collection ID or URL."
+    )