import gradio as gr
from deep_translator import GoogleTranslator

def translate_text(text, direction):
    if direction == "Hindi to English":
        translator = GoogleTranslator(source='hi', target='en')
    else:  # English to Hindi
        translator = GoogleTranslator(source='en', target='hi')
    
    translated = translator.translate(text)
    return translated

# Define the Gradio interface
iface = gr.Interface(
    fn=translate_text,
    inputs=[
        gr.Textbox(label="Enter text to translate"),
        gr.Radio(["Hindi to English", "English to Hindi"], label="Translation Direction")
    ],
    outputs=gr.Textbox(label="Translated Text"),
    title="Hindi-English Translator",
    description="Translate text between Hindi and English"
)

# Launch the interface
iface.launch()