File size: 2,498 Bytes
71c3841 00695de 0158882 71c3841 5736615 71c3841 00695de 5736615 71c3841 0158882 71c3841 5736615 |
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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import uuid
from google.oauth2 import service_account
from googleapiclient.discovery import build
from env_setup import config
import pandas as pd
SCOPES = ['https://www.googleapis.com/auth/spreadsheets']
def authenticate_google_account():
service_account_file = config["google_creds"]
if not service_account_file:
raise ValueError("Service account credentials path is missing in env_setup.py.")
return service_account.Credentials.from_service_account_file(service_account_file, scopes=SCOPES)
def store_data_in_sheet(sheet_id, chunks, summary, overall_sentiment):
creds = authenticate_google_account()
service = build('sheets', 'v4', credentials=creds)
sheet = service.spreadsheets()
call_id = str(uuid.uuid4())
print(f"Call ID: {call_id}")
values = []
if chunks:
first_chunk, first_sentiment, _ = chunks[0]
values.append([call_id, first_chunk, first_sentiment, summary, overall_sentiment])
for chunk, sentiment, _ in chunks[1:]:
values.append(["", chunk, sentiment, "", ""])
header = ["Call ID", "Chunk", "Sentiment", "Summary", "Overall Sentiment"]
all_values = [header] + values
body = {'values': all_values}
try:
result = sheet.values().append(
spreadsheetId=sheet_id,
range="Sheet1!A1",
valueInputOption="RAW",
body=body
).execute()
print(f"{result.get('updates').get('updatedCells')} cells updated in Google Sheets.")
except Exception as e:
print(f"Error updating Google Sheets: {e}")
def fetch_call_data(sheet_id, sheet_range="Sheet1!A1:E"):
try:
creds = authenticate_google_account()
service = build('sheets', 'v4', credentials=creds)
sheet = service.spreadsheets()
result = sheet.values().get(
spreadsheetId=sheet_id,
range=sheet_range
).execute()
rows = result.get("values", [])
print(f"Fetched rows: {rows}") # Log fetched rows
if rows:
headers = rows[0]
data = rows[1:]
df = pd.DataFrame(data, columns=headers)
print(f"DataFrame: {df}") # Log DataFrame
return df
else:
print("No data found in the specified range.")
return pd.DataFrame()
except Exception as e:
print(f"Error fetching data from Google Sheets: {e}")
return pd.DataFrame() |