Update database.py

database.py

@@ -1,23 +1,19 @@
-from sqlalchemy import create_engine, Column, String, Integer, Text, DateTime, ForeignKey
+from sqlalchemy import create_engine, Column, String, Integer, Text, DateTime, ForeignKey, Index
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker, relationship
 from datetime import datetime
 import uuid
 import os
+from pathlib import Path
+from dotenv import load_dotenv
 
-# Neon DB connection
-DATABASE_URL = os.getenv("DATABASE_URL")
-
-if not DATABASE_URL:
-    print("Warning: DATABASE_URL not set. Using sqlite fallback for local dev only.")
-    DATABASE_URL = "sqlite:///./local_chat.db"
+# Load environment variables
+env_path = Path(__file__).parent / '.env'
+load_dotenv(dotenv_path=env_path)
 
-# Handle "postgres://" to "postgresql://" fix
-if DATABASE_URL and DATABASE_URL.startswith("postgres://"):
-    DATABASE_URL = DATABASE_URL.replace("postgres://", "postgresql://", 1)
+DATABASE_URL = os.getenv("DATABASE_URL")
 
 # Create engine
-# sslmode is usually handled in the connection string itself
 engine = create_engine(DATABASE_URL, pool_pre_ping=True)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()
@@ -55,7 +51,7 @@ def get_db():
     try:
         return db
     finally:
-        db.close()
+        pass
 
 def create_session(user_id: str, name: str = "New Chat"):
     """Create a new chat session"""
@@ -137,3 +133,21 @@ def delete_session(session_id: str):
             db.commit()
     finally:
         db.close()
+
+def update_session_name(session_id: str, name: str):
+    """Update the name of a session"""
+    db = SessionLocal()
+    try:
+        session = db.query(Session).filter(Session.id == session_id).first()
+        if session:
+            session.name = name
+            db.commit()
+            db.refresh(session)
+            return {
+                "id": session.id,
+                "user_id": session.user_id,
+                "name": session.name,
+                "created_at": session.created_at.isoformat() if session.created_at else None
+            }
+    finally:
+        db.close()