import os, json
from openai import OpenAI

os.environ["OPENAI_API_KEY"] = "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"
batch_id = "batch_69899f6279a48190abaad4d65407768b"

client = OpenAI()

OUTPUT_DIR = "batchapi/outputs"
os.makedirs(OUTPUT_DIR, exist_ok=True)

batch = client.batches.retrieve(batch_id)

print("Status:", batch.status)
print("Output file:", batch.output_file_id)
print("Error file:", batch.error_file_id)
print("Request counts:", batch.request_counts)


if batch.output_file_id:
    text = client.files.content(batch.output_file_id).text

    for line in text.splitlines():
        record = json.loads(line)
        img = record["custom_id"]
        out = record["response"]["output_parsed"]

        with open(f"{OUTPUT_DIR}/{img}.json", "w") as f:
            json.dump(out, f, indent=2)
else:
    print("No output file — batch likely failed.")