Update app.py

app.py

@@ -1,39 +1,40 @@
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+import os
+import gradio as gr
 from TTS.api import TTS
-import torch
 
-app = FastAPI()
+# Set the environment variable to agree to the terms of service
+os.environ["COQUI_TOS_AGREED"] = "1"
 
 # Initialize the TTS model
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2")
-device = "cuda" if torch.cuda.is_available() else "cpu"
-tts.to(device)
 
 # Get available voices and languages
 available_voices = tts.list_speakers()
 available_languages = tts.list_languages()
 
-class TextToSpeechRequest(BaseModel):
-    text: str
-    voice: str
-    language: str
+def generate_speech(text, voice, language):
+    # Generate speech
+    output_path = "output.wav"
+    tts.tts_to_file(text=text, speaker_wav=voice, language=language, file_path=output_path)
+    return output_path
 
-@app.get("/voices")
-def get_voices():
-    return {"voices": available_voices}
+# Create Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# XTTS v2 Text-to-Speech Generator")
 
-@app.get("/languages")
-def get_languages():
-    return {"languages": available_languages}
+    with gr.Row():
+        text_input = gr.Textbox(label="Input Text", placeholder="Enter text to be synthesized")
 
-@app.post("/generate")
-def generate_speech(request: TextToSpeechRequest):
-    if request.voice not in available_voices:
-        raise HTTPException(status_code=400, detail="Invalid voice selected")
-    if request.language not in available_languages:
-        raise HTTPException(status_code=400, detail="Invalid language selected")
+    with gr.Row():
+        voice_dropdown = gr.Dropdown(label="Select Voice", choices=available_voices)
+        language_dropdown = gr.Dropdown(label="Select Language", choices=available_languages)
 
-    output_path = "output.wav"
-    tts.tts_to_file(text=request.text, speaker=request.voice, language=request.language, file_path=output_path)
-    return {"message": "Speech generated successfully", "file_path": output_path}
+    with gr.Row():
+        generate_button = gr.Button("Generate Speech")
+
+    with gr.Row():
+        audio_output = gr.Audio(label="Generated Speech")
+
+    generate_button.click(generate_speech, inputs=[text_input, voice_dropdown, language_dropdown], outputs=audio_output)
+
+demo.launch(debug=True)