stub/replicate.py

# stub/replicate.py
def run(model_path, input):
    print("Stub called for replicate.run with model_path and input")

    prompt = input["prompt"]    
    if "Introduction".lower() in prompt.lower():
        url = "https://replicate.delivery/pbxt/sLBtHnGDMVK7AV5x24dl29lp9pQnbsfcuMbusXcJEl9kG8rIA/out.mp4"
    elif "Development".lower() in prompt.lower():
        url = "https://replicate.delivery/pbxt/QVgesepHS7pvZE9UWs2SPDMsEeCMrteelrmuegvvr7iITDerIA/out.mp4"
    elif "Climax".lower() in prompt.lower():
        url = "https://replicate.delivery/pbxt/H1bJ3dp0r95OM5XPXoK6gfABF9vOCsFT7gxH0I4ceg75N4XRA/out.mp4"
    elif "Resolution".lower() in prompt.lower():
        url = "https://replicate.delivery/pbxt/qNdKneAbNaRtdK6pZnoAO17JCJfD5neffTw193F1XXUkvBfVE/out.mp4"
    else:
        url = "https://replicate.delivery/pbxt/cgT0Aef4haodP04HybKaOrsHOQKkYcV8mpzGj7WHx3eFMuviA/out.mp4"
    return url