Update app.py

app.py

@@ -19,23 +19,26 @@ def process_fn(input_audio_path):
     try:
         ext = input_audio_path.split(".")[-1].lower()
 
-        detected_key = detect_key(
-            audio_dir=input_audio_path,
+        detected_keys = detect_key(
+            input_audio_path,
             extension=ext,
-            device=DEVICE
+            device=DEVICE,
+            cli=False
         )
 
-        print("Detected key raw output:", detected_key)
+        print("Raw output:", detected_keys)
 
-        if isinstance(detected_key, list):
-            detected_key = detected_key[0]
+        if not detected_keys:
+            raise ValueError("No key detected")
+
+        detected_key = detected_keys[0]  # because function returns list[str]
 
         labels = LabelList()
         labels.labels.append(
             AudioLabel(
                 t=0.0,
                 duration=5.0,
-                label=str(detected_key),
+                label=detected_key,
                 amplitude=0.8,
                 description="Detected by S-KEY",
                 color=OutputLabel.rgb_color_to_int(255, 100, 100, 0.8),
@@ -48,6 +51,7 @@ def process_fn(input_audio_path):
         print("Error:", e)
         raise e
 
+
 # Build Gradio Endpoint
 with gr.Blocks() as demo:
     input_audio = gr.Audio(label="Input Audio", type="filepath").harp_required(True)