neon setup

py_backend/alembic.ini

@@ -1,3 +1,3 @@
 [alembic]
 script_location = alembic
-sqlalchemy.url = postgresql://promptaid:promptaid@localhost:5433/promptaid
+sqlalchemy.url = ${ALEMBIC_DATABASE_URL}

py_backend/alembic/env.py

@@ -1,36 +1,67 @@
 import os
 import sys
 from dotenv import load_dotenv
+
+# Load local .env when running migrations locally
 load_dotenv()
 
+# Allow "from app import ..." imports
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
 from alembic import context
-from sqlalchemy import engine_from_config, pool
+from sqlalchemy import create_engine, pool
 from app.models import Base
 
-target_metadata = Base.metadata
 config = context.config
+target_metadata = Base.metadata
+
+
+def _get_db_url() -> str:
+    """
+    Prefer a dedicated migration URL; otherwise use the app URL.
+    Only adds sslmode=require for remote connections (not localhost).
+    """
+    url = os.getenv("ALEMBIC_DATABASE_URL") or os.getenv("DATABASE_URL")
+    if not url:
+        raise RuntimeError("Set ALEMBIC_DATABASE_URL or DATABASE_URL for Alembic migrations.")
+
+    # Only add sslmode=require for remote connections, not localhost
+    if "sslmode=" not in url and "localhost" not in url and "127.0.0.1" not in url:
+        url = f"{url}{'&' if '?' in url else '?'}sslmode=require"
+    return url
 
-def run_migrations_offline():
-    url = config.get_main_option("sqlalchemy.url")
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode."""
+    url = _get_db_url()
     context.configure(
-        url=url, target_metadata=target_metadata, literal_binds=True, dialect_opts={"paramstyle": "named"}
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+        compare_type=True,
+        compare_server_default=True,
     )
     with context.begin_transaction():
         context.run_migrations()
 
-def run_migrations_online():
-    connectable = engine_from_config(
-        config.get_section(config.config_ini_section),
-        prefix='sqlalchemy.',
-        poolclass=pool.NullPool,
-    )
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode."""
+    url = _get_db_url()
+    connectable = create_engine(url, poolclass=pool.NullPool, future=True)
+
     with connectable.connect() as connection:
-        context.configure(connection=connection, target_metadata=target_metadata)
+        context.configure(
+            connection=connection,
+            target_metadata=target_metadata,
+            compare_type=True,
+            compare_server_default=True,
+        )
         with context.begin_transaction():
             context.run_migrations()
 
+
 if context.is_offline_mode():
     run_migrations_offline()
 else:

py_backend/app/config.py

@@ -17,5 +17,7 @@ class Settings(BaseSettings):
     class Config:
         env_file = ".env"
         env_file_encoding = "utf-8-sig"
+        extra = "ignore"
+        case_sensitive = False
 
 settings = Settings()

py_backend/app/database.py

@@ -1,27 +1,35 @@
 import os
 import logging
-
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, declarative_base
+from sqlalchemy.engine import make_url
+# from sqlalchemy.pool import NullPool  # optional for serverless
 
 from .config import settings
 
 raw_db_url = settings.DATABASE_URL
-clean_db_url = raw_db_url.split("?", 1)[0]
-print(f"database: {clean_db_url.split('@')[-1].split('/')[-1]}")
+
+# Only add sslmode=require for remote connections, not localhost
+if "sslmode=" not in raw_db_url and "localhost" not in raw_db_url and "127.0.0.1" not in raw_db_url:
+    raw_db_url = f"{raw_db_url}{'&' if '?' in raw_db_url else '?'}sslmode=require"
+
+safe_url = make_url(raw_db_url).set(password="***")
+print(f"database url: {safe_url}")  
 
 engine = create_engine(
-    clean_db_url,
-    echo=True,
-    future=True,
+    raw_db_url,            
     pool_pre_ping=True,
+    pool_recycle=300,        
+    # poolclass=NullPool,     
+    # echo=True,              
+    future=True,
 )
 
 SessionLocal = sessionmaker(
     autocommit=False,
     autoflush=False,
     bind=engine,
-    future=True
+    future=True,
 )
 
 Base = declarative_base()

py_backend/entrypoint.sh

@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+set -e
+
+# show the DB host (redacts password automatically in our echo)
+echo "Running alembic upgrade head..."
+alembic upgrade head || { echo "Alembic failed"; exit 1; }
+
+# start app (use the port Spaces injects)
+exec uvicorn app.main:app --host 0.0.0.0 --port ${PORT:-7860}