import gradio as gr
import matplotlib.pyplot as plt
import numpy as np
import os
import soundfile as sf

def create_spectrogram(audio_data, sample_rate):
    # Crea lo spettrogramma
    plt.specgram(audio_data, Fs=sample_rate)

    # Salva lo spettrogramma in un file PNG
    plt.savefig('spectrogram.png')

    # Ritorna il file PNG dello spettrogramma
    return 'spectrogram.png'

def get_audio_info(audio_file):
    # Ottieni le informazioni del file audio
    audio_info = sf.info(audio_file.name)
    
    # Crea una tabella con le informazioni del file audio
    info_table = f"""
    | Informazione | Valore |
    | --- | --- |
    | Durata | {audio_info.duration} secondi |
    | Campioni al secondo | {audio_info.samplerate} Hz |
    | Canali | {audio_info.channels} |
    | Bitrate | {audio_info.samplerate * audio_info.channels * audio_info.subtype.itemsize * 8} bit/s |
    | Estensione | {os.path.splitext(audio_file.name)[1]} |
    """
    
    return info_table

# Crea l'interfaccia Gradio
iface = gr.Interface(
    fn=[get_audio_info, create_spectrogram], 
    inputs=gr.Audio(), 
    outputs=["markdown", "image"]
)
iface.launch()