Update app.py

app.py

@@ -91,6 +91,7 @@ import random
 import base64
 import os
 import urllib.request
+from pydub import AudioSegment
 
 def function(Textbox, Textbox2, Textbox3, Dropdown):
     target = os.environ.get("target")
@@ -216,10 +217,11 @@ def function(Textbox, Textbox2, Textbox3, Dropdown):
                         }
                     )
                     output = output["audio_out"]
-                    mp3_data = BytesIO()
-                    output.export(mp3_data, format="mp3")
-                    mp3_data.seek(0)
-                    return mp3_data.read()
+                    return output
+                    # mp3_data = BytesIO()
+                    # output.export(mp3_data, format="mp3")
+                    # mp3_data.seek(0)
+                    # return mp3_data.read()
                 except Exception as e:
                     print(traceback.format_exc())
                     return "Please Wait!"
@@ -234,8 +236,13 @@ inputs = [
 
 ]
 
+def audio_format(data):
+    audio = AudioSegment.from_file(BytesIO(data), format="mp3")
+    return audio
+
 # outputs = gr.outputs.Image(type="numpy")
-outputs = gr.outputs.Audio(type="audio")
+outputs = gr.outputs.Textbox() if Dropdown == "Option1" else gr.outputs.Audio(type="bytes", label="Audio Output", format="mp3")
+
 
 iface = gr.Interface(fn=function, inputs=inputs, outputs=outputs)