from backend.mistral7b import convert_with_mistral_7b
from backend.mistrallarge2 import convert_with_mistral_large2
from backend.codegen2 import convert_with_codegen2

def route_model_conversion(model_name, cs_code):
    if "Mistral-7B" in model_name:
        return convert_with_mistral_7b(cs_code)
    elif "Mixtral" in model_name:
        return convert_with_mistral_large2(cs_code)
    elif "codegen2" in model_name:
        return convert_with_codegen2(cs_code)
    else:
        return "Unsupported model selected."