Update app.py

app.py

@@ -1,28 +1,25 @@
 import gradio as gr
+from datasets import list_datasets, list_files
 
-def list_subfolders(dataset_path):
-  """Lists all subfolders within the provided dataset path.
+def list_files_and_subfolders(dataset_name):
+    # List files and subfolders in the chosen dataset directory
+    files_and_subfolders = list_files(dataset_name)
 
-  Args:
-      dataset_path: The path to the dataset directory.
+    # Convert list to string for display
+    files_and_subfolders_str = "\n".join(files_and_subfolders)
 
-  Returns:
-      A list of subfolder names or an empty list if no subfolders exist.
-  """
-  try:
-    import os
-    return [f for f in os.listdir(dataset_path) if os.path.isdir(os.path.join(dataset_path, f))]
-  except FileNotFoundError:
-    return ["Dataset path not found!"]
-  except PermissionError:
-    return ["Permission error accessing path!"]
+    return files_and_subfolders_str
 
+# List all available datasets
+datasets = list_datasets()
+
+# Gradio interface
 interface = gr.Interface(
-  fn=list_subfolders,
-  inputs="text",
-  outputs="list",
-  title="List Subfolders",
-  description="Enter the path to your dataset directory and see a list of subfolders.",
+    fn=list_files_and_subfolders,
+    inputs=gr.inputs.Dropdown(choices=datasets, label="Choose Dataset"),
+    outputs="text",
+    title="List Files and Subfolders in Hugging Face Dataset",
+    description="Choose a dataset to list its files and subfolders.",
 )
 
 interface.launch()