add functionality to update google spreadsheet with queries

app.py

@@ -1,16 +1,46 @@
 import os
 import re
 import io
+import json
 from typing import List, Tuple, Union
 from pathlib import Path
 import gradio as gr
 import openai
 import pymupdf
 from docx import Document
+from google.oauth2.service_account import Credentials
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 LEPTON_API_TOKEN = os.environ.get("LEPTON_API_TOKEN", None)
 
+# Set up Google Sheets API credentials
+SCOPES = ['https://www.googleapis.com/auth/spreadsheets']
+SERVICE_ACCOUNT_FILE = os.environ.get("GOOGLE_SECRET_AUTH_JSON", None)
+SPREADSHEET_ID = os.environ.get("SPREADSHEET_ID", None)
+
+def addrow_googlesheets(model, question, document, answer, reasoning, score):
+    try:
+        creds = Credentials.from_service_account_info(json.loads(SERVICE_ACCOUNT_FILE), scopes=SCOPES)
+        service = build('sheets', 'v4', credentials=creds)
+        body = {
+            'values': [[model, question, document, answer, reasoning, score]]
+        }
+        result = service.spreadsheets().values().append(
+            spreadsheetId=SPREADSHEET_ID,
+            range='Sheet1!A:F',
+            valueInputOption='RAW',
+            insertDataOption='INSERT_ROWS',
+            body=body
+        ).execute()
+        # print(f"{result.get('updates').get('updatedCells')} cells appended.")
+        return True
+
+    except HttpError as error:
+        print(f"An error occurred: {error}")
+        return False
+
 PROMPT = """
 Given the following QUESTION, DOCUMENT and ANSWER you must analyze the provided answer and determine whether it is faithful to the contents of the DOCUMENT. The ANSWER must not offer new information beyond the context provided in the DOCUMENT. The ANSWER also must not contradict information provided in the DOCUMENT. Output your final verdict by strictly following this format: "PASS" if the answer is faithful to the DOCUMENT and "FAIL" if the answer is not faithful to the DOCUMENT. Show your reasoning.
 
@@ -179,6 +209,8 @@ def model_call(question, document, answer, client_base_url):
     hallucination, reasoning = parse_patronus_lynx_response(response.choices[0].text)
     score = "🔴 FAIL 🔴" if hallucination else "🟢 PASS 🟢"
     combined_reasoning = " ".join(reasoning)[1:-1]
+    model = "Patronus Lynx 8B v1.1" if client_base_url=="https://yb15a7dy-lynx-v1-1-8b.tin.lepton.run/api/v1/" else "Patronus Lynx 70B"
+    addrow_googlesheets(model, question, document, answer, combined_reasoning, score)
     return combined_reasoning, score
 
 def return_approximate_token_size(text):

requirements.txt

@@ -1,3 +1,5 @@
 openai
 PyMuPDF
-python-docx
+python-docx
+google-auth
+google-api-python-client