Update migration environment and run script: import additional models in env.py, enhance run.sh to create storage directories and initialize SQLite database with alembic migrations.

migrations/env.py

@@ -14,7 +14,8 @@ if app_dir not in sys.path:
 
 # Import your models here
 try:
-    from app.models.database import Base
+    from app.models import Base
+    from app.models.models import Audiobook, TextChunk  # Import models to register them
 except ImportError as e:
     raise Exception(
         f"Failed to import models. PYTHONPATH: {os.environ.get('PYTHONPATH')}, "

scripts/run.sh

@@ -6,6 +6,32 @@ export PORT=${PORT:-7860}
 export HOST=${HOST:-"0.0.0.0"}
 export WORKERS=${WORKERS:-1}
 export LOG_LEVEL=${LOG_LEVEL:-"info"}
+export DATABASE_URL=${DATABASE_URL:-"sqlite:///app/storage/audiobooks.db"}
+
+# Create storage directories if they don't exist
+mkdir -p /app/storage/audio
+mkdir -p /app/storage/text
+mkdir -p /app/storage/temp
+
+# Initialize database if using SQLite
+if [[ "$DATABASE_URL" == sqlite* ]]; then
+    echo "Using SQLite database at: $DATABASE_URL"
+    # Create the database directory if it doesn't exist
+    DB_DIR=$(dirname "${DATABASE_URL#sqlite:///}")
+    mkdir -p "$DB_DIR"
+    
+    # Initialize alembic if not already initialized
+    if [ ! -d "migrations" ]; then
+        echo "Initializing alembic..."
+        alembic init migrations
+    fi
+    
+    # Create initial migration if none exists
+    if [ ! -d "migrations/versions" ] || [ -z "$(ls -A migrations/versions)" ]; then
+        echo "Creating initial migration..."
+        alembic revision --autogenerate -m "Initial migration"
+    fi
+fi
 
 # Wait for database to be ready (if using external database)
 if [ ! -z "$DATABASE_URL" ] && [[ "$DATABASE_URL" != sqlite* ]]; then
@@ -21,12 +47,10 @@ fi
 
 # Run database migrations
 echo "Running database migrations..."
-alembic upgrade head
-
-# Create necessary directories if they don't exist
-mkdir -p /app/storage/audio
-mkdir -p /app/storage/text
-mkdir -p /app/storage/temp
+alembic upgrade head || {
+    echo "Warning: Database migration failed. This might be expected for first run."
+    echo "Continuing with application startup..."
+}
 
 echo "Starting CSM-1B TTS API server..."
 echo "Host: $HOST"