File size: 1,511 Bytes
1aa1e51 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import gspread
from oauth2client.service_account import ServiceAccountCredentials
import pandas as pd
import os
import json
def get_gspread_client():
scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
google_creds_json = os.environ.get("GOOGLE_CREDENTIALS_JSON")
if google_creds_json:
# Use credentials from Hugging Face secret
creds_dict = json.loads(google_creds_json)
creds = ServiceAccountCredentials.from_json_keyfile_dict(creds_dict, scope)
else:
# Fallback to local credentials file
creds = ServiceAccountCredentials.from_json_keyfile_name("credentials.json", scope)
client = gspread.authorize(creds)
return client
def upload_to_google_sheets(session_data):
client = get_gspread_client()
sheet = client.open("Bipolar Chatbot Logs").sheet1
# Convert session_data to a pandas DataFrame
df = pd.DataFrame(session_data)
# Get existing headers from the sheet
try:
existing_headers = sheet.row_values(1)
except gspread.exceptions.APIError:
existing_headers = []
# Define column order
columns = ['query', 'response', 'sources', 'feedback', 'source_feedback']
df = df[columns]
# If the sheet is empty, write the headers
if not existing_headers:
sheet.append_row(columns)
# Append the DataFrame to the sheet
for row in df.itertuples(index=False, name=None):
sheet.append_row(list(row))
|