import gradio as gr
import urllib.request
import requests
from pydub import AudioSegment

def function1(prompt):
    response = requests.post("https://matthijs-speecht5-tts-demo.hf.space/run/predict", json={
        "data": [
            prompt,
            "KSP (male)",
        ]
    }).json()
    data = response["data"][0]["name"]
    data = "https://matthijs-speecht5-tts-demo.hf.space/file="+data
    speech = urllib.request.urlretrieve(data, "speech.wav")
    audio = AudioSegment.from_wav(data).export("output.mp3", format="mp3")
    return audio

iface = gr.Interface(fn=function1, inputs="text", outputs=[gr.Audio(label="Audio")])
iface.launch()