from transformers import pipeline

def get_translation_pipeline():
    # Use a public model that doesn't require login
    model_name = "Salesforce/codet5-base-multi-sum"
    return pipeline("text2text-generation", model=model_name, tokenizer=model_name)

def translate_cobol_to_csharp(pipe, cobol_code: str) -> str:
    prompt = f"Translate this COBOL code to C#:\n\n{cobol_code}"
    result = pipe(prompt, max_length=512, do_sample=False)[0]["generated_text"]
    return result