Spaces:
Running
Running
| import gradio as gr | |
| import os | |
| import json | |
| from acrcloud.recognizer import ACRCloudRecognizer | |
| # 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' # Default host, can be adjusted | |
| # ACRCloud recognizer configuration | |
| config = { | |
| 'host': acr_host, | |
| 'access_key': acr_access_key, | |
| 'access_secret': acr_access_secret, | |
| 'timeout': 10 # seconds | |
| } | |
| # Initialize ACRCloud recognizer | |
| acr = ACRCloudRecognizer(config) | |
| def identify_audio(file): | |
| # Gradio provides a file object, and file.name contains the path | |
| file_path = file.name # Gradio file object already provides a file path | |
| # Get the duration of the audio file in milliseconds | |
| duration_ms = int(acr.get_duration_ms_by_file(file_path)) | |
| results = [] | |
| # Process audio in 10-second chunks | |
| for i in range(0, duration_ms // 1000, 10): | |
| res = acr.recognize_by_file(file_path, i, 10) | |
| results.append(f"Time {i}s: {res.strip()}") | |
| # 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) | |
| return { | |
| "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.JSON(label="Audio Metadata"), | |
| title="Audio Search by File", | |
| description="Upload an audio file to identify it using ACRCloud." | |
| ) | |
| # Launch the Gradio interface | |
| iface.launch() | |