import time
from ModelMarketSDK.ModelMarket import Mixtral8x7BModelMarketTestnet

# Membaca private key dan public key dari file teks
with open('private_key.txt', 'r') as file:
    PRIVATE_KEY = file.read().strip()

with open('public_key.txt', 'r') as file:
    PUBLIC_KEY = file.read().strip()

model_market = Mixtral8x7BModelMarketTestnet(PRIVATE_KEY, PUBLIC_KEY)

# Fungsi untuk mengirim permintaan dan mendapatkan respons
def get_response():
    chat = [{"role": "system", "content": "You are a helpful assistant!"}, {"role":"user", "content": "What is 2+2?"}]

    # the generate_self_requesting method takes the same input
    # but does not return the response but rather the url of the node that is processing
    # the request and the result_code (unique code for each request)
    node_url, result_code = model_market.generate_self_requesting(3000, chat)

    full_resp = ""
    done = False
    while not done:
        # the get_next_output function takes the url, the result_code and what has been
        # generated so far and returns the next piece of content as a str and a bool
        # indicating whether the AI generation is done
        resp, done = model_market.get_next_output(node_url, result_code, full_resp)
        full_resp += resp
        print(resp, end="")
        time.sleep(0.1)
    print("\n")  # Tambahkan baris baru setelah setiap respons

# Mengulang proses sebanyak 20 kali
for i in range(20):
    print(f"Iteration {i + 1}:")
    get_response()