add spreadsheet syncing

app.ipynb

@@ -16,6 +16,7 @@
    "outputs": [],
    "source": [
     "import json\n",
+    "import gspread\n",
     "import gradio as gr\n",
     "\n",
     "\n",
@@ -151,10 +152,12 @@
     "        # update overall ratings\n",
     "        task['overall'] = overall\n",
     "        task['notes'] = notes\n",
-    "        # save the task to json file\n",
     "        try:\n",
+    "            # save the task to json file\n",
     "            with open(f'./data/demo_task_{task_id}.json', 'w', encoding='utf-8') as task_file:\n",
     "                json.dump(task, task_file, ensure_ascii=False, indent=4)\n",
+    "            # save the task to google spreadsheet\n",
+    "            self.save_gsheet(task_id, ratings)\n",
     "            gr.Info(f'Task demo_task_{task_id} is saved!')\n",
     "        except:\n",
     "            raise gr.Error(f'Could not save the task demo_task_{task_id} :(')\n",
@@ -177,15 +180,46 @@
     "        # erase overall ratings\n",
     "        task['overall'] = ''\n",
     "        task['notes'] = ''\n",
-    "        # save the reset task to json file\n",
     "        try:\n",
+    "            # save the reset task to json file\n",
     "            with open(f'./data/demo_task_{task_id}.json', 'w', encoding='utf-8') as task_file:\n",
     "                json.dump(task, task_file, ensure_ascii=False, indent=4)\n",
-    "            gr.Warning(f'Task demo_task_{task_id} is reset!')\n",
+    "            # save the reset task to google spreadsheet\n",
+    "            self.reset_gsheet(task_id)\n",
+    "            gr.Info(f'Task demo_task_{task_id} is reset!')\n",
     "        except:\n",
     "            raise gr.Error(f'Could not reset the task demo_task_{task_id} :(')\n",
     "        return '', '', '', '', '', '', '', '', '', ''\n",
     "\n",
+    "    def save_gsheet(self, id, ratings):\n",
+    "        \"\"\"Save the task to google spreadsheet.\"\"\"\n",
+    "        # parse the ratings\n",
+    "        groundedness1, fluency1, utility1, notes1, \\\n",
+    "        groundedness2, fluency2, utility2, notes2, \\\n",
+    "        overall, notes = ratings\n",
+    "        try:\n",
+    "            # configure gsheet credentials\n",
+    "            gc = gspread.service_account('./gsheet_service_account.json')\n",
+    "            sheet_id = '1D2sfE9YXKtd7cKlgalo5UnuNKC-GhxlGqHVYUlkQlCY'\n",
+    "            sh = gc.open_by_key(sheet_id).worksheet('demo-app')\n",
+    "            # update task ratings in the worksheet\n",
+    "            sh.update(range_name=f'C{3+int(id)}:J{3+int(id)}',\n",
+    "                      values=[[groundedness1, fluency1, utility1, groundedness2, fluency2, utility2, overall, notes]])\n",
+    "        except:\n",
+    "            gr.Warning(f'Could not save the task demo_task_{task_id} to the spreadsheet :(')\n",
+    "\n",
+    "    def reset_gsheet(self, id):\n",
+    "        \"\"\"Reset the task ratings in google spreadsheet.\"\"\"\n",
+    "        try:\n",
+    "            # configure gsheet credentials\n",
+    "            gc = gspread.service_account('./gsheet_service_account.json')\n",
+    "            sheet_id = '1D2sfE9YXKtd7cKlgalo5UnuNKC-GhxlGqHVYUlkQlCY'\n",
+    "            sh = gc.open_by_key(sheet_id).worksheet('demo-app')\n",
+    "            # update task ratings in the worksheet\n",
+    "            sh.batch_clear([f'C{3+int(id)}:J{3+int(id)}'])\n",
+    "        except:\n",
+    "            gr.Warning(f'Could not reset the task demo_task_{task_id} in the spreadsheet :(')\n",
+    "\n",
     "    def launch_interface(self):\n",
     "        \"\"\"Launch the A/B Evaluation RAG task interface.\"\"\"\n",
     "        gr.close_all()\n",