import csv
import pandas as pd
 
#ładowanie danych
# dane 0
dane_in_0 = []
with open('dev-0/in.tsv') as data:
    for idx, line in enumerate(data.readlines()):
        dane_in_0.append(line.replace('\n', ''))

# dane 1
dane_in_1 = []
with open('dev-1/in.tsv') as data:
    for idx, line in enumerate(data.readlines()):
        dane_in_1.append(line.replace('\n', ''))

# dane A
dane_in_A = []
with open('dev-1/in.tsv') as data:
    for idx, line in enumerate(data.readlines()):
        dane_in_A.append(line.replace('\n', ''))


#słowa damskie i męskie
m = ['samochod', 'bmw', 'porshe', 'mercedes' ,'mecz', 'piłka', 'gol', 'windows', 'linux', 'serwer', 'ubuntu', 'żona', 'dziewczyna', 'siłownia', 'klata' , 'biceps', 'dzik', 'kumpel', 'ziom', 'gry', 'lol', 'cs', 'sport', 'piwo', 'wódka', 'komputer', 'sport', 'dres']
w = ['sukienka', 'kwiaty', 'paznokcie', 'koleżanka', 'koleżanki', 'przyjaciółka', 'przyjaciółki', 'włosy', 'lakier', 'hybryda', 'ginekolog', 'okres', 'torebka', 'torebki','kosmetycznka', 'makijaż', 'ciasto', 'mąż', 'lekarz', 'ciąża', 'krwawienie', 'spódniczka', 'fitness', 'ciuchy']

#stemowanie i normalizacja damskich i męskich wyrazów do 6 znaku
m = [x[:6].lower() for x in m]
w = [x[:6].lower() for x in w]

def decyzja(dane_in, m, w):
    res = []
    for line in dane_in:
        m_words = 0
        w_words = 0
        t = str(line).lower().split()
        for i in t:
            if i in m:
                m_words += 1
            elif i in w:
                w_words += 1
        if m_words >= w_words:
            res.append(1)
        else:
            res.append(0)
    return res

#print("dev-0")
tab_resul_0 = pd.DataFrame(decyzja(dane_in_0, m, w))
tab_resul_0.to_csv('dev-0/out.tsv', sep = '\t')

#print("dev-1")
tab_resul_1 = pd.DataFrame(decyzja(dane_in_1, m, w))
tab_resul_1.to_csv('dev-1/out.tsv', sep = '\t')

#print("test-A")
tab_resul_A = pd.DataFrame(decyzja(dane_in_A, m, w))
tab_resul_A.to_csv('test-A/out.tsv', sep = '\t')