Voicechat_Hindi/app.py

"""Deploying AI Voice Chatbot Gradio App."""
from gradio import Audio, Interface, Textbox
from typing import Tuple

from Voicechat_Hindi.utils import (TextGenerationPipeline, from_en_translation,
                   html_audio_autoplay, stt, to_en_translation, tts,
                   tts_to_bytesio)

max_answer_length = 100
desired_language = "hi"
response_generator_pipe = TextGenerationPipeline(max_length=max_answer_length)


def main(audio: object) -> Tuple[str, str, str, object]:
    """Calls functions for deploying gradio app.
    It responds both verbally and in text
    by taking voice input from user.
    Args:
        audio (object): recorded speech of user
    Returns:
        tuple containing
        - user_speech_text (str) : recognized speech
        - bot_response_hi (str) : translated answer of bot
        - bot_response_en (str) : bot's original answer
        - html (object) : autoplayer for bot's speech
    """
    user_speech_text = stt(audio, desired_language)
    tranlated_text = to_en_translation(user_speech_text, desired_language)
    bot_response_en = response_generator_pipe(tranlated_text)
    bot_response_hi = from_en_translation(bot_response_en, desired_language)
    bot_voice = tts(bot_response_hi, desired_language)
    bot_voice_bytes = tts_to_bytesio(bot_voice)
    html = html_audio_autoplay(bot_voice_bytes)
    return user_speech_text, bot_response_hi, bot_response_en, html


interface = Interface(
    fn=main,
    inputs=[
        Audio(
            source="microphone",
            type="filepath",
        ),
    ],
    outputs=[
        Textbox(label="You said: "),
        Textbox(label="AI: "),
        Textbox(label="AI (English): "),
        "html",
    ],
    live=True,
    allow_flagging="never",
)

if __name__ == '__main__':
    interface.launch(debug=True)