Refactor application imports and migration script: update model import paths in env.py and migrate.py, enhance app initialization in __init__.py, and modify PYTHONPATH in run.sh for improved directory structure.

alembic/env.py

@@ -5,7 +5,7 @@ from sqlalchemy import pool
 
 from alembic import context
 
-from app.models import Base
+from app.models.database import Base
 from app.config import get_settings
 
 # this is the Alembic Config object, which provides

app/__init__.py

@@ -1 +1,4 @@
-"""App package.""" 
+"""App package initialization."""
+from app.models import Base
+
+__all__ = ['Base'] 

scripts/migrate.py

@@ -1,8 +1,9 @@
+"""Database migration script."""
 import os
-import sys
 import logging
 from alembic.config import Config
 from alembic import command
+from app.models.database import Base
 
 # Configure logging
 logging.basicConfig(
@@ -32,7 +33,7 @@ def run_migrations():
         
     except Exception as e:
         logger.error(f"Error during database migration: {str(e)}")
-        sys.exit(1)
+        raise
 
 if __name__ == "__main__":
     run_migrations() 

scripts/run.sh

@@ -9,7 +9,7 @@ export LOG_LEVEL=${LOG_LEVEL:-"info"}
 export DATABASE_URL=${DATABASE_URL:-"sqlite:///app/storage/audiobooks.db"}
 
 # Set Python path to include both app directories
-export PYTHONPATH="/app:/app/app:${PYTHONPATH:-}"
+export PYTHONPATH="/app:/app/app:${PYTHONPATH:-}:$(dirname $(dirname $(realpath $0)))"
 
 # Create storage directories if they don't exist
 mkdir -p /app/storage/audio