Allowing multiple events to be added and states for ChatWindow

atomicgamedeveloper · atomicgamedeveloper · commit 12516320a17c · 2024-05-09T00:34:39.000+02:00
diff --git a/calendar.js b/calendar.js
@@ -4,8 +4,6 @@ const process = require('process');
 const { authenticate } = require('@google-cloud/local-auth');
 const { google } = require('googleapis');
 
-console.log("I am working.");
-
 // If modifying these scopes, delete token.json.
 const SCOPES = ['https://www.googleapis.com/auth/calendar'];
 // The file token.json stores the user's access and refresh tokens, and is
@@ -14,56 +12,76 @@ const SCOPES = ['https://www.googleapis.com/auth/calendar'];
 const TOKEN_PATH = path.join(process.cwd(), 'token.json');
 const CREDENTIALS_PATH = path.join(process.cwd(), 'credentials.json');
 
-/**
- * Creates a new event on the user's primary calendar.
- * @param {google.auth.OAuth2} auth An authorized OAuth2 client.
- * @param {Object} eventBody The event details.
- */
-async function createEvent(auth, eventBody) {
+async function createEvents(auth, eventBodiesArray) {
     const calendar = google.calendar({ version: 'v3', auth });
 
-    try {
-        const event = await calendar.events.insert({
+    const promises = eventBodiesArray.map(eventBody => {
+        return calendar.events.insert({
             calendarId: 'primary',
             resource: eventBody,
+        }).then(event => {
+            console.log(`Event created: ${event.data.htmlLink}`);
+            return event; // Return event for further processing if needed.
+        }).catch(error => {
+            console.error('Error creating event', error);
+            return null; // Return null or appropriate error handling.
         });
-        console.log(`Event created: ${event.data.htmlLink}`);
+    });
+
+    try {
+        const results = await Promise.all(promises);
+        // Optional: Process results or perform further actions here.
     } catch (error) {
-        console.error('Error creating event', error);
+        console.error('An error occurred with Promise.all', error);
     }
 }
 
-/**
- * Creates an object literal representing the event details.
- * @param {string} summary The summary of the event.
- * @param {string} location The location of the event.
- * @param {string} description The description of the event.
- * @param {Date} startDateTime The start date and time of the event.
- * @param {Date} endDateTime The end date and time of the event.
- * @returns {Object} The event object literal.
- */
+function createEventObjects(optionsArray) {
+    // Initialize an array to hold all event objects
+    let events = [];
+
+    // Iterate over each set of options
+    optionsArray.forEach(options => {
+        let event = {
+            'summary': options.summary,
+            'location': options.location,
+            'description': options.description,
+            // Initialize start and end events without dateTime or date
+            'start': {},
+            'end': {}
+        };
+
+        // Check if it's an all day event
+        if (options.allDay) {
+            event.start.date = options.start;
+            event.end.date = options.end;
+            event.start.timeZone = 'GMT+2';
+            event.end.timeZone = 'GMT+2';
+        } else {
+            event.start.dateTime = options.start;
+            event.end.dateTime = options.end;
+            event.start.timeZone = options.startTimeZone || 'Etc/GMT+2';
+            event.end.timeZone = options.endTimeZone || 'Etc/GMT+2';
+        }
+
+        // Add recurrence if provided
+        if (options.recurrence) {
+            event.recurrence = options.recurrence;
+        }
+
+        // Add reminders if provided
+        if (options.reminders) {
+            event.reminders = {
+                'useDefault': false,
+                'overrides': options.reminders
+            };
+        }
+
+        // Add the constructed event to the events array
+        events.push(event);
+    });
 
-function createEventObject(summary, location, description, startDateTime, endDateTime) {
-    return {
-        'summary': summary,
-        'location': location,
-        'description': description,
-        'start': {
-            'dateTime': startDateTime.toISOString(),
-            'timeZone': 'America/Los_Angeles', // Adjust to the desired timezone
-        },
-        'end': {
-            'dateTime': endDateTime.toISOString(),
-            'timeZone': 'America/Los_Angeles', // Adjust to the desired timezone
-        },
-        'reminders': {
-            'useDefault': false,
-            'overrides': [
-                { 'method': 'email', 'minutes': 24 * 60 },
-                { 'method': 'popup', 'minutes': 10 },
-            ],
-        },
-    };
+    return events; // Returns an array of event objects
 }
 
 /**
@@ -128,46 +146,23 @@ async function authorize() {
     return client;
 }
 
-/**
- * Lists the next 10 events on the user's primary calendar.
- * @param {google.auth.OAuth2} auth An authorized OAuth2 client.
- */
-async function listEvents(auth) {
-    const calendar = google.calendar({ version: 'v3', auth });
-    const res = await calendar.events.list({
-        calendarId: 'primary',
-        timeMin: new Date().toISOString(),
-        maxResults: 5,
-        singleEvents: true,
-        orderBy: 'startTime',
-    });
-    const events = res.data.items;
-    if (!events || events.length === 0) {
-        console.log('No upcoming events found.');
-        return;
-    }
-    console.log('Upcoming 10 events:');
-    events.map((event, i) => {
-        const start = event.start.dateTime || event.start.date;
-        console.log(`${start} - ${event.summary}`);
-    });
-}
-
-authorize().then(listEvents).catch(console.error);
-
-// Example usage
 authorize().then(auth => {
+    // Preparing the date format for an all-day event
+    const startDate = new Date('2024-05-15').toISOString().substring(0, 10); // Converting to 'YYYY-MM-DD' format
+    const endDate = new Date('2024-05-16').toISOString().substring(0, 10);  // All-day event end date should be the next day in 'YYYY-MM-DD' format
 
     // Creating an event object
-    const newEvent = createEventObject(
-        'Team Meeting', // Summary
-        'Online', // Location
-        'Discuss project progress', // Description
-        new Date('2024-05-15T09:00:00'), // Start date and time
-        new Date('2024-05-15T10:00:00') // End date and time
-    );
-
-    // Creating the event on Google Calendar
-    createEvent(auth, newEvent).catch(console.error);
+    const newEvent = createEventObjects([{
+        summary: 'Team Meeting - All Day', // Summary
+        location: 'Online', // Location
+        description: 'Discuss project progress', // Description
+        start: startDate, // Start date for all-day event
+        end: endDate, // End date for all-day event
+        allDay: true, // Marking this event as an all-day event
+        // Time zone is irrelevant for all-day events but set for consistency if needed
+    }]);
+
+    // Assuming createEvent is a function that creates an event on Google Calendar using the provided authentication and event object
+    createEvents(auth, newEvent).catch(console.error);
 
 }).catch(console.error);
diff --git a/interaction.js b/interaction.js
@@ -19,11 +19,17 @@ const defaultSystemPrompt = "You are a helpful chatbot.";
 const GPT4 = "gpt-4-turbo-preview";
 const GPT3_5 = 'gpt-3.5-turbo-0125';
 
+const CHAT_WINDOW_STATES = {
+    CREATE: 'Create',
+    SPECIFY: 'Specify',
+};
+
 class ChatWindow {
     #messages = [];
     #USER = "user";
     #ASSISTANT = "assistant";
     #SYSTEM = "system";
+    #STATE = CHAT_WINDOW_STATES.CREATE;
 
     constructor(model = GPT3_5, systemPrompt = defaultSystemPrompt) {
         if (model != GPT4 && model != GPT3_5) {
@@ -34,6 +40,17 @@ class ChatWindow {
         this.#messagesAppend(systemMessage);
     }
 
+    getState() {
+        return this.#STATE;
+    }
+
+    setState(state) {
+        if (state != CHAT_WINDOW_STATES.CREATE && state != CHAT_WINDOW_STATES.SPECIFY) {
+            throw new Error(`State '${model}' does not exist.`)
+        }
+        this.#STATE = state;
+    }
+
     #makeMessage(role, message) {
         if (role != this.#USER && role != this.#ASSISTANT && role != this.#SYSTEM) {
             throw new Error(`Role '${role}' is not a valid role.`)
@@ -48,6 +65,7 @@ class ChatWindow {
     async #streamResponse(messages) {
         let stream = await openai.chat.completions.create({
             'model': this.model,
+            //'response_format': { 'type': 'json_object' },
             'messages': messages,
             'stream': true
         });
@@ -70,16 +88,52 @@ class ChatWindow {
 
     #generateMessageHTML(message) {
         let newParagraph = document.createElement('p');
-        newParagraph.innerHTML = message.role + ":\n" + message.content;
+        newParagraph.innerHTML = message.role + ":\n\n"
+        newParagraph.innerHTML += message.content;
         chatHistory.appendChild(newParagraph);
         chatHistory.scrollTo({ top: chatHistory.scrollHeight });
     }
 
+    #parseJSONToObject(jsonString) {
+        try {
+            let obj = JSON.parse(jsonString);
+            return obj;
+        } catch (e) {
+            // Log error or handle it as needed
+            console.error('Error parsing JSON', e);
+            return null; // or return an empty object {}, based on your needs
+        }
+    }
+
+    #generateMessageHTMLStreamed(message) {
+        return new Promise((resolve) => {
+            let newParagraph = document.createElement("p");
+            newParagraph.innerHTML = `<strong>${message.role}</strong>:<br>`;
+            chatHistory.appendChild(newParagraph);
+
+            let index = 0;
+            function typeMessage() {
+                if (index < message.content.length) {
+                    let chunkSize = Math.floor(Math.random() * 4) + 2; // 2-3 characters
+                    let part = message.content.slice(index, index + chunkSize);
+                    newParagraph.innerHTML += part;
+                    index += chunkSize;
+
+                    chatHistory.scrollTo({ top: chatHistory.scrollHeight });
+                    setTimeout(typeMessage, Math.random() * 100 + 100);
+                } else {
+                    resolve();
+                }
+            }
+            typeMessage();
+        });
+    }
+
     getMessages() {
         return this.#messages;
     }
 
-    sendMessage(prompt) {
+    sendUserMessage(prompt) {
         let userMessage = this.#makeMessage(this.#USER, prompt);
         this.#messagesAppend(userMessage);
         this.#generateMessageHTML(userMessage);
@@ -88,14 +142,40 @@ class ChatWindow {
         this.#generateMessageHTML(emptyResponseMessage);
         this.#streamResponse(messages);
     }
+
+    async sendSystemMessage(prompt) {
+        let systemMessage = this.#makeMessage(this.#SYSTEM, prompt);
+        await this.#generateMessageHTMLStreamed(systemMessage);
+    }
+}
+
+function sleep(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms))
 }
 
-const chatWindow = new ChatWindow(GPT3_5);
+const chatWindow = new ChatWindow(GPT4);
+(async () => {
+    await chatWindow.sendSystemMessage("Welcome to CalendarMe!");
+    await sleep(1000);
+    await chatWindow.sendSystemMessage("Please write your plans here in as much detail as you like.")
+})();
 
 textInputBox.addEventListener('keydown', function (e) {
     if (e.key === 'Enter') {
+        let today = new Date().toISOString().split('T')[0];
+        let time = new Date().toTimeString().split(' ')[0];
+        let day = new Date().toLocaleString('en-us', { weekday: 'long' });
         let textFromInputBox = textInputBox.value;
+        let planning_prompt = textFromInputBox;
+
         textInputBox.value = "";
-        chatWindow.sendMessage(textFromInputBox);
+
+        let constructedPrompt = planning_prompt//`Given the following current date and time: ${day}, ${today}T${time} and planning prompt: '${planning_prompt}', format the prompt's contents as JSON objects with the following keys: summary, location (Optional.), description, start, end, recurrence (Optional. array of RRULE strings), reminders (Optional. useDefault, overrides), timeZone (Etc/GMT+2), allDay (boolean), in an array that can be parsed to create calendar events. Please use 1-2 emojis per complex sentence in the title's lhs and description to make them more personal.`;
+
+        if (chatWindow.getState() == CHAT_WINDOW_STATES.CREATE) {
+            constructedPrompt = `Given the following current date and time: ${day}, ${today}T${time} and planning prompt: '${planning_prompt}', format the prompt's contents as JSON objects with the following keys: summary, location (Optional.), description, start, end, recurrence (Optional. array of RRULE strings), reminders (Optional. useDefault, overrides), timeZone (Etc/GMT+2), allDay (Optional. boolean), in an array that can be parsed to create calendar events. Please use 1-2 emojis per complex sentence in the title's lhs and description to make them more personal.`;
+            chatWindow.setState(CHAT_WINDOW_STATES.SPECIFY);
+        }
+        chatWindow.sendUserMessage(constructedPrompt);
     }
 });
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -29,6 +29,7 @@
   "dependencies": {
     "@google-cloud/local-auth": "^2.1.0",
     "googleapis": "^105.0.0",
+    "json": "^11.0.0",
     "openai": "^4.40.2"
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -29,6 +29,7 @@`
`29`	`29`	`"dependencies": {`
`30`	`30`	`"@google-cloud/local-auth": "^2.1.0",`
`31`	`31`	`"googleapis": "^105.0.0",`
	`32`	`+ "json": "^11.0.0",`
`32`	`33`	`"openai": "^4.40.2"`
`33`	`34`	`}`
`34`	`35`	`}`