feature: show full stack trace on server error

imaginairy/http/app.py

@@ -1,18 +1,24 @@
+import logging
 import os.path
+import sys
+import traceback
 from asyncio import Lock
 
-from fastapi import FastAPI, Query
+from fastapi import FastAPI, Query, Request
 from fastapi.concurrency import run_in_threadpool
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import StreamingResponse
+from fastapi.responses import JSONResponse, StreamingResponse
 from fastapi.staticfiles import StaticFiles
 
 from imaginairy.http.models import ImagineWebPrompt
 from imaginairy.http.stablestudio import routes
 from imaginairy.http.utils import generate_image
 
+logger = logging.getLogger(__name__)
+
 gpu_lock = Lock()
 
+
 app = FastAPI()
 
 app.add_middleware(
@@ -48,3 +54,13 @@ static_folder = os.path.dirname(os.path.abspath(__file__)) + "/stablestudio/dist
 print(f"static_folder: {static_folder}")
 
 app.mount("/", StaticFiles(directory=static_folder, html=True), name="static")
+
+
+@app.exception_handler(Exception)
+async def exception_handler(request: Request, exc: Exception):
+    print(f"Unhandled error: {exc}", file=sys.stderr)
+    traceback.print_exc(file=sys.stderr)
+    return JSONResponse(
+        status_code=500,
+        content={"message": "Internal Server Error"},
+    )