Update app.py

app.py

@@ -9,7 +9,7 @@ async def get_splits(dataset_name: str) -> Dict[str, List[Dict]]:
         response = await session.get(URL)
         return response.json()
 
-async def get_valid_datasets() -> Dict[str, List[str]]:
+async def get_valid_datasets_old() -> Dict[str, List[str]]:
     URL = f"https://datasets-server.huggingface.co/valid"
     async with httpx.AsyncClient() as session:
         response = await session.get(URL)
@@ -18,6 +18,17 @@ async def get_valid_datasets() -> Dict[str, List[str]]:
         # The one to watch: https://huggingface.co/rungalileo
         # rungalileo/medical_transcription_40
 
+async def get_valid_datasets() -> Dict[str, List[str]]:
+    URL = f"https://datasets-server.huggingface.co/valid"
+    async with httpx.AsyncClient() as session:
+        response = await session.get(URL)
+        try:
+            datasets = response.json()["valid"]
+        except (KeyError, json.JSONDecodeError):
+            datasets = []  # Set a default value if the response is not in the expected format
+        return gr.Dropdown.update(choices=datasets, value="awacke1/ChatbotMemory.csv")
+
+
 async def get_first_rows(dataset: str, config: str, split: str) -> Dict[str, Dict[str, List[Dict]]]:
     URL = f"https://datasets-server.huggingface.co/first-rows?dataset={dataset}&config={config}&split={split}"
     async with httpx.AsyncClient() as session:
@@ -77,8 +88,9 @@ with gr.Blocks() as demo:
         URLcenter = gr.Textbox(label="Dataset URL", placeholder="URL")
         btn = gr.Button("Use Dataset")
         #URLoutput = gr.Textbox(label="Output",placeholder="URL Output")
-        URLoutput = gr.HTML(label="Output",placeholder="URL Output")
-    
+        #URLoutput = gr.HTML(label="Output",placeholder="URL Output")
+        URLoutput = gr.HTML(label="Output")
+
     with gr.Row():
         dataset = gr.DataFrame(wrap=True, interactive=True)