from transformers import MarianMTModel, MarianTokenizer

def translate_text(input_text, source_lang="en", target_lang="hi"):
    # Load pre-trained model and tokenizer for MarianMT
    model_name = f'Helsinki-NLP/opus-mt-{source_lang}-{target_lang}'
    model = MarianMTModel.from_pretrained(model_name)
    tokenizer = MarianTokenizer.from_pretrained(model_name)

    # Tokenize the input text
    input_ids = tokenizer.encode(input_text, return_tensors="pt")

    # Generate translation
    translation_ids = model.generate(input_ids)

    # Decode the translated text
    translated_text = tokenizer.decode(translation_ids[0], skip_special_tokens=True)

    return translated_text

if __name__ == "__main__":
    # Example usage
    input_text = "Hello, how are you?"
    translated_text = translate_text(input_text)
    
    print(f"Input Text: {input_text}")
    print(f"Translated Text: {translated_text}")