import requests
import json
kataloog="/mnt/c/jaagup/22/12/bfm/"
fnimed=open("vastus2.txt").readlines()
koht=int(open("koht.txt").read())
while koht<len(fnimed):
  fnimi=fnimed[koht].strip()
  print(koht, fnimi)
  fsisu=open(kataloog+"tekstid_puhas/"+fnimi).read()
  r=requests.post("https://api.tartunlp.ai/bert/ner/v1", json={"text": fsisu})
  if r.status_code==200:
    obj=r.json()["result"]
    v=[]
    for lause in obj:
      for sona in lause:
         if sona["ner"]!="O":
           v.append(sona["ner"]+","+sona["word"])
    print("\n".join(v), file=open(kataloog+"vastused_failiti/v"+fnimi, "w")) 
    json.dump(obj, open(kataloog+"vastused_failiti_pikalt/j"+fnimi, "w", encoding="utf-8"), ensure_ascii=False)
    print(str(len(v)))
    koht+=1    
    print(koht, file=open("koht.txt", "w"))
  else:
    print(fnimi+" kood ", r.status_code)