from ollama import chat
from ollama import ChatResponse
import os
import json
import sys

algkaust="/home/jaagup/26/05/llama4/"
kaust=algkaust+"loodusetaerr/"
viip0=open(algkaust+"viip5.txt", "r", encoding="utf-8").read()
fnimed=os.listdir(kaust)
fnimed.sort()
numbrifail=algkaust+"mnr"+sys.argv[1]+".txt"
fnr=int(open(numbrifail).read())
while fnr<len(fnimed):
 try:
  fnimi=kaust+fnimed[fnr]
  print(fnr, fnimi)
  viip=viip0+open(fnimi, encoding="utf-8").read()
  for nr in range(2):
   try:
    response: ChatResponse = chat(model='gemma4', messages=[
     {
      'role': 'user',
      'content': viip,
     },
    ])
    v=response['message']['content']
    algus=v.find("{")
    ots=v.rfind("}")
    d=json.loads(v[algus:ots+1])
    json.dump(d, open(algkaust+"jsonvastused2/"+fnimed[fnr], "w", encoding="utf-8"), ensure_ascii=False, indent=2)
    break
   except:
    pass
 except Exception as ex:
   print(ex)
 fnr+=1
 with open(numbrifail, "w") as f2:
  print(fnr, file=f2)
 if fnr % 10000 == 0: break