#start with first demarcator
#have a flag that indicates if you've hit demarcator, if flag = 0, append cell to list
#remove words followed by colon
#when you hit colon, start deleting from end of buffer until reaching space
# search for either space space space heifen space or space space space number .
# keep scanning cell until you hit a demarcator, then add everything into a bullet point

import csv

bulletFile = open("InclusionExclusionBulletPoint.tsv","w",encoding="latin-1")

with open("EligibilityCriteriaDocument.txt","r",encoding="latin-1") as tsvFile:
    tsvContents = csv.reader(tsvFile, delimiter='\t')
    goodCount = 0
    count = 0
    for row in tsvContents:
        count +=1
        #if count!= 60567:
            #continue
        inclusion = row[10]
        exclusion = row[11]
        nctNumber = row[1]
        intervention = row[4]
        condition = row[3]
        gender = row[5]
        age = row[6]
        nctNumber = str(nctNumber)
        intervention = str(intervention)
        condition = str(condition)
        gender = str(gender)
        age = str(age)            
        inclusionList = []
        exclusionList = []
        flag = 0
        buffer = ""
        inclusion = str(inclusion)
        exclusion = str(exclusion)
        for i in range(len(inclusion)):

            if inclusion[i] == ":":
                k= i-1
                while inclusion[k] != " ":
                    k -= 1
                buffer = buffer[:k]
                #print(inclusion)
                continue
                
            if i+4 < len(inclusion) and inclusion[i] == " " and inclusion[i+1] == " " and inclusion[i+2] == " " and inclusion[i+3] == "-" and inclusion[i+4] == " ":
                if flag == 1:
                    inclusionList.append(buffer.strip(" "))
                    bulletFile.write(nctNumber + "\t" + condition + "\t" + "indication" + "\t" + buffer.strip(" ") + "\t" + intervention + "\n")  
                buffer = ""
                i = i+4

                flag = 1
                continue
            
            if i+4 < len(inclusion) and inclusion[i] == " " and inclusion[i+1] == " " and inclusion[i+2] == " " and inclusion[i+3].isdigit():
                k = i+4
                
                while inclusion[k].isdigit():
                    k += 1

                if inclusion[k] == ".":
                    if flag == 1:
                        inclusionList.append(buffer.strip(" "))
                        bulletFile.write(nctNumber + "\t" + condition + "\t" + "indication" + "\t" + buffer.strip(" ") + "\t" + intervention + "\n")
                    buffer = ""
                    i = k+1
                    flag = 1
                    continue


            buffer += inclusion[i]

        if flag == 0:
            inclusionList.append(inclusion)
            bulletFile.write(nctNumber + "\t" + condition + "\t" + "indication" + "\t" + inclusion + "\t" + intervention + "\n")
            
        else:
            inclusionList.append(buffer.strip(" "))
            bulletFile.write(nctNumber + "\t" + condition + "\t" + "indication" + "\t" + buffer.strip(" ") + "\t" + intervention + "\n")
            goodCount += 1

        bulletFile.write(nctNumber + "\t" + condition + "\t" + "indication" + "\t" + gender + "\t" + intervention + "\n")
        bulletFile.write(nctNumber + "\t" + condition + "\t" + "indication" + "\t" + age + "\t" + intervention + "\n") 



        #write id number \t indication \t item \n to the file wherever you do inclusionlist.append


        flag = 0
        buffer = ""
        for i in range(len(exclusion)):
            
            if exclusion[i] == ":":
                k= i-1
                while exclusion[k] != " ":
                    k -= 1
                buffer = buffer[:k]
                #print(exclusion)
                continue
                
            if i+4 < len(exclusion) and exclusion[i] == " " and exclusion[i+1] == " " and exclusion[i+2] == " " and exclusion[i+3] == "-" and exclusion[i+4] == " ":
                if flag == 1:
                    exclusionList.append(buffer.strip(" "))
                    bulletFile.write(nctNumber + "\t" + condition + "\t" + "counterindication" + "\t" + buffer.strip(" ") + "\t" + intervention + "\n")

                buffer = ""
                i = i+4

                flag = 1
                continue
            
            if i+4 < len(exclusion) and exclusion[i] == " " and exclusion[i+1] == " " and exclusion[i+2] == " " and exclusion[i+3].isdigit():
                k = i+4
                
                while exclusion[k].isdigit():
                    k += 1

                if exclusion[k] == ".":
                    if flag == 1:
                        exclusionList.append(buffer.strip(" "))
                        bulletFile.write(nctNumber + "\t" + condition + "\t" + "counterindication" + "\t" + buffer.strip(" ") + "\t" + intervention + "\n")

                    buffer = ""
                    i = k+1
                    flag = 1
                    continue


            buffer += exclusion[i]

        if flag == 0:
            exclusionList.append(exclusion)
            bulletFile.write(nctNumber + "\t" + condition + "\t" + "counterindication" + "\t" + exclusion + "\t" + intervention + "\n")

            
        else:
            exclusionList.append(buffer.strip(" "))
            bulletFile.write(nctNumber + "\t" + condition + "\t" + "counterindication" + "\t" + buffer.strip(" ") + "\t" + intervention + "\n")
            goodCount += 1
            #copy paste, replace "inclusion" with "exclusion"

        #write id number \t indication \t item \n to the file wherever you do exclusionlist.append




        #take row[0] for id number 

        #print(inclusionList)
        #write to new file
        # id \t inclusion/exclusion \t item \n    


bulletFile.close()