How to edit and save text files (.py) in Google Colab?

I cloned a github repo using !git clone https://github.com/llSourcell/Pokemon_GAN.git. I wanted to modify a .py file inside Colab. So i used %load filename.py as suggested here (How to load/edit/run/save text files (.py) into an IPython notebook cell?). But whenever i run this command, i get disconnected after some time. I was wondering if there is some other way to edit .py file without undergoing the hassle of downloading it to pc,editing and then re uploading it. Thanks in advance.

You can use Ipython magic commands. Use below command

%pycat code.py

A pop up will appear displaying the code. You can copy it and edit it locally.
Remove the file using below command

!rm code.py

Copy the edited code to a cell in notebook and add below command at the top of the cell

%%writefile code.py

Run the cell. A file will be created with the contents present in the cell.

Updates: Now there are lot more easy and convenient options.

1. In the files section, there is an option to upload files or you can double click on the file, make changes and ctrl+s to save those changes.
2. You can also use https://github.com/abhishekkrthakur/colabcode to edit using visual studio code server.

Colab includes a text editor you can use to create, open, and delete .py files directly.

All is done in the Files view (see below).

• To create or delete a file, right click and choose "New file" or "Delete file".
• To edit a file, double click on it. It appears on the right portion of your screen. Make any changes you wish. Changes are saved automatically.

Unfortunately, it seems, colab do not support %load line magic (yet), and yet, you can see the file content using !cat your_file.py and then manually, copy the output contents, write them to a new cell and write %%writefile your_new_file_name.py at the top of the new cell to save this back to the instance. Note that, this will not be saved to your google drive yet.

Example:
!ls
output: colabData/

%%writefile something.py
print("everything's fine.")

!ls
output: colabData/ something.py

%run something.py
output: everything's fine.