import json

import openai

from ai_scientist.perform_review import load_paper, perform_review

client = openai.OpenAI(api_key="ollama", base_url="http://localhost:11434/v1")
model = "mistral-nemo"

# Load paper from pdf file (raw text)
# paper_txt = load_paper("review_test.pdf")

### include the path to the aider chat of the research, from results/{MODEL-USED}
paper_txt = load_paper("/home/ubuntu/ollama-ai-sci/AI-Scientist/results/nanoGPT/20240918_182054_refined_augmented_preprocessing/20240918_182054_refined_augmented_preprocessing_aider.txt")

# Get the review dict of the review
review = perform_review(
    paper_txt,
    model,
    client,
    num_reflections=5,
    num_fs_examples=1,
    num_reviews_ensemble=1,
    temperature=0.1,
)
with open("review.txt", "w") as f:
    f.write(json.dumps(review, indent=4))