Merge pull request #8 from Promptly-Technologies-LLC/5-test-and-hook-up-warnings-component

Create setup page for setting the API key and creating an assistant

Author: chriscarrollsmith

.env.example

@@ -1,2 +1,2 @@
-OPENAI_API_KEY=
+OPENAI_API_KEY=
 ASSISTANT_ID=

README.md

@@ -13,29 +13,21 @@ git clone https://github.com/Promptly-Technologies-LLC/openai-assistants-python-
 cd openai-assistants-python-quickstart
 ```
 
-### 2. Set your [OpenAI API key](https://platform.openai.com/api-keys)
-
-```shell
-cp .env.example .env
-```
-
-### 3. Install dependencies
+### 2. Install dependencies
 
 ```shell
 uv venv
 uv pip install -r pyproject.toml
 ```
 
-### 4. Create an assistant
+### 3. Run the FastAPI server
 
 ```shell
-uv run create_assistant.py
+uv run uvicorn main:app --host 0.0.0.0 --port 8000 --reload
 ```
 
-### 5. Run the FastAPI server
+### 4. Navigate to [http://localhost:8000](http://localhost:8000).
 
-```shell
-uv run uvicorn main:app --host 0.0.0.0 --port 8000 --reload
-```
+### 5. Set your OpenAI API key and create an assistant in the GUI
 
-### 6. Navigate to [http://localhost:8000](http://localhost:8000).
+If your OPENAI_API_KEY or ASSISTANT_ID are not set, you will be redirected to the setup page where you can set them. (The values will be saved in a .env file in the root of the project.) Once set, you will be redirected to the home page and can begin a chat session.

__pycache__/main.cpython-312.pyc

@@ -1,17 +1,16 @@
 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, api_keys, assistants
 
 
 logger = logging.getLogger("uvicorn.error")
 
-dotenv.load_dotenv()
-
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Optional startup logic
@@ -22,9 +21,10 @@ 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)
+app.include_router(api_keys.router)
+app.include_router(assistants.router)
 
 # Mount static files (e.g., CSS, JS)
 app.mount("/static", StaticFiles(directory=os.path.join(os.getcwd(), "static")), name="static")
@@ -36,6 +36,11 @@ async def lifespan(app: FastAPI):
 async def read_home(request: Request):
     logger.info("Home page requested")
 
+    # Check if environment variables are missing
+    load_dotenv(override=True)
+    if not os.getenv("OPENAI_API_KEY") or not os.getenv("ASSISTANT_ID"):
+        return RedirectResponse(url="/setup")
+    
     categories = {
         "Basic chat": "basic-chat",
         "File search": "file-search",
@@ -107,6 +112,23 @@ async def read_all(request: Request, messages: list = [], thread_id: str = None)
         }
     )
 
+# Add new setup route
+@app.get("/setup")
+async def read_setup(request: Request, message: str = None):
+    # Check if assistant ID is missing
+    load_dotenv(override=True)
+    if not os.getenv("OPENAI_API_KEY"):
+        message="OpenAI API key is missing."
+    elif not os.getenv("ASSISTANT_ID"):
+        message="Assistant ID is missing."
+    else:
+        message="All set up!"
+    
+    return templates.TemplateResponse(
+        "setup.html",
+        {"request": request, "message": message}
+    )
+
 if __name__ == "__main__":
     import uvicorn
 

main.py

@@ -0,0 +1,32 @@
+import logging
+from fastapi import APIRouter, HTTPException, Form
+from fastapi.responses import RedirectResponse
+from utils.create_assistant import update_env_file
+
+# Configure logger
+logger: logging.Logger = logging.getLogger("uvicorn.error")
+
+router = APIRouter(prefix="/api-keys", tags=["API Keys"])
+
+@router.post("/set")
+async def set_openai_api_key(api_key: str = Form()):
+    """
+    Set the OpenAI API key in the application's environment variables.
+    
+    Args:
+        api_key: OpenAI API key received from form submission
+    
+    Returns:
+        RedirectResponse: Redirects to home page on success
+    
+    Raises:
+        HTTPException: If there's an error updating the environment file
+    """
+    try:
+        update_env_file("OPENAI_API_KEY", api_key, logger)
+        return RedirectResponse(url="/", status_code=303)
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Failed to update API key: {str(e)}"
+        )

routers/api_keys.py

@@ -0,0 +1,38 @@
+from fastapi import APIRouter, Depends
+from fastapi.responses import RedirectResponse
+from openai import AsyncOpenAI
+from utils.create_assistant import create_or_update_assistant, request
+import logging
+import os
+from dotenv import load_dotenv
+
+# Configure logger
+logger = logging.getLogger("uvicorn.error")
+
+# Load environment variables
+load_dotenv()
+
+router = APIRouter(
+    prefix="/assistants",
+    tags=["assistants"]
+)
+
+
+@router.post("/create-update")
+async def create_update_assistant(
+    client: AsyncOpenAI = Depends(lambda: AsyncOpenAI())
+):
+    """
+    Create a new assistant or update an existing one.
+    Returns the assistant ID and status of the operation.
+    """
+    assistant_id = os.getenv("ASSISTANT_ID")
+
+    assistant_id: str = await create_or_update_assistant(
+        client=client,
+        assistant_id=assistant_id,
+        request=request,
+        logger=logger
+    )
+
+    return RedirectResponse(url="/", status_code=303)

routers/assistants.py

@@ -452,3 +452,42 @@ pre {
   color: #b8b8b8;
   margin-right: 8px;
 }
+
+.setupContainer {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  gap: 24px;
+  max-width: 600px;
+  width: 100%;
+  padding: 40px 20px;
+}
+
+.setupForm {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  gap: 20px;
+  width: 100%;
+}
+
+.inputWrapper {
+  margin-bottom: 20px;
+  width: 100%;
+  max-width: 400px;
+  display: flex;
+  justify-content: center;
+}
+
+.setupForm form {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  width: 100%;
+}
+
+.setupMessage {
+  font-size: 1.1em;
+  text-align: center;
+  margin-bottom: 8px;
+}