Redirect user to warnings.html if assistant ID or OpenAI API key is missing

Author: chriscarrollsmith

main.py

@@ -1,16 +1,18 @@
 import os
 import logging
-import dotenv
+from dotenv import load_dotenv
 from contextlib import asynccontextmanager
 from fastapi import FastAPI, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-from routers import messages, assistants_files, tool_outputs
+from fastapi.responses import RedirectResponse
+from routers import files, messages, tools
 
 
 logger = logging.getLogger("uvicorn.error")
 
-dotenv.load_dotenv()
+# Get the assistant ID from .env file
+load_dotenv()
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -22,8 +24,8 @@ async def lifespan(app: FastAPI):
 
 # Mount routers
 app.include_router(messages.router)
-app.include_router(assistants_files.router)
-app.include_router(tool_outputs.router)
+app.include_router(files.router)
+app.include_router(tools.router)
 
 
 # Mount static files (e.g., CSS, JS)
@@ -36,6 +38,12 @@ async def lifespan(app: FastAPI):
 async def read_home(request: Request):
     logger.info("Home page requested")
 
+    # Check if assistant ID is missing
+    if not os.getenv("ASSISTANT_ID"):
+        return RedirectResponse(url="/warnings", message="Assistant ID is missing")
+    elif not os.getenv("OPENAI_API_KEY"):
+        return RedirectResponse(url="/warnings", message="OpenAI API key is missing")
+    
     categories = {
         "Basic chat": "basic-chat",
         "File search": "file-search",
@@ -107,6 +115,14 @@ async def read_all(request: Request, messages: list = [], thread_id: str = None)
         }
     )
 
+# Add new warnings route
+@app.get("/warnings")
+async def read_warnings(request: Request):
+    return templates.TemplateResponse(
+        "warnings.html",
+        {"request": request}
+    )
+
 if __name__ == "__main__":
     import uvicorn