Add about

src/hellocomputer/main.py

@@ -2,7 +2,7 @@ import json
 from pathlib import Path
 
 from fastapi import FastAPI
-from fastapi.responses import HTMLResponse, RedirectResponse, FileResponse
+from fastapi.responses import FileResponse, HTMLResponse, RedirectResponse
 from fastapi.staticfiles import StaticFiles
 from starlette.middleware.sessions import SessionMiddleware
 from starlette.requests import Request

src/hellocomputer/routers/analysis.py

@@ -1,9 +1,9 @@
 from fastapi import APIRouter
 from fastapi.responses import PlainTextResponse
 
-from hellocomputer.sessions import SessionDB
 from hellocomputer.db import StorageEngines
 from hellocomputer.extraction import extract_code_block
+from hellocomputer.sessions import SessionDB
 
 from ..config import settings
 from ..models import Chat

src/hellocomputer/routers/auth.py

@@ -2,9 +2,10 @@ from authlib.integrations.starlette_client import OAuth, OAuthError
 from fastapi import APIRouter
 from fastapi.responses import HTMLResponse, RedirectResponse
 from starlette.requests import Request
+
 from hellocomputer.config import settings
-from hellocomputer.users import UserDB
 from hellocomputer.db import StorageEngines
+from hellocomputer.users import UserDB
 
 router = APIRouter()
 

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 ..sessions import SessionDB
 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
         )

src/hellocomputer/routers/sessions.py

@@ -3,8 +3,11 @@ from uuid import uuid4
 from fastapi import APIRouter
 from fastapi.responses import PlainTextResponse
 from starlette.requests import Request
-from hellocomputer.users import OwnershipDB
+from typing import List
+
 from hellocomputer.db import StorageEngines
+from hellocomputer.users import OwnershipDB
+
 from ..config import settings
 
 # Scheme for the Authorization header
@@ -13,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)
@@ -32,3 +26,15 @@ async def get_greeting() -> str:
         "Hi! I'm a helpful assistant. Please upload or select a file "
         "and I'll try to analyze it following your orders"
     )
+
+
+@router.get("/sessions")
+async def get_sessions(request: Request) -> List[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,
+    )
+    return ownership.sessions(user_email)

src/hellocomputer/static/about.html

@@ -0,0 +1,49 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Hola, computer!</title>
+    <link rel="icon" type="image/x-icon" href="/app/img/favicon.ico">
+    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/css/bootstrap.min.css" rel="stylesheet">
+    <link href="https://fonts.googleapis.com/css2?family=Share+Tech+Mono&display=swap" rel="stylesheet">
+    <style>
+        .login-container {
+            max-width: 400px;
+            margin: 0 auto;
+            padding: 50px 0;
+        }
+
+        .logo {
+            display: block;
+            margin: 0 auto 20px auto;
+        }
+
+        .techie-font {
+            font-family: 'Share Tech Mono', monospace;
+            font-size: 24px;
+            text-align: center;
+            margin-bottom: 20px;
+        }
+    </style>
+</head>
+
+<body>
+    <div class="container">
+        <div class="login-container text-center">
+            <h1 class="h3 mb-3 fw-normal techie-font">Hola, computer!</h1>
+            <img src="/app/img/assistant.webp" alt="Logo" class="logo img-fluid">
+            <p class="techie-font">
+                Hola, computer! is a web assistant that allows you to query excel files using natural language. It may
+                not be as powerful as Excel, but it has an efficient query backend that can process your data faster
+                and more efficiently than Excel.
+            </p>
+            <a href="/"><button class="btn btn-secondary w-100">Back</button></a>
+        </div>
+    </div>
+
+    <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/js/bootstrap.bundle.min.js"></script>
+</body>
+
+</html>

src/hellocomputer/static/index.html

@@ -63,9 +63,8 @@
                                     <label for="datasetLabel" class="form-label">Sesson name</label>
                                     <input type="text" class="form-control" id="datasetLabel"
                                         aria-describedby="labelHelp">
-                                    <div id="labelHelp" class="form-text">You'll be able to recover this file in the
-                                        future
-                                        with this name
+                                    <div id="labelHelp" class="form-text">
+                                        You'll be able to recover this file in the future with this name
                                     </div>
                                 </div>
                                 <div class="modal-body">
@@ -86,8 +85,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">
+                    Load a session
                 </button>
 
                 <!-- Modal -->
@@ -96,15 +96,17 @@
                     <div class="modal-dialog modal-dialog-scrollable">
                         <div class="modal-content">
                             <div class="modal-header">
-                                <h5 class="modal-title" id="staticBackdropLabel">Current sessions</h5>
+                                <h5 class="modal-title" id="staticBackdropLabel">Available sessions</h5>
                                 <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">
+                                <ul id="userSessions">
+                                </ul>
                             </div>
                             <div class="modal-footer">
