Update app.py

app.py

@@ -1,5 +1,6 @@
 from gradio_client import Client, handle_file
 import os
+import gradio as gr
 
 # Get Hugging Face token from environment variable
 hf_token = os.getenv("your_huggingface_token")  # Make sure this is set in your environment
@@ -7,26 +8,37 @@ hf_token = os.getenv("your_huggingface_token")  # Make sure this is set in your
 # Initialize the Gradio client with the token
 client = Client("ResembleAI/Chatterbox", hf_token)
 
-# Example text input, audio prompt URL, and other parameters
-text_input = "Now let's make my mum's favourite. So three mars bars into the pan. Then we add the tuna and just stir for a bit, just let the chocolate and fish infuse. A sprinkle of olive oil and some tomato ketchup. Now smell that. Oh boy this is going to be incredible."
-audio_prompt_url = 'https://github.com/gradio-app/gradio/raw/main/test/test_files/audio_sample.wav'
-exaggeration_input = 0.5
-temperature_input = 0.8
-seed_num_input = 0
-cfgw_input = 0.5
+# Define the function to call the model and return the generated audio file
+def generate_tts_audio(text_input, audio_prompt_url, exaggeration_input, temperature_input, seed_num_input, cfgw_input):
+    try:
+        result = client.predict(
+            text_input=text_input,
+            audio_prompt_path_input=handle_file(audio_prompt_url),
+            exaggeration_input=exaggeration_input,
+            temperature_input=temperature_input,
+            seed_num_input=seed_num_input,
+            cfgw_input=cfgw_input,
+            api_name="/generate_tts_audio"
+        )
+        return result['filepath']  # Return the filepath to the generated audio
+    except Exception as e:
+        # Handle errors and display them to the user
+        return f"An error occurred: {e}"
 
-# Use try-except to handle any errors that might occur during the prediction
-try:
-    result = client.predict(
-        text_input=text_input,
-        audio_prompt_path_input=handle_file(audio_prompt_url),
-        exaggeration_input=exaggeration_input,
-        temperature_input=temperature_input,
-        seed_num_input=seed_num_input,
-        cfgw_input=cfgw_input,
-        api_name="/generate_tts_audio"
-    )
-    print(result)  # Print the result (filepath to generated audio)
-except Exception as e:
-    # Handle the error by printing the exception message
-    print(f"An error occurred: {e}")
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=generate_tts_audio,
+    inputs=[
+        gr.Textbox(label="Text to Synthesize", placeholder="Enter your text here..."),
+        gr.Textbox(label="Audio Prompt URL", placeholder="Enter the URL of the audio prompt (Optional)", default='https://github.com/gradio-app/gradio/raw/main/test/test_files/audio_sample.wav'),
+        gr.Slider(minimum=0, maximum=1, default=0.5, label="Exaggeration"),
+        gr.Slider(minimum=0, maximum=1, default=0.8, label="Temperature"),
+        gr.Number(label="Seed Number", value=0),
+        gr.Slider(minimum=0, maximum=1, default=0.5, label="CFG/Pace")
+    ],
+    outputs=gr.Audio(label="Generated Audio")
+)
+
+# Launch the Gradio interface
+if __name__ == "__main__":
+    interface.launch()