Updated app.py to query faster

app.py

@@ -5,6 +5,7 @@ from typing import Optional
 import psycopg2
 from dotenv import load_dotenv
 from psycopg2.extras import RealDictCursor
+from psycopg2 import pool
 import numpy as np
 import os
 
@@ -43,15 +44,48 @@ print(f"Loaded untrained: {untrained_features.shape}")
 # Database Connection
 # -------------------------
 
+db_pool = None
+
+@app.on_event("startup")
+def startup_event():
+    global db_pool
+    try:
+        db_pool = pool.SimpleConnectionPool(
+            1, 20,
+            host=os.getenv("DB_HOST"),
+            database=os.getenv("DB_NAME"),
+            user=os.getenv("DB_USER"),
+            password=os.getenv("DB_PASSWORD"),
+            port=6543,
+            cursor_factory=RealDictCursor
+        )
+        print("Database connection pool created")
+    except Exception as e:
+        print(f"Error creating connection pool: {e}")
+
+@app.on_event("shutdown")
+def shutdown_event():
+    global db_pool
+    if db_pool:
+        db_pool.closeall()
+        print("Database connection pool closed")
+
+class PooledConnection:
+    def __init__(self, pool):
+        self.pool = pool
+        self.conn = pool.getconn()
+        
+    def close(self):
+        self.pool.putconn(self.conn)
+        
+    def cursor(self, *args, **kwargs):
+        return self.conn.cursor(*args, **kwargs)
+        
+    def __getattr__(self, name):
+        return getattr(self.conn, name)
+
 def get_db():
-    return psycopg2.connect(
-        host=os.getenv("DB_HOST"),
-        database=os.getenv("DB_NAME"),
-        user=os.getenv("DB_USER"),
-        password=os.getenv("DB_PASSWORD"),
-        port=6543,
-        cursor_factory=RealDictCursor
-    )
+    return PooledConnection(db_pool)
 
 
 # -------------------------
@@ -172,7 +206,7 @@ async def root():
 
 
 @app.get("/test-db")
-async def test_db():
+def test_db():
     try:
         conn = get_db()
         cur = conn.cursor()
@@ -185,7 +219,7 @@ async def test_db():
 
 
 @app.get("/movies/search")
-async def search_movies(q: str, n: int = 10):
+def search_movies(q: str, n: int = 10):
     conn = get_db()
     cur = conn.cursor()
     cur.execute("""
@@ -202,7 +236,7 @@ async def search_movies(q: str, n: int = 10):
 
 
 @app.get("/movies/{movie_id}")
-async def get_movie(movie_id: int):
+def get_movie(movie_id: int):
     conn = get_db()
     cur = conn.cursor()
     cur.execute("""