# File name: graph_client.py
from concurrent.futures import ThreadPoolExecutor
import json
import requests
from concurrent.futures import ThreadPoolExecutor, as_completed
import time

# english_text = (
#     "It was the best of times, it was the worst of times, it was the age "
#     "of wisdom, it was the age of foolishness, it was the epoch of belief"
# )
# response = requests.post("http://127.0.0.1:8000/", json=english_text)
# french_text = response.text

# print(french_text)

test_image_url = "https://static.wixstatic.com/media/4d6b49_42b9435ce1104008b1b5f7a3c9bfcd69~mv2.jpg/v1/fill/w_454,h_333,fp_0.50_0.50,q_90/4d6b49_42b9435ce1104008b1b5f7a3c9bfcd69~mv2.jpg"
english_text = (
    "It was the best of times, it was the worst of times, it was the age "
    "of wisdom, it was the age of foolishness, it was the epoch of belief"
)


def send_text_request(number):
    json = {"text": english_text}
    response = requests.post("http://127.0.0.1:8000/", json=json)
    embeddings = response.text
    return number, embeddings

def process_text(numbers, max_workers=10):
    with ThreadPoolExecutor(max_workers=max_workers) as executor:
        futures = [executor.submit(send_text_request, number) for number in numbers]
        for future in as_completed(futures):
            n_result, result = future.result()
            result = json.loads(result)
            print (f"{n_result} : {len(result[0])}")

# def process_text(numbers, max_workers=10):
#     for n in numbers:
#         n_result, result = send_text_request(n)
#         result = json.loads(result)
#         print (f"{n_result} : {len(result[0])}")


if __name__ == "__main__":
    # n_calls = 100000
    n_calls = 1000
    numbers = list(range(n_calls))
    start_time = time.monotonic()
    process_text(numbers)
    end_time = time.monotonic()
    total_time = end_time - start_time
    avg_time_ms = total_time / n_calls * 1000
    calls_per_sec = n_calls / total_time
    print(f"Average time taken: {avg_time_ms:.2f} ms")
    print(f"Number of calls per second: {calls_per_sec:.2f}")