from microbit import *
import music

#transkriptimi
muzika = [{"A":'E4:4', "G":'G4:4', "U":'B4:4', "C":'D5:4'}, {"A":'E5:4', "G":'G5:4', "U":'B5:4', "C":'D:4'}]
#translatimi
muzikaL = [{"A":'E4:8', "G":'G4:8', "U":'B4:8', "C":'D5:8'}, {"A":'E5:8', "G":'G5:8', "U":'B5:8', "C":'D:8'}]

komp = {"T":"A", "C":"G", "A":"U", "G":"C"}

tab = {"U" :
         {"U": {
            "U":"Fenilalanina",
            "C":"Fenilalaanina",
            "A":"Leucina",
            "G":"Leucina"
          },
          "C": {
            "U":"Serina",
            "C":"Serina",
            "A":"Serina",
            "G":"Serina"
          },
          "A": {
            "U":"Tirozina",
            "C":"Tirozina",
            "A":"STOP",
            "G":"STOP"
          },
          "G": {
            "U":"Cisteina",
            "C":"Cisteina",
            "A":"STOP",
            "G":"Triptofan"
          }
          },
       "C" :
         {"U": {
            "U":"Leucina",
            "C":"Leucina",
            "A":"Leucina",
            "G":"Leucina"
          },
          "C": {
            "U":"Prolina",
            "C":"Prolina",
            "A":"Prolina",
            "G":"Prolina"
          },
          "A": {
            "U":"Histidina",
            "C":"Histidina",
            "A":"Glutamina",
            "G":"Glutamina"
          },
          "G": {
            "U":"Arginina",
            "C":"Arginina",
            "A":"Arginina",
            "G":"Arginina"
          }
          },
       "A":
         {"U": {
            "U":"Izoleucina",
            "C":"Izoleucina",
            "A":"Izoleucina",
            "G":"Metionina"
          },
          "C": {
            "U":"Treonina",
            "C":"Treonina",
            "A":"Treonina",
            "G":"Treonina"
          },
          "A": {
            "U":"Asparagina",
            "C":"Asparagina",
            "A":"Lizina",
            "G":"Lizina"
          },
          "G": {
            "U":"Serina",
            "C":"Serina",
            "A":"Arginina",
            "G":"Arginina"
          }
          },
       "G":
         {"U": {
            "U":"Valina",
            "C":"Valina",
            "A":"Valina",
            "G":"Valina"
          },
          "C": {
            "U":"Alanina",
            "C":"Alanina",
            "A":"Alanina",
            "G":"Alanina"
          },
          "A": {
            "U":"Acid aspartik",
            "C":"Acid aspartik",
            "A":"Acid glutamik",
            "G":"Acid glutamik"
          },
          "G": {
            "U":"Glicina",
            "C":"Glicina",
            "A":"Glicina",
            "G":"Glicina"
          }
         }}


def transkripto(ADN):
    ARN = [komp[baze] for baze in ADN]
    return ARN

def translato(ARN):
    i = 0
    aminoacid = [None, None, None]
    for x in ARN:
        aminoacid[i] = x
        i += 1
        if i == 3:
            i = 0
            aminoacidi = tab[aminoacid[0]][aminoacid[1]][aminoacid[2]]
            pause = len(aminoacidi) * (5/6) * 400
            display.scroll(aminoacidi, wait = False)
            music.play(muzikaL[0][aminoacid[0]])
            music.play(muzikaL[1][aminoacid[0]])
            music.play(muzikaL[0][aminoacid[1]])
            music.play(muzikaL[1][aminoacid[1]])
            music.play(muzikaL[0][aminoacid[2]])
            music.play(muzikaL[1][aminoacid[2]])
            sleep(pause)

bazat = ["T", "C", "A", "G"]
ARN = []
ADN = []
i = 0

while True:
    A = 'A'
    G = 'G'
    C = 'C'
    T = 'T'

    ADN = [A,A,G, A,G,A, G,G,G, G,G,C, T,G,A, T,T,G, G,T,G, G,T,C, T,T,C, T,A,C, T,G,T, T,G,G, T,A,C, T,G,T, T,G,G]
    ARN = transkripto(ADN)
    ADN = []
    for x in ARN:
        display.scroll(x, wait=False)
        music.play(muzika[0][x])
        music.play(muzika[1][x])
    display.scroll("->")
    translato(ARN)
    i = 0
    ARN = []