Updated calendar tool to support service accounts

app.py

@@ -8,68 +8,50 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def add_event_to_calendar(event_name:str, start:str, end:str)-> str: # it's important to specify the return type
-    """A tool that allows you to create events on google calendar. 
-        Time should be represented in this format: 2015-05-28T09:00:00+utc. for now let utc be 01:00
-    Args:
-        event_name: The name of the event.
-        start: The start time of the event represented as a datetime str.
-        end: The end time of the event respresented as a datetime str.
+def add_event_to_calendar(event_name: str, start: str, end: str) -> str:
+    """A tool that creates events on Google Calendar using a service account.
+       Time format: 2015-05-28T09:00:00+01:00 (WAT = UTC+1)
     """
-    import datetime
-    import os.path
-
-    from google.auth.transport.requests import Request
-    from google.oauth2.credentials import Credentials
-    from google_auth_oauthlib.flow import InstalledAppFlow
+    from google.oauth2 import service_account
     from googleapiclient.discovery import build
     from googleapiclient.errors import HttpError
 
-
+    # Service account configuration
+    SERVICE_ACCOUNT_FILE = "credentials.json"  # Your service account JSON file
     SCOPES = ["https://www.googleapis.com/auth/calendar"]
-    
-    creds = None
-    # The file token.json stores the user's access and refresh tokens, and is
-    # created automatically when the authorization flow completes for the first
-    # time.
-    if os.path.exists("token.json"):
-        creds = Credentials.from_authorized_user_file("token.json", SCOPES)
-    # If there are no (valid) credentials available, let the user log in.
-    if not creds or not creds.valid:
-        if creds and creds.expired and creds.refresh_token:
-            creds.refresh(Request())
-        else:
-            flow = InstalledAppFlow.from_client_secrets_file(
-                "credentials.json", SCOPES
-            )
-            creds = flow.run_local_server(port=0)
-        # Save the credentials for the next run
-        with open("token.json", "w") as token:
-            token.write(creds.to_json())
+    CALENDAR_ID = "primary"  # Or your specific calendar ID
 
     try:
+        # Authenticate with service account
+        creds = service_account.Credentials.from_service_account_file(
+            SERVICE_ACCOUNT_FILE, scopes=SCOPES
+        )
+
         service = build("calendar", "v3", credentials=creds)
 
-        # Call the Calendar API
         event = {
             'summary': event_name,
             'start': {
                 'dateTime': start,
-                'timeZone': '', # Africa/Lagos
+                'timeZone': 'Africa/Lagos',  # Fixed timezone
             },
-            
             'end': {
                 'dateTime': end,
-                'timeZone': '',
+                'timeZone': 'Africa/Lagos',
             },
         }
 
-        event = service.events().insert(calendarId='primary', body=event).execute()
+        event = service.events().insert(
+            calendarId=CALENDAR_ID,
+            body=event
+        ).execute()
 
-    except HttpError as error:
-        print(f"An error occurred: {error}")
-    return f"Event '{event_name}' has been added successfully to calendar✅"
+        return f"Event '{event_name}' added successfully! 🎉"
 
+    except HttpError as error:
+        return f"❌ Error: {error}"
+    except Exception as e:
+        return f"❌ Unexpected error: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: