"""Main application entry point""" import uvicorn from fastapi import FastAPI from fastapi.responses import FileResponse, HTMLResponse from fastapi.staticfiles import StaticFiles from fastapi.middleware.cors import CORSMiddleware from pathlib import Path from apscheduler.schedulers.asyncio import AsyncIOScheduler from apscheduler.triggers.cron import CronTrigger from datetime import datetime # Import modules from .core.config import config from .core.database import Database from .services.token_manager import TokenManager from .services.proxy_manager import ProxyManager from .services.load_balancer import LoadBalancer from .services.sora_client import SoraClient from .services.generation_handler import GenerationHandler from .services.concurrency_manager import ConcurrencyManager from .api import routes as api_routes from .api import admin as admin_routes # Initialize scheduler (uses system local timezone by default) scheduler = AsyncIOScheduler() # Initialize FastAPI app app = FastAPI( title="Sora2API", description="OpenAI compatible API for Sora", version="1.0.0" ) # CORS middleware app.add_middleware( CORSMiddleware, allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"], ) # Initialize components db = Database() token_manager = TokenManager(db) proxy_manager = ProxyManager(db) concurrency_manager = ConcurrencyManager() load_balancer = LoadBalancer(token_manager, concurrency_manager) sora_client = SoraClient(proxy_manager) generation_handler = GenerationHandler(sora_client, token_manager, load_balancer, db, proxy_manager, concurrency_manager) # Set dependencies for route modules api_routes.set_generation_handler(generation_handler) admin_routes.set_dependencies(token_manager, proxy_manager, db, generation_handler, concurrency_manager, scheduler) # Include routers app.include_router(api_routes.router) app.include_router(admin_routes.router) # Static files static_dir = Path(__file__).parent.parent / "static" static_dir.mkdir(exist_ok=True) app.mount("/static", StaticFiles(directory=str(static_dir)), name="static") # Cache files (tmp directory) tmp_dir = Path(__file__).parent.parent / "tmp" tmp_dir.mkdir(exist_ok=True) app.mount("/tmp", StaticFiles(directory=str(tmp_dir)), name="tmp") # Frontend routes @app.get("/", response_class=HTMLResponse) async def root(): """Redirect to login page""" return """

Redirecting to login...

""" @app.get("/login", response_class=FileResponse) async def login_page(): """Serve login page""" return FileResponse(str(static_dir / "login.html")) @app.get("/manage", response_class=FileResponse) async def manage_page(): """Serve management page""" return FileResponse(str(static_dir / "manage.html")) @app.on_event("startup") async def startup_event(): """Initialize database on startup""" # Get config from setting.toml config_dict = config.get_raw_config() # Check if database exists is_first_startup = not db.db_exists() # Initialize database tables await db.init_db() # Handle database initialization based on startup type if is_first_startup: print("🎉 First startup detected. Initializing database and configuration from setting.toml...") await db.init_config_from_toml(config_dict, is_first_startup=True) print("✓ Database and configuration initialized successfully.") else: print("🔄 Existing database detected. Checking for missing tables and columns...") await db.check_and_migrate_db(config_dict) print("✓ Database migration check completed.") # Load admin credentials and API key from database admin_config = await db.get_admin_config() config.set_admin_username_from_db(admin_config.admin_username) config.set_admin_password_from_db(admin_config.admin_password) config.api_key = admin_config.api_key # Load cache configuration from database cache_config = await db.get_cache_config() config.set_cache_enabled(cache_config.cache_enabled) config.set_cache_timeout(cache_config.cache_timeout) config.set_cache_base_url(cache_config.cache_base_url or "") # Sync cache timeout to file cache instance generation_handler.file_cache.set_timeout(cache_config.cache_timeout) # Load generation configuration from database generation_config = await db.get_generation_config() config.set_image_timeout(generation_config.image_timeout) config.set_video_timeout(generation_config.video_timeout) # Load token refresh configuration from database token_refresh_config = await db.get_token_refresh_config() config.set_at_auto_refresh_enabled(token_refresh_config.at_auto_refresh_enabled) # Load call logic configuration from database call_logic_config = await db.get_call_logic_config() config.set_call_logic_mode(call_logic_config.call_mode) print(f"✓ Call logic mode: {call_logic_config.call_mode}") # Load POW service configuration from database pow_service_config = await db.get_pow_service_config() config.set_pow_service_mode(pow_service_config.mode) config.set_pow_service_server_url(pow_service_config.server_url or "") config.set_pow_service_api_key(pow_service_config.api_key or "") config.set_pow_service_proxy_enabled(pow_service_config.proxy_enabled) config.set_pow_service_proxy_url(pow_service_config.proxy_url or "") print(f"✓ POW service mode: {pow_service_config.mode}") # Initialize concurrency manager with all tokens all_tokens = await db.get_all_tokens() await concurrency_manager.initialize(all_tokens) print(f"✓ Concurrency manager initialized with {len(all_tokens)} tokens") # Start file cache cleanup task await generation_handler.file_cache.start_cleanup_task() # Start token refresh scheduler if enabled if token_refresh_config.at_auto_refresh_enabled: scheduler.add_job( token_manager.batch_refresh_all_tokens, CronTrigger(hour=0, minute=0), # Every day at 00:00 (system local timezone) id='batch_refresh_tokens', name='Batch refresh all tokens', replace_existing=True ) scheduler.start() print("✓ Token auto-refresh scheduler started (daily at 00:00)") else: print("⊘ Token auto-refresh is disabled") @app.on_event("shutdown") async def shutdown_event(): """Cleanup on shutdown""" await generation_handler.file_cache.stop_cleanup_task() if scheduler.running: scheduler.shutdown() if __name__ == "__main__": uvicorn.run( "src.main:app", host=config.server_host, port=config.server_port, reload=False )