From 34245ed004a2c0fcc56f41b6576cbd9ead52bae6 Mon Sep 17 00:00:00 2001
From: Guillem Borrell <borrell.guillem@bcg.com>
Date: Tue, 11 Jun 2024 17:49:05 +0200
Subject: [PATCH] Module import and favicon

---
 requirements.in           | 1 +
 requirements.txt          | 2 ++
 src/hellocomputer/main.py | 7 ++++++-
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/requirements.in b/requirements.in
index 031ffd6..640ff7d 100644
--- a/requirements.in
+++ b/requirements.in
@@ -7,6 +7,7 @@ s3fs
 aiofiles
 duckdb
 polars
+pyarrow
 pyjwt[crypto]
 python-multipart
 authlib
diff --git a/requirements.txt b/requirements.txt
index f6c087e..ad5d5c8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -122,6 +122,7 @@ numpy==1.26.4
     # via
     #   langchain
     #   langchain-community
+    #   pyarrow
 openai==1.33.0
 orjson==3.10.4
     # via
@@ -132,6 +133,7 @@ packaging==23.2
     #   langchain-core
     #   marshmallow
 polars==0.20.31
+pyarrow==16.1.0
 pycparser==2.22
     # via cffi
 pydantic==2.7.3
diff --git a/src/hellocomputer/main.py b/src/hellocomputer/main.py
index 4d6bf59..a21da9e 100644
--- a/src/hellocomputer/main.py
+++ b/src/hellocomputer/main.py
@@ -2,7 +2,7 @@ import json
 from pathlib import Path
 
 from fastapi import FastAPI
-from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi.responses import HTMLResponse, RedirectResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from starlette.middleware.sessions import SessionMiddleware
 from starlette.requests import Request
@@ -29,6 +29,11 @@ async def homepage(request: Request):
         return HTMLResponse(f.read())
 
 
+@app.get("/favicon.ico")
+async def favicon():
+    return FileResponse(static_path / "img" / "favicon.ico")
+
+
 app.include_router(health.router)
 app.include_router(sessions.router)
 app.include_router(files.router)