Update database.py

database.py

@@ -1,19 +1,17 @@
-from sqlalchemy import create_engine, Column, String, Integer, Text, DateTime, ForeignKey, Index
+from sqlalchemy import create_engine, Column, String, Integer, Text, DateTime, ForeignKey
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker, relationship
 from datetime import datetime
 import uuid
 import os
 
-# For HF Spaces: Use environment variable directly (set via Secrets)
-# For local dev: Can still use .env file
-DATABASE_URL = os.getenv("DATABASE_URL")
-
-if not DATABASE_URL:
-    raise ValueError("DATABASE_URL environment variable is not set!")
+# Use SQLite for reliable deployment on HF Spaces
+# The /data directory is persistent in HF Spaces
+DATABASE_URL = "sqlite:////data/chat_history.db"
 
 # Create engine
-engine = create_engine(DATABASE_URL, pool_pre_ping=True)
+# check_same_thread=False is needed for SQLite with FastAPI
+engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()
 
@@ -42,6 +40,12 @@ class Message(Base):
 # Database functions
 def init_db():
     """Create all tables"""
+    # Ensure /data directory exists (for HF Spaces)
+    if DATABASE_URL.startswith("sqlite:////data"):
+        os.makedirs("/data", exist_ok=True)
+    elif DATABASE_URL.startswith("sqlite:///./data"):
+        os.makedirs("./data", exist_ok=True)
+        
     Base.metadata.create_all(bind=engine)
 
 def get_db():