Update app.py

app.py

@@ -1,13 +1,12 @@
 import gradio as gr
 import os
+import json
 from acrcloud.recognizer import ACRCloudRecognizer
-import tempfile
-import shutil
 
 # Retrieve ACRCloud credentials from environment variables
 acr_access_key = os.environ.get('ACR_ACCESS_KEY')
 acr_access_secret = os.environ.get('ACR_ACCESS_SECRET')
-acr_host = 'identify-ap-southeast-1.acrcloud.com' # os.environ.get('ACR_HOST', 'eu-west-1.api.acrcloud.com')
+acr_host = 'identify-ap-southeast-1.acrcloud.com'  # Default host, can be adjusted
 
 # ACRCloud recognizer configuration
 config = {
@@ -37,34 +36,37 @@ def identify_audio(file):
     # Full recognition result
     full_result = acr.recognize_by_file(file_path, 0)
 
+    # Parse full_result from string to JSON
+    full_result_parsed = json.loads(full_result)
+
+    # Extract desired details from parsed result
+    if 'metadata' in full_result_parsed and 'music' in full_result_parsed['metadata']:
+        music_data = full_result_parsed['metadata']['music'][0]
+        title = music_data.get('title', 'Unknown Title')
+        artists = ', '.join([artist['name'] for artist in music_data.get('artists', [])])
+        album = music_data.get('album', {}).get('name', 'Unknown Album')
+        release_date = music_data.get('release_date', 'Unknown Release Date')
+
+        full_result_text = f"Title: {title}\nArtists: {artists}\nAlbum: {album}\nRelease Date: {release_date}"
+    else:
+        full_result_text = "No music metadata found in the result."
+
     # Recognize using file buffer
     with open(file_path, 'rb') as f:
         buf = f.read()
         buffer_result = acr.recognize_by_filebuffer(buf, 0)
 
-    # Parse and structure the results in a more readable format
-    partial_results = "\n".join(results)
-    full_result_parsed = f"Title: {full_result['metadata']['music'][0]['title']}\n" \
-                         f"Artist: {full_result['metadata']['music'][0]['artists'][0]['name']}\n" \
-                         f"Album: {full_result['metadata']['music'][0]['album']['name']}\n" \
-                         f"Release Date: {full_result['metadata']['music'][0]['release_date']}\n" \
-                         f"Duration: {full_result['metadata']['music'][0]['duration_ms'] // 1000}s"
-
-    buffer_result_parsed = f"Track ID: {buffer_result['metadata']['music'][0]['external_metadata']['spotify']['track']['id']}\n" \
-                           f"Track Name: {buffer_result['metadata']['music'][0]['external_metadata']['spotify']['track']['name']}\n" \
-                           f"Artist: {buffer_result['metadata']['music'][0]['external_metadata']['spotify']['track']['artists'][0]['name']}\n"
-
     return {
-        "Partial Results": partial_results,
-        "Full Result": full_result_parsed,
-        "Buffer Result": buffer_result_parsed
+        "Partial Results": results,
+        "Full Result": full_result_text,
+        "Buffer Result": buffer_result
     }
 
 # Create Gradio interface
 iface = gr.Interface(
     fn=identify_audio,
     inputs=gr.File(label="Upload Audio File"),
-    outputs=[gr.Textbox(label="Partial Results"), gr.Textbox(label="Full Result"), gr.Textbox(label="Buffer Result")],
+    outputs=gr.JSON(label="Audio Metadata"),
     title="Audio Search by File",
     description="Upload an audio file to identify it using ACRCloud."
 )