From 833a446899b125b1114f573ccd36a0e741ebe154 Mon Sep 17 00:00:00 2001
From: Guillem Borrell <borrell.guillem@bcg.com>
Date: Wed, 12 Jun 2024 12:45:09 +0200
Subject: [PATCH] Save session when file is uploaded

---
 src/hellocomputer/routers/files.py    | 11 ++++++++++-
 src/hellocomputer/routers/sessions.py | 13 ++-----------
 src/hellocomputer/static/index.html   |  9 +++++----
 src/hellocomputer/static/script.js    | 24 ++++++++++++++++++++++++
 4 files changed, 41 insertions(+), 16 deletions(-)

diff --git a/src/hellocomputer/routers/files.py b/src/hellocomputer/routers/files.py
index 0af6552..da1057e 100644
--- a/src/hellocomputer/routers/files.py
+++ b/src/hellocomputer/routers/files.py
@@ -3,10 +3,12 @@ import aiofiles
 # import s3fs
 from fastapi import APIRouter, File, UploadFile
 from fastapi.responses import JSONResponse
+from starlette.requests import Request
 
 from ..config import settings
 from ..db import StorageEngines
 from ..sessions import SessionDB
+from ..users import OwnershipDB
 
 router = APIRouter()
 
@@ -21,7 +23,7 @@ router = APIRouter()
 
 
 @router.post("/upload", tags=["files"])
-async def upload_file(file: UploadFile = File(...), sid: str = ""):
+async def upload_file(request: Request, file: UploadFile = File(...), sid: str = ""):
     async with aiofiles.tempfile.NamedTemporaryFile("wb") as f:
         content = await file.read()
         await f.write(content)
@@ -39,6 +41,13 @@ async def upload_file(file: UploadFile = File(...), sid: str = ""):
             .dump()
         )
 
+        OwnershipDB(
+            StorageEngines.gcs,
+            gcs_access=settings.gcs_access,
+            gcs_secret=settings.gcs_secret,
+            bucket=settings.gcs_bucketname,
+        ).set_ownersip(request.session.get("user").get("email"), sid)
+
         return JSONResponse(
             content={"message": "File uploaded successfully"}, status_code=200
         )
diff --git a/src/hellocomputer/routers/sessions.py b/src/hellocomputer/routers/sessions.py
index adc2532..f1446f1 100644
--- a/src/hellocomputer/routers/sessions.py
+++ b/src/hellocomputer/routers/sessions.py
@@ -16,17 +16,8 @@ router = APIRouter()
 
 
 @router.get("/new_session")
-async def get_new_session(request: Request) -> str:
-    user_email = request.session.get("user").get("email")
-    ownership = OwnershipDB(
-        StorageEngines.gcs,
-        gcs_access=settings.gcs_access,
-        gcs_secret=settings.gcs_secret,
-        bucket=settings.gcs_bucketname,
-    )
-    sid = str(uuid4())
-
-    return ownership.set_ownersip(user_email, sid)
+async def get_new_session() -> str:
+    return str(uuid4())
 
 
 @router.get("/greetings", response_class=PlainTextResponse)
diff --git a/src/hellocomputer/static/index.html b/src/hellocomputer/static/index.html
index 8465d3f..b4ee779 100644
--- a/src/hellocomputer/static/index.html
+++ b/src/hellocomputer/static/index.html
@@ -86,8 +86,9 @@
                 </div>
 
                 <!-- Button trigger modal -->
-                <button type="button" class="btn btn-primary" data-bs-toggle="modal" data-bs-target="#staticBackdrop">
-                    Or load an existing session
+                <button type="button" class="btn btn-primary" data-bs-toggle="modal" data-bs-target="#staticBackdrop"
+                    id="loadSessionsButton">
+                    Browse sessions
                 </button>
 
                 <!-- Modal -->
@@ -100,8 +101,8 @@
                                 <button type="button" class="btn-close" data-bs-dismiss="modal"
                                     aria-label="Close"></button>
                             </div>
-                            <div class="modal-body">
-                                <p>Current sessions</p>
+                            <div class="modal-body" id="userSessions">
+                                <p>Current sessions...</p>
                             </div>
                             <div class="modal-footer">
                                 <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Close</button>
diff --git a/src/hellocomputer/static/script.js b/src/hellocomputer/static/script.js
index 63d2de6..fd14e46 100644
--- a/src/hellocomputer/static/script.js
+++ b/src/hellocomputer/static/script.js
@@ -13,6 +13,7 @@ $('#menu-toggle').click(function (e) {
     toggleMenuArrow(document.getElementById('menu-toggle'));
 });
 
+// Hide sidebar on mobile devices
 document.addEventListener("DOMContentLoaded", function () {
     console.log('Width: ' + window.innerWidth + ' Height: ' + window.innerHeight);
     if ((window.innerWidth <= 800) && (window.innerHeight <= 600)) {
@@ -25,8 +26,10 @@ document.addEventListener("DOMContentLoaded", function () {
 
 const textarea = document.getElementById('chatTextarea');
 const sendButton = document.getElementById('sendButton');
+const sessions = document.getElementById('userSessions');
 const chatMessages = document.querySelector('.chat-messages');
 
+// Auto resize textarea
 textarea.addEventListener('input', function () {
     this.style.height = 'auto';
     this.style.height = (this.scrollHeight <= 150 ? this.scrollHeight : 150) + 'px';
@@ -53,6 +56,7 @@ async function fetchResponse(message, newMessage) {
     }
 }
 
+// Function to add AI message
 function addAIMessage(messageContent) {
     const newMessage = document.createElement('div');
     newMessage.classList.add('message', 'bg-white', 'p-2', 'mb-2', 'rounded');
@@ -123,6 +127,7 @@ document.addEventListener("DOMContentLoaded", function () {
     fetchGreeting();
 });
 
+// Function upload the data file
 document.addEventListener("DOMContentLoaded", function () {
     const fileInput = document.getElementById('inputGroupFile01');
     const uploadButton = document.getElementById('uploadButton');
@@ -158,3 +163,22 @@ document.addEventListener("DOMContentLoaded", function () {
         }
     });
 });
+
+// Function to get the user sessions
+document.addEventListener("DOMContentLoaded", function () {
+    const sessionsButton = document.getElementById('loadSessionsButton');
+    sessionsButton.addEventListener('click', async function fetchSessions() {
+        try {
+            const response = await fetch('/sessions');
+            if (!response.ok) {
+                throw new Error('Network response was not ok ' + response.statusText);
+            }
+            const data = await response.text();
+            sessions.innerHTML = data;
+        } catch (error) {
+            sessions.innerHTML = 'Error: ' + error.message;
+        }
+    }
+    );
+}
+);
\ No newline at end of file