From db49cc857ade2e23b91f13f0e7042bb1a4bf8a24 Mon Sep 17 00:00:00 2001
From: Mariko Medlock <marikomedlock@google.com>
Date: Fri, 28 Jul 2023 16:45:15 -0400
Subject: [PATCH] Fix notebook template to be valid ipynb.

---
 .../resources/export/notebook_template.ipynb  | 51 +++++++++++++------
 1 file changed, 36 insertions(+), 15 deletions(-)

diff --git a/service/src/main/resources/export/notebook_template.ipynb b/service/src/main/resources/export/notebook_template.ipynb
index e9a39af4a..582b211be 100644
--- a/service/src/main/resources/export/notebook_template.ipynb
+++ b/service/src/main/resources/export/notebook_template.ipynb
@@ -1,15 +1,36 @@
-import pandas
-import os
-
-# This query was generated for dataset ${underlayName} in study ${studyName} at ${timestamp}.
-${entityName}_sql = """
-${formattedSql}
-"""
-
-${entityName}_df = pandas.read_gbq(
-    dataset_${entityName}_sql,
-    dialect="standard",
-    use_bqstorage_api=("BIGQUERY_STORAGE_API_ENABLED" in os.environ),
-    progress_bar_type="tqdm_notebook")
-
-${entityName}_df.head(5)
+{
+  "metadata": {
+    "kernelspec": {
+      "display_name": "Python 3",
+      "language": "python",
+      "name": "python3"
+    }
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "import pandas\n",
+        "import os\n",
+        "\n",
+        "# This query was generated for dataset ${underlayName} in study ${studyName} at ${timestamp}.\n",
+        "${entityName}_sql = \"\"\"\n",
+        "${formattedSql}\n",
+        "\"\"\"\n",
+        "\n",
+        "${entityName}_df = pandas.read_gbq(\n",
+        "    dataset_${entityName}_sql,\n",
+        "    dialect=\"standard\",\n",
+        "    use_bqstorage_api=(\"BIGQUERY_STORAGE_API_ENABLED\" in os.environ),\n",
+        "    progress_bar_type=\"tqdm_notebook\")\n",
+        "\n",
+        "${entityName}_df.head(5)\n"
+      ]
+    }
+  ],
+  "nbformat": 4,
+  "nbformat_minor": 2
+}
\ No newline at end of file