Update database.py

database.py

@@ -11,7 +11,6 @@ class Database:
         self.pool: Optional[asyncpg.Pool] = None
 
     async def connect(self) -> None:
-        # Render free tier: 512 MB RAM, keep pool small
         self.pool = await asyncpg.create_pool(
             dsn=config.DATABASE_URL,
             min_size=1,
@@ -43,7 +42,7 @@ class Database:
                     updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
                 )
             """)
-              await conn.execute("""
+            await conn.execute("""
                 CREATE TABLE IF NOT EXISTS messages (
                     id SERIAL PRIMARY KEY,
                     user_id BIGINT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
@@ -71,7 +70,6 @@ class Database:
                     updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
                 )
             """)
-
             logger.info("Database tables created/verified")
 
     async def upsert_user(
@@ -108,7 +106,6 @@ class Database:
                 ORDER BY created_at DESC
                 LIMIT $2
             """, user_id, limit)
-            # Reverse to chronological order for the LLM
             return [
                 {"role": r["role"], "content": r["content"], "created_at": r["created_at"]}
                 for r in reversed(rows)
@@ -153,10 +150,11 @@ class Database:
 
     async def count_unsummarized(self, user_id: int) -> int:
         async with self._acquire() as conn:
-            return await conn.fetchval("""
+            val = await conn.fetchval("""
                 SELECT COUNT(*) FROM messages
                 WHERE user_id = $1 AND is_summarized = FALSE
-            """, user_id) or 0
+            """, user_id)
+            return val or 0
 
     async def clear_history(self, user_id: int) -> int:
         async with self._acquire() as conn:
@@ -173,7 +171,6 @@ class Database:
             logger.info("Cleared %d messages and summary for user %s", count, user_id)
             return count
 
-
     async def get_stats(self, user_id: int) -> Dict[str, Any]:
         async with self._acquire() as conn:
             user_count = await conn.fetchval("SELECT COUNT(*) FROM users")
@@ -189,4 +186,3 @@ class Database:
 
 
 db = Database()
-