Once the code is generated, the user can run this code on Google Colab. In order to do that, first, the user needs to "download code", which downloads the code in a Jupiter notebook format. Then the user needs to use "upload code" button to upload this file to Google Colab. See screenshot of the two buttons below
After that, Google Colab environment opens and the user can run the code. This process is overly complicated and should be simplified. Instead of "download code" and "upload code", there can be one button "Run on Google Colab" that takes care of uploading the generated code to Colab.