Update app.py

app.py

@@ -1,12 +1,20 @@
 import gradio as gr
+import os
 import csv
 import random
 import pickle
 import uuid
 from langchain.vectorstores import FAISS
 from langchain.embeddings import HuggingFaceEmbeddings
+from googleapiclient.discovery import build
+from google.oauth2 import service_account
 
 USER_ID = uuid.uuid4()
+SERVICE_ACCOUNT_JSON = os.environ.get('GOOGLE_SHEETS_CREDENTIALS')
+creds = service_account.Credentials.from_service_account_info(json.loads(SERVICE_ACCOUNT_JSON))
+SPREADSHEET_ID = '1o0iKPxWYKYKEPjqB2YwrTgrLzvGyb9ULj9tnw_cfJb0'
+service = build('sheets', 'v4', credentials=creds)
+
 with open("article_list.pkl","rb") as articles:
     article_list = tuple(pickle.load(articles))
 INDEXES = ["miread_large", "miread_contrastive", "scibert_contrastive"]
@@ -41,10 +49,14 @@ def get_article():
 
 
 def send_result(l_output, r_output, prompt, pick):
-    with open('results.csv','a') as res_file:
-      writer = csv.writer(res_file)
-      row = [USER_ID,l_output,r_output,prompt,pick]
-      writer.writerow(row)
+    # with open('results.csv','a') as res_file:
+    #   writer = csv.writer(res_file)
+    #   writer.writerow(row)
+    row = [USER_ID,l_output,r_output,prompt,pick]
+    row = [str(x) for x in row]
+    body = {'values': [row]}
+    result = service.spreadsheets().values().append(spreadsheetId=SPREADSHEET_ID, range='A1:E1',majorDimension="ROWS", valueInputOption='RAW', body=body).execute()
+    print(f"Appended {result['updates']['updatedCells']} cells.")
     new_prompt = get_article()
     return new_prompt,gr.State.update(value=new_prompt)