import gradio as gr
import speech_recognition as sr
from deep_translator import GoogleTranslator

# Inicializa o reconhecedor de fala
recognizer = sr.Recognizer()

# Função para transcrever e traduzir o áudio
def transcrever_e_traduzir(audio_file):
    with sr.AudioFile(audio_file) as source:
        audio = recognizer.record(source)  # Lê todo o áudio
        try:
            # Tenta transcrever o áudio para texto usando pocketsphinx (reconhecimento offline)
            text_en = recognizer.recognize_sphinx(audio)  # Modificado para pocketsphinx
            
            # Se a transcrição for bem-sucedida, traduz
            translated_text = GoogleTranslator(source='en', target='pt').translate(text_en)
            
            return f"Texto original (Inglês): {text_en}\n\nTexto traduzido (Português): {translated_text}"
        
        except sr.UnknownValueError:
            return "A transcrição não pôde ser compreendida."
        except sr.RequestError as e:
            return f"Erro ao acessar o serviço de reconhecimento de fala: {e}"

# Cria a interface com Gradio
interface = gr.Interface(
    fn=transcrever_e_traduzir,
    inputs=gr.Audio(type="filepath"),  # Input de áudio via upload
    outputs="text",  # Saída em formato de texto
    title="Transcrição e Tradução de Áudio",
    description="Faça upload de um arquivo de áudio em inglês e receba a transcrição e tradução para português."
)

# Executa a interface
interface.launch()