-                                <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Close</button>
+                                <button type="button" class="btn btn-secondary" data-bs-dismiss="modal"
+                                    id="sessionCloseButton">Close</button>
                             </div>
                         </div>
                     </div>

src/hellocomputer/static/login.html

@@ -35,6 +35,8 @@
             <h1 class="h3 mb-3 fw-normal techie-font">Hola, computer!</h1>
             <img src="/app/img/assistant.webp" alt="Logo" class="logo img-fluid">
             <a href="/login"><button type="submit" class="btn btn-primary w-100">Login</button></a>
+            <p></p>
+            <a href="/app/about.html"><button class="btn btn-secondary w-100">About</button></a>
         </div>
     </div>
 

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)) {
@@ -27,6 +28,7 @@ const textarea = document.getElementById('chatTextarea');
 const sendButton = document.getElementById('sendButton');
 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 +55,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 +126,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 +162,32 @@ document.addEventListener("DOMContentLoaded", function () {
         }
     });
 });
+
+// Function to get the user sessions
+document.addEventListener("DOMContentLoaded", function () {
+    const sessionsButton = document.getElementById('loadSessionsButton');
+    const sessions = document.getElementById('userSessions');
+
+    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 = JSON.parse(await response.text());
+            sessions.innerHTML = '';
+            data.forEach(item => {
+                const listItem = document.createElement('li');
+                const button = document.createElement('button');
+                button.textContent = item;
+                button.addEventListener("click", function () { alert(`You clicked on ${item}`); });
+                listItem.appendChild(button);
+                sessions.appendChild(listItem);
+            });
+        } catch (error) {
+            sessions.innerHTML = 'Error: ' + error.message;
+        }
+    }
+    );
+}
+);

src/hellocomputer/users.py

@@ -1,11 +1,12 @@
 import json
 import os
+from datetime import datetime
 from pathlib import Path
+from typing import List
 from uuid import UUID, uuid4
 
 import duckdb
 import polars as pl
-from datetime import datetime
 
 from .db import DDB, StorageEngines
 
@@ -79,7 +80,7 @@ class OwnershipDB(DDB):
               '{sid}' as sid,
               '{now}' as timestamp
           )
-        TO '{self.path_prefix}/{record_id}.csv' (FORMAT JSON)"""
+        TO '{self.path_prefix}/{record_id}.csv'"""
 
         try:
             self.db.sql(query)
@@ -88,3 +89,25 @@ class OwnershipDB(DDB):
             self.db.sql(query)
 
         return sid
+
+    def sessions(self, user_email: str) -> List[str]:
+        try:
+            return (
+                self.db.sql(f"""
+            SELECT
+                sid
+            FROM
+                '{self.path_prefix}/*.csv'
+            WHERE
+                email = '{user_email}
+            ORDER BY
+                timestamp ASC
+            LIMIT 10'
+        """)
+                .pl()
+                .to_series()
+                .to_list()
+            )
+        # If the table does not exist
+        except duckdb.duckdb.IOException:
+            return []

test/test_data.py

@@ -1,8 +1,8 @@
 from pathlib import Path
 
 import hellocomputer
-from hellocomputer.sessions import SessionDB
 from hellocomputer.db import StorageEngines
+from hellocomputer.sessions import SessionDB
 
 TEST_STORAGE = StorageEngines.local
 TEST_XLS_PATH = (

test/test_query.py

@@ -2,11 +2,11 @@ from pathlib import Path
 
 import hellocomputer
 import pytest
-from hellocomputer.sessions import SessionDB
 from hellocomputer.config import settings
 from hellocomputer.db import StorageEngines
 from hellocomputer.extraction import extract_code_block
 from hellocomputer.models import Chat
+from hellocomputer.sessions import SessionDB
 
 TEST_XLS_PATH = (
     Path(hellocomputer.__file__).parents[2]

test/test_user.py

@@ -2,7 +2,7 @@ from pathlib import Path
 
 import hellocomputer
 from hellocomputer.db import StorageEngines
-from hellocomputer.users import UserDB, OwnershipDB
+from hellocomputer.users import OwnershipDB, UserDB
 
 TEST_STORAGE = StorageEngines.local
 TEST_OUTPUT_FOLDER = Path(hellocomputer.__file__).parents[2] / "test" / "output"
@@ -28,7 +28,13 @@ def test_user_exists():
 def test_assign_owner():
     assert (
         OwnershipDB(storage_engine=TEST_STORAGE, path=TEST_OUTPUT_FOLDER).set_ownersip(
-            "something.something@something", "1234", "test"
+            "something.something@something", "testsession", "test"
         )
-        == "1234"
+        == "testsession"
     )
+
+
+def test_get_sessions():
+    assert OwnershipDB(storage_engine=TEST_STORAGE, path=TEST_OUTPUT_FOLDER).sessions(
+        "something.something@something"
+    ) == ["testsession"]