Colabs cell to upload a given structure within `output_dir` to Michelanglo

colabs_mike.ipynb

#@title Upload to Michelanglo (optional)
#@markdown [Michelanglo](https://michelanglo.sgc.ox.ac.uk/) is a website that
#@markdown allows the creation, annotation and sharing of a webpage with an interactive protein viewport.
#@markdown ([examples](https://michelanglo.sgc.ox.ac.uk/gallery)).
#@markdown The created pages are private —they have a 1 in a quintillion change to be guessed within 5 tries.

#@markdown Registered users (optional) can add interactive annotations to pages.
#@markdown A page created by a guest is editable by registered users with the URL to it
#@markdown (this can be altered in the page settings).
#@markdown Leave blank for guest:

username = ''  #@param {type:"string"}
password = ''  #@param {type:"string"}

#@markdown Choose initial model.

chosen_rank = 1  #@param {type:"integer"}

import os
os.system(f'pip3 install michelanglo-api')
import site
site.main()
from michelanglo_api import MikeAPI
if not username:
  mike = MikeAPI.guest_login()
else:
  mike = MikeAPI(username, password)

filenames = [for filename in os.listdir('output_dir') if f'rank_{chosen_rank}' filename]
if len(filenames) == 0:
     raise ValueError(f'Could not find rank {chosen_rank} in {output_dir}')
elif any(['_relaxed' in filename for filename in filenames]): # relaxed is better!
     filename = [filename for filename in filenames if '_relaxed' in filename ][0]
else:
     filename = filenames[0]
     
page = mike.convert_pdb(filename=filename)
page.show_link()