Update app.py

app.py

@@ -15,6 +15,8 @@ SERVICE_ACCOUNT_JSON = os.environ.get('GOOGLE_SHEET_CREDENTIALS')
 creds = service_account.Credentials.from_service_account_info(json.loads(SERVICE_ACCOUNT_JSON))
 SPREADSHEET_ID = '1o0iKPxWYKYKEPjqB2YwrTgrLzvGyb9ULj9tnw_cfJb0'
 service = build('sheets', 'v4', credentials=creds)
+LEFT_MODEL = None
+RIGHT_MODEL = None
 
 with open("article_list.pkl","rb") as articles:
     article_list = tuple(pickle.load(articles))
@@ -35,8 +37,10 @@ vecdbs = [FAISS.load_local(index_name, faiss_embedder)
           for index_name, faiss_embedder in zip(INDEXES, faiss_embedders)]
 
 def get_matchup():
+    global LEFT_MODEL, RIGHT_MODEL
     choices = INDEXES
     left, right = random.sample(choices,2)
+    LEFT_MODEL, RIGHT_MODEL = left, right
     return left, right
 
 def get_comp(prompt):
@@ -50,10 +54,11 @@ def get_article():
 
 
 def send_result(l_output, r_output, prompt, pick):
+    global LEFT_MODEL, RIGHT_MODEL
     # with open('results.csv','a') as res_file:
     #   writer = csv.writer(res_file)
     #   writer.writerow(row)
-    row = [USER_ID,prompt,l_output,r_output,pick]
+    row = [USER_ID,prompt,LEFT_MODEL,RIGHT_MODEL,pick]
     row = [str(x) for x in row]
     body = {'values': [row]}
     result = service.spreadsheets().values().append(spreadsheetId=SPREADSHEET_ID, range='A1:E1', valueInputOption='RAW', body=body).execute()