parece uma situação bastante delicada e entendo sua preocupação. apesar de também me parecer uma grande red flag a empresa colocar essa pressão sobre você no período de experiência, sei que isso pode deixar qualquer pessoa em dúvida sobre a melhor decisão a tomar.

mas independentemente do resultado da segunda etapa, ao entregar uma solução significativa, você tem a oportunidade de deixar um legado, sem contar que o aprendizado que você adquiriu ao se dedicar na solução até a entrega dela é muito valioso! mesmo que não continue na empresa, sua reputação e credibilidade como alguém que entrega, de fato, permanece com você e pode ser um diferencial em futuras entrevistas e oportunidades...

ser um profissional que agrega valor, independente das circunstâncias, tende a ser recompensado de alguma forma, seja na empresa atual ou em novas oportunidades.

boa sorte na segunda etapa, e espero que tudo se resolva da melhor forma para você!