Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 import urllib.request
 from pydub import AudioSegment
 import numpy as np
+import os
 
 def function1(prompt):
     response = requests.post("https://tommy24-testing3.hf.space/run/predict", json={
@@ -17,9 +18,22 @@ def function1(prompt):
         ]
     }).json()
     data = response["data"][0]["name"]
-    data = "https://matthijs-speecht5-tts-demo.hf.space/file="+data
-    file_name, headers = urllib.request.urlretrieve(data, "speech.mp3")
-    return data
-
-iface = gr.Interface(fn=function1, inputs="text", outputs="text")
-iface.launch()
+    data = "https://matthijs-speecht5-tts-demo.hf.space/file=" + data
+    
+    # Download the audio file and save it in WAV format
+    file_name, headers = urllib.request.urlretrieve(data, "speech.wav")
+    
+    # Load the audio file using pydub
+    sound = AudioSegment.from_wav("speech.wav")
+    
+    # Set the path and file name for the converted MP3 file
+    mp3_file = "speech.mp3"
+    
+    # Export the audio in MP3 format
+    sound.export(mp3_file, format="mp3")
+    
+    # Delete the WAV file
+    os.remove("speech.wav")
+    
+    # Return the path to the MP3 file
+    return os.path.abspath(mp3_file)