dev: check working

app.py

@@ -1,4 +1,8 @@
 from fastapi import FastAPI
+import os
+import subprocess
+import gdown
+import h5py
 
 app = FastAPI()
 
@@ -6,10 +10,6 @@ app = FastAPI()
 def greet_json():
     return {"Hello": "World!"}
 
-#-----------------------------------------------------------------------------------------
-
-import os
-
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
 os.environ["FONTCONFIG_PATH"] = "/tmp/fontconfig"
@@ -24,41 +24,26 @@ bundle = WAV2VEC2_BASE
 model = bundle.get_model()
 print("Model downloaded successfully!")
 
-#-----------------------------------------------------------------------------------------
-import subprocess
-
 def reencode_audio(input_path, output_path):
     command = [
         'ffmpeg', '-i', input_path, '-acodec', 'pcm_s16le', '-ar', '16000', '-ac', '1', output_path
     ]
     subprocess.run(command, check=True)
 
-#-----------------------------------------------------------------------------------------
-
-import gdown
-import h5py
-import io
-
-file_id = '1zhisRgRi2qBFX73VFhzh-Ho93MORQqVa'
-url = f"https://drive.google.com/uc?id={file_id}"
-
-response = gdown.download(url, None, quiet=False)
-
-with h5py.File(io.BytesIO(response.content), 'r') as h5_file:
-    print("Keys:", list(h5_file.keys()))
-    dataset_name = list(h5_file.keys())[0]
-    data = h5_file[dataset_name][:]
-    print(data)
+def download_and_read_h5(file_id):
+    url = f"https://drive.google.com/uc?id={file_id}"
+    output_path = f"/tmp/{file_id}.h5"
 
-file_id = '1wIaycDFGTF3e0PpAHKk-GLnxk4cMehOU'
-url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, output_path, quiet=False)
 
-response2 = gdown.download(url, None, 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)
 
-with h5py.File(io.BytesIO(response2.content), '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)
 
-#-----------------------------------------------------------------------------------------
+file_id_2 = '1wIaycDFGTF3e0PpAHKk-GLnxk4cMehOU'
+download_and_read_h5(file_id_2)