File size: 5,610 Bytes
36cbcc0 29f247e 36cbcc0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 | """FastAPI application initialization"""
from fastapi import FastAPI
from fastapi.responses import HTMLResponse, FileResponse
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager
from pathlib import Path
from .core.config import config
from .core.database import Database
from .services.flow_client import FlowClient
from .services.proxy_manager import ProxyManager
from .services.token_manager import TokenManager
from .services.load_balancer import LoadBalancer
from .services.concurrency_manager import ConcurrencyManager
from .services.generation_handler import GenerationHandler
from .api import routes, admin
@asynccontextmanager
async def lifespan(app: FastAPI):
"""Application lifespan manager"""
# Startup
print("=" * 60)
print("Flow2API Starting...")
print("=" * 60)
# Get config from setting.toml
config_dict = config.get_raw_config()
# Check if database exists (determine if first startup)
is_first_startup = not db.db_exists()
# Initialize database tables structure
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 config from database
admin_config = await db.get_admin_config()
if admin_config:
config.set_admin_username_from_db(admin_config.username)
config.set_admin_password_from_db(admin_config.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 "")
# 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 debug configuration from database
debug_config = await db.get_debug_config()
config.set_debug_enabled(debug_config.enabled)
# Initialize concurrency manager
tokens = await token_manager.get_all_tokens()
await concurrency_manager.initialize(tokens)
# Start file cache cleanup task
await generation_handler.file_cache.start_cleanup_task()
print(f"β Database initialized")
print(f"β Total tokens: {len(tokens)}")
print(f"β Cache: {'Enabled' if config.cache_enabled else 'Disabled'} (timeout: {config.cache_timeout}s)")
print(f"β File cache cleanup task started")
print(f"β Server running on http://{config.server_host}:{config.server_port}")
print("=" * 60)
yield
# Shutdown
print("Flow2API Shutting down...")
# Stop file cache cleanup task
await generation_handler.file_cache.stop_cleanup_task()
print("β File cache cleanup task stopped")
# Initialize components
db = Database()
proxy_manager = ProxyManager(db)
flow_client = FlowClient(proxy_manager)
token_manager = TokenManager(db, flow_client)
concurrency_manager = ConcurrencyManager()
load_balancer = LoadBalancer(token_manager, concurrency_manager)
generation_handler = GenerationHandler(
flow_client,
token_manager,
load_balancer,
db,
concurrency_manager,
proxy_manager # ζ·»ε proxy_manager εζ°
)
# Set dependencies
routes.set_generation_handler(generation_handler)
admin.set_dependencies(token_manager, proxy_manager, db)
# Create FastAPI app
app = FastAPI(
title="Flow2API",
description="OpenAI-compatible API for Google VideoFX (Veo)",
version="1.0.0",
lifespan=lifespan
)
# CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# Include routers
app.include_router(routes.router)
app.include_router(admin.router)
# Static files - serve tmp directory for cached files
tmp_dir = Path(__file__).parent.parent / "tmp"
tmp_dir.mkdir(exist_ok=True)
app.mount("/tmp", StaticFiles(directory=str(tmp_dir)), name="tmp")
# HTML routes for frontend
static_path = Path(__file__).parent.parent / "static"
@app.get("/", response_class=HTMLResponse)
async def index():
"""Redirect to login page"""
login_file = static_path / "login.html"
if login_file.exists():
return FileResponse(str(login_file))
return HTMLResponse(content="<h1>Flow2API</h1><p>Frontend not found</p>", status_code=404)
@app.get("/login", response_class=HTMLResponse)
async def login_page():
"""Login page"""
login_file = static_path / "login.html"
if login_file.exists():
return FileResponse(str(login_file))
return HTMLResponse(content="<h1>Login Page Not Found</h1>", status_code=404)
@app.get("/manage", response_class=HTMLResponse)
async def manage_page():
"""Management console page"""
manage_file = static_path / "manage.html"
if manage_file.exists():
return FileResponse(str(manage_file))
return HTMLResponse(content="<h1>Management Page Not Found</h1>", status_code=404)
|