Update backend/backend.py

backend/backend.py

@@ -7,7 +7,7 @@ from oauth2client.service_account import ServiceAccountCredentials
 
 class Backend:
     def __init__(self, sheet_name: str, credentials: str):
-        creds_dict = json.loads(credentials)
+        creds_dict = json.load(open(credentials))
         scope = [
             "https://spreadsheets.google.com/feeds",
             "https://www.googleapis.com/auth/drive",
@@ -26,82 +26,60 @@ class Backend:
     def add_row(
         self,
         index_in_dataset,
-        audio_id,
-        interpretation,
+        interpretation_id,
         transcription_id,
         user_id,
+        answer,
         user_name,
-        ssr_answer,
-        user_ssr_answer,
-        ssr_label,
-        ssd_answer,
-        user_ssd_answer,
-        ssd_label,
+        is_correct=None,
     ):
         timestamp = datetime.utcnow().isoformat()
         self.sheet.append_row(
             [
                 index_in_dataset,
-                audio_id,
-                interpretation,
+                interpretation_id,
                 transcription_id,
                 user_id,
                 user_name,
-                ssr_answer,
-                user_ssr_answer,
-                ssr_label,
-                ssd_answer,
-                user_ssd_answer,
-                ssd_label,
+                answer,
+                is_correct if is_correct is not None else "",
                 timestamp,
             ]
         )
 
     def update_row(
-        self,
-        index_in_dataset,
-        audio_id,
-        user_id,
-        new_ssr_answer=None,
-        new_user_ssr_answer=None,
-        new_ssr_label=None,
-        new_ssd_answer=None,
-        new_user_ssd_answer=None,
-        new_ssd_label=None,
+        self, index_in_dataset, interpretation_id, user_id, new_answer, is_correct=None
     ):
         records = self.get_all_rows().to_dict("records")
         for idx, row in enumerate(records):
             if (
-                row["audio_id"] == audio_id
+                row["interpretation_id"] == interpretation_id
                 and row["index_in_dataset"] == index_in_dataset
                 and row["user_id"] == user_id
             ):
+                # Update the is_correct field if provided
                 sheet_row = (
                     idx + 2
                 )  # +2 because sheet rows are 1-indexed and header is row 1
+                if row["answer"] != new_answer:
+                    self.sheet.update_cell(
+                        sheet_row, self.header.index("answer") + 1, new_answer
+                    )
+                    self.sheet.update_cell(
+                        sheet_row,
+                        self.header.index("timestamp") + 1,
+                        datetime.utcnow().isoformat(),
+                    )
+                    # update is_correct column
+                    self.sheet.update_cell(
+                        sheet_row,
+                        self.header.index("is_correct") + 1,
+                        is_correct if is_correct is not None else "",
+                    )
 
-                def update_cell_if_needed(column_name, new_value):
-                    if new_value is not None and row[column_name] != new_value:
-                        self.sheet.update_cell(
-                            sheet_row, self.header.index(column_name) + 1, new_value
-                        )
-
-                update_cell_if_needed("ssr_answer", new_ssr_answer)
-                update_cell_if_needed("user_ssr_answer", new_user_ssr_answer)
-                update_cell_if_needed("ssr_label", new_ssr_label)
-                update_cell_if_needed("ssd_answer", new_ssd_answer)
-                update_cell_if_needed("user_ssd_answer", new_user_ssd_answer)
-                update_cell_if_needed("ssd_label", new_ssd_label)
-
-                # Always update timestamp if any answer was updated
-                self.sheet.update_cell(
-                    sheet_row,
-                    self.header.index("timestamp") + 1,
-                    datetime.utcnow().isoformat(),
-                )
                 return True
         return False
 
-    def get_answer_count(self, audio_id):
+    def get_answer_count(self, interpretation_id):
         df = self.get_all_rows()
-        return df[df["audio_id"] == audio_id]["user_id"].nunique()
+        return df[df["interpretation_id"] == interpretation_id]["user_id"].nunique()