{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a5e875f5",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    
    "\n# Code that generates the output\n\n",
    "answers = pd.DataFrame({\n",
    "    \"id\": [1, 2],\n",
    "    \"question\": [\"What is the capital of France?\", \"What is the capital of Germany?\"],\n",
    "    \"passages\": [\n",
    "        [\n",
    "            {\"passage\": \"Paris is the capital and most populous city of France.\", \n",
    "             \"doc_IDs\": [\"<urn:uuid:1234abcd-5678-efgh-9101-ijklmnopqrst>\", \"<urn:uuid:1234abcd-5678-efgh-9202-ijklmnopqrst>\"]},\n",
    "            {\"passage\": \"France is located in Western Europe.\", \n",
    "             \"doc_IDs\": [\"<urn:uuid:1234abcd-5678-efgh-9101-ijklmnopqrst>\"]}\n",
    "        ],\n",
    "        [\n",
    "            {\"passage\": \"Berlin is the capital of Germany.\", \n",
    "             \"doc_IDs\": [\"<urn:uuid:1234abcd-5678-efgh-9101-ijklmnopqrst>\"]}\n",
    "        ]\n",
    "    ],\n",
    "    \"final_prompt\": [\n",
    "        \"Using the following - Paris is the capital and most populous city of France - and - France is located in Western Europe - answer the question: What is the capital of France?\",\n",
    "        \"Using the following - Berlin is the capital of Germany - answer the question: What is the capital of Germany?\"\n",
    "    ],\n",
    "    \"answer\": [\"Paris\", \"Berlin\"]\n",
    "})\n\n",
    "# Convert to JSON format\n",
    "answers_json = answers.to_json(orient='records', lines=True, force_ascii=False)\n",
    "\n",
    "# Or just save to a file\n",
    "answers.to_json(\"answers.jsonl\", orient='records', lines=True, force_ascii=False)\n",
    "\n",
    "# Load the file to make sure it is correct\n",
    "loaded_answers = pd.read_json(\"answers.jsonl\", lines=True)\n\n",
    "# Print the loaded answers\n",
    "print(loaded_answers)\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}