Allow user to customize system prompt from setup.html

Author: chriscarrollsmith

routers/setup.py

@@ -55,6 +55,7 @@ async def read_setup(
     # Variable initializations
     current_tools: List[str] = []
     current_model: Optional[str] = None
+    current_instructions: Optional[str] = None
     # Populate with all models extracted from user-provided HTML, sorted
     available_models: List[str] = sorted([
         "gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-3.5-turbo-1106", "gpt-3.5-turbo-16k", 
@@ -84,6 +85,7 @@ async def read_setup(
                 assistant = await client.beta.assistants.retrieve(assistant_id)
                 current_tools = [tool.type for tool in assistant.tools]
                 current_model = assistant.model  # Get the model from the assistant
+                current_instructions = assistant.instructions # Get instructions
             except Exception as e:
                 logger.error(f"Failed to retrieve assistant {assistant_id}: {e}")
                 # If we can't retrieve the assistant, proceed as if it doesn't exist
@@ -100,6 +102,7 @@ async def read_setup(
             "assistant_id": assistant_id,
             "current_tools": current_tools,
             "current_model": current_model,
+            "current_instructions": current_instructions,
             "available_models": available_models # Pass available models to template
         }
     )
@@ -109,6 +112,7 @@ async def read_setup(
 async def create_update_assistant(
     tool_types: List[ToolTypes] = Form(...),
     model: str = Form(...),
+    instructions: str = Form(...),
     client: AsyncOpenAI = Depends(lambda: AsyncOpenAI())
 ) -> RedirectResponse:
     """
@@ -122,6 +126,7 @@ async def create_update_assistant(
         assistant_id=current_assistant_id,
         tool_types=tool_types,
         model=model,
+        instructions=instructions,
         logger=logger
     )
 

static/styles.css

@@ -173,7 +173,7 @@ pre {
   display: flex;
   flex-direction: column;
   justify-content: center;
-  align-items: center;
+  align-items: flex-start;
   width: 100%;
   height: 100%;
   padding: 20px;
@@ -188,7 +188,7 @@ pre {
   display: flex;
   flex-direction: column;
   gap: 12px;
-  align-items: center;
+  align-items: flex-start;
   width: 100%;
 }
 
@@ -622,7 +622,7 @@ pre {
 .setupSection form {
   display: flex;
   flex-direction: column;
-  align-items: center; /* Center form elements */
+  align-items: flex-start; /* Changed from center to flex-start */
   gap: 20px;
   width: 100%;
 }
@@ -633,7 +633,7 @@ pre {
   flex-direction: column;
   align-items: flex-start; /* Align labels to the left */
   gap: 10px; /* Vertical gap between labels */
-  width: auto; /* Fit content width */
+  width: 100%; /* Changed from auto to 100% */
   margin-bottom: 1rem; /* Restore some margin below the group */
 }
 
@@ -651,6 +651,14 @@ pre {
   margin-bottom: 8px;
 }
 
+#file-list-container {
+    display: flex;
+    flex-direction: column;
+    align-items: flex-start;
+    gap: 20px;
+    width: 100%;
+  }
+
 .fileList {
   list-style: none;
   padding: 0;
@@ -699,3 +707,13 @@ pre {
   margin-left: 10px; 
   font-style: italic;
 }
+
+/* Original selector: .centered-button-container */
+/* New selector increases specificity to override .setupSection form > div */
+.setupSection form .centered-button-container {
+  display: flex;
+  justify-content: center; /* Center button horizontally */
+  align-items: center; /* Center button vertically (good practice) */
+  width: 100%;
+  margin-top: 10px; /* Optional: add some space above the button */
+}

templates/components/file-viewer.html

@@ -8,8 +8,10 @@ <h3>Upload Files</h3>
     enctype="multipart/form-data">
   <input type="file" name="file" required>
   <input type="hidden" name="purpose" value="assistants">
-  <button type="submit" class="button">Upload File</button>
-  <span id="upload-indicator" class="htmx-indicator">Uploading...</span>
+  <div class="centered-button-container">
+    <button type="submit" class="button">Upload File</button>
+    <span id="upload-indicator" class="htmx-indicator">Uploading...</span>
+  </div>
 </form>
 
 <h3>Existing Files</h3>

templates/setup.html

@@ -53,6 +53,18 @@ <h3>Setup Required</h3>
                     </select>
                   </div>
 
+                  <div style="margin-top: 15px;">
+                    <label for="instructions-input">System Prompt:</label>
+                    <textarea 
+                      name="instructions" 
+                      id="instructions-input" 
+                      class="input" 
+                      rows="4" 
+                      placeholder="e.g., You are a helpful assistant."
+                      style="width: 100%; resize: none;"
+                    >{% if current_instructions %}{{ current_instructions }}{% else %}You are a helpful assistant.{% endif %}</textarea>
+                  </div>
+
                   <p class="setupMessage" style="margin-top: 15px;">Select tools for your assistant:</p>
                   <div>
                     <label>
@@ -68,12 +80,14 @@ <h3>Setup Required</h3>
                       Get Weather (Custom Function)
                     </label>
                   </div>
-                  <button 
-                    type="submit"
-                    class="button"
-                  >
-                    {% if assistant_id %}Update{% else %}Create{% endif %} Assistant
-                  </button>
+                  <div class="centered-button-container">
+                    <button 
+                      type="submit"
+                      class="button"
+                    >
+                      {% if assistant_id %}Update{% else %}Create{% endif %} Assistant
+                    </button>
+                  </div>
                 </form>
               </div>
               {% if assistant_id and "file_search" in current_tools %}

utils/create_assistant.py

@@ -70,6 +70,7 @@ async def create_or_update_assistant(
         assistant_id: str | None,
         tool_types: List[ToolTypes],
         model: str,
+        instructions: str,
         logger: logging.Logger
 ) -> str | None:
     """
@@ -82,7 +83,7 @@ async def create_or_update_assistant(
             # Update the existing assistant
             assistant = await client.beta.assistants.update(
                 assistant_id,
-                instructions="You are a helpful assistant.",
+                instructions=instructions,
                 name="Quickstart Assistant",
                 model=model,
                 tools=tool_params
@@ -91,7 +92,7 @@ async def create_or_update_assistant(
         else:
             # Create a new assistant
             assistant = await client.beta.assistants.create(
-                instructions="You are a helpful assistant.",
+                instructions=instructions,
                 name="Quickstart Assistant",
                 model=model,
                 tools=tool_params