dev: check working

app.py

@@ -30,20 +30,26 @@ def reencode_audio(input_path, output_path):
     ]
     subprocess.run(command, check=True)
 
-def download_and_read_h5(file_id):
-    url = f"https://drive.google.com/uc?id={file_id}"
-    output_path = f"/tmp/{file_id}.h5"
+def download_and_read_h5(file_id, dataset_name):
+    file_path = f"/tmp/{file_id}.h5"
+    # dataset_name = "your_dataset_name" 
+
+    with h5py.File(file_path, "r") as h5_file:
+        if dataset_name in h5_file:
+            obj = h5_file[dataset_name]
+            if isinstance(obj, h5py.Dataset):
+                data = obj[:]  
+                print("Data loaded successfully")
+                return data
+            else:
+                print(f"'{dataset_name}' is a group. Inspect its contents.")
+                print(list(obj.keys())) 
+        else:
+            print(f"Dataset '{dataset_name}' not found in the file.")
 
-    gdown.download(url, output_path, quiet=False)
-
-    with h5py.File(output_path, 'r') as h5_file:
-        print("Keys:", list(h5_file.keys()))
-        dataset_name = list(h5_file.keys())[0]
-        data = h5_file[dataset_name][:]
-        print(data)
 
 file_id_1 = '1zhisRgRi2qBFX73VFhzh-Ho93MORQqVa'
-download_and_read_h5(file_id_1)
+download_and_read_h5(file_id_1, "check_1")
 
 file_id_2 = '1wIaycDFGTF3e0PpAHKk-GLnxk4cMehOU'
-download_and_read_h5(file_id_2)
+download_and_read_h5(file_id_2, "check_2")