Create app.py

app.py

@@ -0,0 +1,158 @@
+import time
+import logging
+import uvicorn
+from contextlib import asynccontextmanager
+from fastapi import FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+
+from models import (
+    ImageGenerationRequest, 
+    ImageGenerationResponse, 
+    ErrorResponse,
+    ModelsResponse,
+    ModelInfo
+)
+from image_generator import image_generator
+from config import config
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Application lifespan management"""
+    logger.info("Starting TTI Frame API...")
+    yield
+    logger.info("Shutting down TTI Frame API...")
+    image_generator.cleanup()
+
+
+# Create FastAPI app
+app = FastAPI(
+    title="TTI Frame - OpenAI Compatible Text-to-Image API",
+    description="A FastAPI wrapper providing OpenAI-compatible endpoints for text-to-image generation",
+    version="1.0.0",
+    lifespan=lifespan
+)
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Configure as needed
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Mount static files for serving images
+app.mount("/images", StaticFiles(directory=config.OUTPUT_DIR), name="images")
+
+
+@app.get("/")
+async def root():
+    """Root endpoint"""
+    return {
+        "message": "TTI Frame - OpenAI Compatible Text-to-Image API",
+        "version": "1.0.0",
+        "docs": "/docs"
+    }
+
+
+@app.get("/v1/models", response_model=ModelsResponse)
+async def list_models():
+    """List available models (OpenAI compatible)"""
+    models = [
+        ModelInfo(
+            id="dall-e-3",
+            created=1677649963,
+            owned_by="tti-frame"
+        ),
+        ModelInfo(
+            id="dall-e-2", 
+            created=1677649963,
+            owned_by="tti-frame"
+        )
+    ]
+    
+    return ModelsResponse(data=models)
+
+
+@app.post("/v1/images/generations", response_model=ImageGenerationResponse)
+async def create_image(request: ImageGenerationRequest):
+    """
+    Generate images from text prompts (OpenAI compatible)
+    
+    Creates images based on a text prompt using advanced diffusion models.
+    Supports various sizes, qualities, and response formats.
+    """
+    try:
+        logger.info(f"Received image generation request: {request.prompt[:50]}...")
+        
+        # Validate request
+        if not request.prompt or not request.prompt.strip():
+            raise HTTPException(
+                status_code=400, 
+                detail="Prompt cannot be empty"
+            )
+        
+        if len(request.prompt) > 4000:
+            raise HTTPException(
+                status_code=400,
+                detail="Prompt too long. Maximum 4000 characters allowed."
+            )
+        
+        # Generate images
+        image_data = await image_generator.generate_images(request)
+        
+        response = ImageGenerationResponse(
+            created=int(time.time()),
+            data=image_data
+        )
+        
+        logger.info(f"Successfully generated {len(image_data)} images")
+        return response
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Image generation failed: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail=f"Image generation failed: {str(e)}"
+        )
+
+
+@app.get("/health")
+async def health_check():
+    """Health check endpoint"""
+    return {"status": "healthy", "timestamp": int(time.time())}
+
+
+@app.exception_handler(Exception)
+async def global_exception_handler(request, exc):
+    """Global exception handler"""
+    logger.error(f"Unhandled exception: {exc}")
+    return JSONResponse(
+        status_code=500,
+        content=ErrorResponse(
+            error={
+                "message": "Internal server error",
+                "type": "server_error",
+                "code": "internal_error"
+            }
+        ).dict()
+    )
+
+
+if __name__ == "__main__":
+    uvicorn.run(
+        "main:app",
+        host=config.HOST,
+        port=config.PORT,
+        reload=True,
+        log_level="info"
+    )