dynamic prompt management

frontend/src/pages/AdminPage/AdminPage.module.css

@@ -93,6 +93,28 @@
   margin-bottom: var(--go-ui-spacing-xl);
 }
 
+/* Prompt subsections */
+.promptSubsection {
+  margin-bottom: var(--go-ui-spacing-2xl);
+  padding: var(--go-ui-spacing-lg);
+  background-color: var(--go-ui-color-gray-5);
+  border-radius: var(--go-ui-border-radius-lg);
+  border: var(--go-ui-width-separator-thin) solid var(--go-ui-color-gray-20);
+}
+
+.promptSubsection:last-child {
+  margin-bottom: 0;
+}
+
+.promptSubsectionTitle {
+  font-size: var(--go-ui-font-size-lg);
+  font-weight: var(--go-ui-font-weight-semibold);
+  color: var(--go-ui-color-gray-90);
+  margin: 0 0 var(--go-ui-spacing-lg) 0;
+  padding-bottom: var(--go-ui-spacing-sm);
+  border-bottom: var(--go-ui-width-separator-thin) solid var(--go-ui-color-gray-30);
+}
+
 .modelsTable table {
   width: 100%;
   border-collapse: collapse;

frontend/src/pages/AdminPage/AdminPage.tsx

@@ -33,15 +33,26 @@ export default function AdminPage() {
     p_code: string;
     label: string;
     metadata_instructions?: string;
+    image_type: string;
+    is_active: boolean;
+  }>>([]);
+  
+  const [imageTypes, setImageTypes] = useState<Array<{
+    image_type: string;
+    label: string;
   }>>([]);
   
   // Prompt management state
   const [showEditPromptForm, setShowEditPromptForm] = useState(false);
+  const [showAddPromptForm, setShowAddPromptForm] = useState(false);
+  const [addingPromptType, setAddingPromptType] = useState<'crisis_map' | 'drone_image' | null>(null);
   const [editingPrompt, setEditingPrompt] = useState<any>(null);
   const [newPromptData, setNewPromptData] = useState({
     p_code: '',
     label: '',
-    metadata_instructions: ''
+    metadata_instructions: '',
+    image_type: 'crisis_map',
+    is_active: false
   });
   
      // Model management state
@@ -72,6 +83,7 @@ export default function AdminPage() {
     if (isAuthenticated) {
       fetchModels();
       fetchPrompts();
+      fetchImageTypes();
     }
   }, [isAuthenticated]);
 
@@ -113,12 +125,26 @@ export default function AdminPage() {
       });
   };
 
+  const fetchImageTypes = () => {
+    fetch('/api/image-types')
+      .then(r => r.json())
+      .then(imageTypesData => {
+        console.log('Image types data received:', imageTypesData);
+        setImageTypes(imageTypesData || []);
+      })
+      .catch(() => {
+        // Handle error silently
+      });
+  };
+
   const handleEditPrompt = (prompt: any) => {
     setEditingPrompt(prompt);
     setNewPromptData({
       p_code: prompt.p_code,
       label: prompt.label || '',
-      metadata_instructions: prompt.metadata_instructions || ''
+      metadata_instructions: prompt.metadata_instructions || '',
+      image_type: prompt.image_type || 'crisis_map',
+      is_active: prompt.is_active || false
     });
     setShowEditPromptForm(true);
   };
@@ -132,7 +158,9 @@ export default function AdminPage() {
         },
         body: JSON.stringify({
           label: newPromptData.label,
-          metadata_instructions: newPromptData.metadata_instructions
+          metadata_instructions: newPromptData.metadata_instructions,
+          image_type: newPromptData.image_type,
+          is_active: newPromptData.is_active
         }),
       });
 
@@ -141,7 +169,7 @@ export default function AdminPage() {
         fetchPrompts();
         setShowEditPromptForm(false);
         setEditingPrompt(null);
-        setNewPromptData({ p_code: '', label: '', metadata_instructions: '' });
+        setNewPromptData({ p_code: '', label: '', metadata_instructions: '', image_type: 'crisis_map', is_active: false });
       } else {
         const errorData = await response.json();
         alert(`Failed to update prompt: ${errorData.error || 'Unknown error'}`);
@@ -151,6 +179,70 @@ export default function AdminPage() {
     }
   };
 
+  const handleTogglePromptActive = async (promptCode: string, imageType: string) => {
+    try {
+      const response = await fetch(`/api/prompts/${promptCode}/toggle-active?image_type=${imageType}`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+      });
+
+      if (response.ok) {
+        // Refresh prompts to show updated status
+        fetchPrompts();
+      } else {
+        const errorData = await response.json();
+        alert(`Failed to toggle prompt active status: ${errorData.detail || 'Unknown error'}`);
+      }
+    } catch (error) {
+      alert('Error toggling prompt active status');
+    }
+  };
+
+  const handleAddPrompt = (imageType: 'crisis_map' | 'drone_image') => {
+    setAddingPromptType(imageType);
+    setNewPromptData({
+      p_code: '',
+      label: '',
+      metadata_instructions: '',
+      image_type: imageType,
+      is_active: false
+    });
+    setShowAddPromptForm(true);
+  };
+
+  const handleSaveNewPrompt = async () => {
+    try {
+      const response = await fetch('/api/prompts', {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify(newPromptData),
+      });
+
+      if (response.ok) {
+        // Refresh prompts and close form
+        fetchPrompts();
+        setShowAddPromptForm(false);
+        setAddingPromptType(null);
+        setNewPromptData({
+          p_code: '',
+          label: '',
+          metadata_instructions: '',
+          image_type: 'crisis_map',
+          is_active: false
+        });
+      } else {
+        const errorData = await response.json();
+        alert(`Failed to create prompt: ${errorData.detail || 'Unknown error'}`);
+      }
+    } catch (error) {
+      alert(`Error creating prompt: ${error}`);
+    }
+  };
+
   const toggleModelAvailability = async (modelCode: string, currentStatus: boolean) => {
     try {
       const response = await fetch(`/api/models/${modelCode}/toggle`, {
@@ -731,72 +823,159 @@ Model "${newModelData.label}" added successfully!
                    </div>
                  </Container>
 
-                 {/* Prompts Section */}
+                 {/* Prompts Management Section */}
                  <Container
-                   heading="Prompts Management"
+                   heading="Prompt Management"
                    headingLevel={2}
                    withHeaderBorder
                    withInternalPadding
                  >
                    <div className={styles.modelManagementArea}>
-                     {/* Prompts Table */}
-                     <div className={styles.modelsTable}>
-                       <table>
-                         <thead>
-                           <tr>
-                             <th>Code</th>
-                             <th>Label</th>
-                             <th>Actions</th>
-                           </tr>
-                         </thead>
-                         <tbody>
-                           {availablePrompts.map(prompt => (
-                             <tr key={prompt.p_code}>
-                               <td className={styles.modelCode}>{prompt.p_code}</td>
-                               <td className={styles.promptLabel}>{prompt.label || 'No label'}</td>
-                               <td>
-                                 <div className={styles.modelActions}>
+                     
+                     {/* Crisis Maps Sub-section */}
+                     <div className={styles.promptSubsection}>
+                       <h4 className={styles.promptSubsectionTitle}>Crisis Maps</h4>
+                       <div className={styles.modelsTable}>
+                         <table>
+                           <thead>
+                             <tr>
+                               <th>Code</th>
+                               <th>Label</th>
+                               <th>Status</th>
+                               <th>Actions</th>
+                             </tr>
+                           </thead>
+                           <tbody>
+                             {availablePrompts
+                               .filter(prompt => prompt.image_type === 'crisis_map')
+                               .map(prompt => (
+                               <tr key={prompt.p_code}>
+                                 <td className={styles.modelCode}>{prompt.p_code}</td>
+                                 <td className={styles.promptLabel}>{prompt.label || 'No label'}</td>
+                                 <td>
                                    <Button
-                                     name={`view-${prompt.p_code}`}
-                                     variant="secondary"
+                                     name={`toggle-crisis-${prompt.p_code}`}
+                                     variant={prompt.is_active ? "primary" : "secondary"}
                                      size={1}
-                                     onClick={() => {
-                                       setTestResults(`=== Prompt Details ===\nCode: ${prompt.p_code}\nLabel: ${prompt.label}\n\nMetadata Instructions:\n${prompt.metadata_instructions || 'No instructions available'}`);
-                                       setTestResultsTitle(`Prompt: ${prompt.p_code}`);
-                                       setShowTestResultsModal(true);
-                                     }}
+                                     onClick={() => handleTogglePromptActive(prompt.p_code, 'crisis_map')}
                                    >
-                                     View
+                                     {prompt.is_active ? 'Active' : 'Inactive'}
                                    </Button>
+                                 </td>
+                                 <td>
+                                   <div className={styles.modelActions}>
+                                     <Button
+                                       name={`view-${prompt.p_code}`}
+                                       variant="secondary"
+                                       size={1}
+                                       onClick={() => {
+                                         setTestResults(`=== Prompt Details ===\nCode: ${prompt.p_code}\nLabel: ${prompt.label}\nImage Type: ${prompt.image_type}\nActive: ${prompt.is_active}\n\nMetadata Instructions:\n${prompt.metadata_instructions || 'No instructions available'}`);
+                                         setTestResultsTitle(`Prompt: ${prompt.p_code}`);
+                                         setShowTestResultsModal(true);
+                                       }}
+                                     >
+                                       View
+                                     </Button>
+                                     <Button
+                                       name={`edit-${prompt.p_code}`}
+                                       variant="secondary"
+                                       size={1}
+                                       onClick={() => handleEditPrompt(prompt)}
+                                     >
+                                       Edit
+                                     </Button>
+                                   </div>
+                                 </td>
+                               </tr>
+                             ))}
+                           </tbody>
+                         </table>
+                       </div>
+
+                       {/* Add Crisis Map Prompt Button */}
+                       <div className={styles.addModelButtonContainer}>
+                         <Button
+                           name="add-crisis-prompt"
+                           variant="primary"
+                           onClick={() => handleAddPrompt('crisis_map')}
+                         >
+                           Add New Crisis Map Prompt
+                         </Button>
+                       </div>
+                     </div>
+
+                     {/* Drone Images Sub-section */}
+                     <div className={styles.promptSubsection}>
+                       <h4 className={styles.promptSubsectionTitle}>Drone Images</h4>
+                       <div className={styles.modelsTable}>
+                         <table>
+                           <thead>
+                             <tr>
+                               <th>Code</th>
+                               <th>Label</th>
+                               <th>Status</th>
+                               <th>Actions</th>
+                             </tr>
+                           </thead>
+                           <tbody>
+                             {availablePrompts
+                               .filter(prompt => prompt.image_type === 'drone_image')
+                               .map(prompt => (
+                               <tr key={prompt.p_code}>
+                                 <td className={styles.modelCode}>{prompt.p_code}</td>
+                                 <td className={styles.promptLabel}>{prompt.label || 'No label'}</td>
+                                 <td>
                                    <Button
-                                     name={`edit-${prompt.p_code}`}
-                                     variant="secondary"
+                                     name={`toggle-drone-${prompt.p_code}`}
+                                     variant={prompt.is_active ? "primary" : "secondary"}
                                      size={1}
-                                     onClick={() => handleEditPrompt(prompt)}
+                                     onClick={() => handleTogglePromptActive(prompt.p_code, 'drone_image')}
                                    >
-                                     Edit
+                                     {prompt.is_active ? 'Active' : 'Inactive'}
                                    </Button>
-                                 </div>
-                               </td>
-                             </tr>
-                           ))}
-                         </tbody>
-                       </table>
-                     </div>
+                                 </td>
+                                 <td>
+                                   <div className={styles.modelActions}>
+                                     <Button
+                                       name={`view-${prompt.p_code}`}
+                                       variant="secondary"
+                                       size={1}
+                                       onClick={() => {
+                                         setTestResults(`=== Prompt Details ===\nCode: ${prompt.p_code}\nLabel: ${prompt.label}\nImage Type: ${prompt.image_type}\nActive: ${prompt.is_active}\n\nMetadata Instructions:\n${prompt.metadata_instructions || 'No instructions available'}`);
+                                         setTestResultsTitle(`Prompt: ${prompt.p_code}`);
+                                         setShowTestResultsModal(true);
+                                       }}
+                                     >
+                                       View
+                                     </Button>
+                                     <Button
+                                       name={`edit-${prompt.p_code}`}
+                                       variant="secondary"
+                                       size={1}
+                                       onClick={() => handleEditPrompt(prompt)}
+                                     >
+                                       Edit
+                                     </Button>
+                                   </div>
+                                 </td>
+                               </tr>
+                             ))}
+                           </tbody>
+                         </table>
+                       </div>
 
-                     {/* Add Prompt Button */}
-                     <div className={styles.addModelButtonContainer}>
-                       <Button
-                         name="add-prompt"
-                         variant="primary"
-                         onClick={() => {
-                           // TODO: Implement add prompt functionality
-                           alert('Add prompt functionality coming soon!');
-                         }}
-                       >
-                         Add New Prompt
-                       </Button>
+                       {/* Add Drone Image Prompt Button */}
+                       <div className={styles.addModelButtonContainer}>
+                         <Button
+                           name="add-drone-prompt"
+                           variant="primary"
+                           onClick={() => handleAddPrompt('drone_image')}
+                         >
+                           Add New Drone Image Prompt
+                         </Button>
+                       </div>
                      </div>
+
                    </div>
                  </Container>
 
@@ -991,6 +1170,28 @@ Model "${newModelData.label}" added successfully!
                     className={styles.formInput}
                   />
                 </div>
+                <div className={styles.formField}>
+                  <label className={styles.formLabel}>Image Type:</label>
+                  <SelectInput
+                    name="prompt-image-type"
+                    value={newPromptData.image_type}
+                    onChange={(value) => setNewPromptData(prev => ({ ...prev, image_type: value || 'crisis_map' }))}
+                    options={imageTypes}
+                    keySelector={(o) => o.image_type}
+                    labelSelector={(o) => o.label}
+                  />
+                </div>
+                <div className={styles.formField}>
+                  <label className={styles.formLabel}>Active Status:</label>
+                  <div className={styles.addModelFormCheckbox}>
+                    <input
+                      type="checkbox"
+                      checked={newPromptData.is_active}
+                      onChange={(e) => setNewPromptData(prev => ({ ...prev, is_active: e.target.checked }))}
+                    />
+                    <span>Active for this image type</span>
+                  </div>
+                </div>
                 <div className={styles.formField}>
                   <label className={styles.formLabel}>Metadata Instructions:</label>
                   <textarea
@@ -1009,7 +1210,7 @@ Model "${newModelData.label}" added successfully!
                   onClick={() => {
                     setShowEditPromptForm(false);
                     setEditingPrompt(null);
-                    setNewPromptData({ p_code: '', label: '', metadata_instructions: '' });
+                    setNewPromptData({ p_code: '', label: '', metadata_instructions: '', image_type: 'crisis_map', is_active: false });
                   }}
                 >
                   Cancel
@@ -1026,6 +1227,94 @@ Model "${newModelData.label}" added successfully!
           </div>
         </div>
       )}
+
+      {/* Add Prompt Form Modal */}
+      {showAddPromptForm && (
+        <div className={styles.modalOverlay} onClick={() => setShowAddPromptForm(false)}>
+          <div className={styles.modalContent} onClick={(e) => e.stopPropagation()}>
+            <div className={styles.modalBody}>
+              <h3 className={styles.modalTitle}>
+                Add New {addingPromptType === 'crisis_map' ? 'Crisis Map' : 'Drone Image'} Prompt
+              </h3>
+              <div className={styles.modalForm}>
+                <div className={styles.formField}>
+                  <label className={styles.formLabel}>Code:</label>
+                  <TextInput
+                    name="prompt-code"
+                    value={newPromptData.p_code}
+                    onChange={(value) => setNewPromptData(prev => ({ ...prev, p_code: value || '' }))}
+                    placeholder="e.g., CUSTOM_CRISIS_MAP_001"
+                    className={styles.formInput}
+                  />
+                </div>
+                <div className={styles.formField}>
+                  <label className={styles.formLabel}>Label:</label>
+                  <TextInput
+                    name="prompt-label"
+                    value={newPromptData.label}
+                    onChange={(value) => setNewPromptData(prev => ({ ...prev, label: value || '' }))}
+                    placeholder="Enter prompt description..."
+                    className={styles.formInput}
+                  />
+                </div>
+                <div className={styles.formField}>
+                  <label className={styles.formLabel}>Image Type:</label>
+                  <TextInput
+                    name="prompt-image-type"
+                    value={newPromptData.image_type}
+                    onChange={() => {}} // Disabled - automatically set
+                    disabled={true}
+                    className={styles.formInput}
+                  />
+                </div>
+                <div className={styles.formField}>
+                  <label className={styles.formLabel}>Active Status:</label>
+                  <div className={styles.addModelFormCheckbox}>
+                    <input
+                      type="checkbox"
+                      checked={newPromptData.is_active}
+                      onChange={(e) => setNewPromptData(prev => ({ ...prev, is_active: e.target.checked }))}
+                    />
+                    <span>Active for this image type</span>
+                  </div>
+                </div>
+                <div className={styles.formField}>
+                  <label className={styles.formLabel}>Metadata Instructions:</label>
+                  <textarea
+                    name="prompt-instructions"
+                    value={newPromptData.metadata_instructions}
+                    onChange={(e) => setNewPromptData(prev => ({ ...prev, metadata_instructions: e.target.value }))}
+                    placeholder="Enter metadata extraction instructions..."
+                    className={`${styles.formInput} ${styles.textarea}`}
+                    rows={8}
+                  />
+                </div>
+              </div>
+              <div className={styles.modalButtons}>
+                <Button
+                  name="cancel-add-prompt"
+                  variant="tertiary"
+                  onClick={() => {
+                    setShowAddPromptForm(false);
+                    setAddingPromptType(null);
+                    setNewPromptData({ p_code: '', label: '', metadata_instructions: '', image_type: 'crisis_map', is_active: false });
+                  }}
+                >
+                  Cancel
+                </Button>
+                <Button
+                  name="save-new-prompt"
+                  variant="primary"
+                  onClick={handleSaveNewPrompt}
+                  disabled={!newPromptData.p_code || !newPromptData.label}
+                >
+                  Create Prompt
+                </Button>
+              </div>
+            </div>
+          </div>
+        </div>
+      )}
     </PageContainer>
   );
 }

frontend/src/pages/UploadPage/UploadPage.tsx

@@ -616,7 +616,7 @@ export default function UploadPage() {
           },
           body: new URLSearchParams({
             title: title || 'Generated Caption',
-            prompt: imageType === 'drone_image' ? 'DEFAULT_DRONE_IMAGE' : 'DEFAULT_CRISIS_MAP',
+            // No prompt specified - backend will use active prompt for image type
             ...(modelName && { model_name: modelName })
           })
         },
@@ -737,7 +737,7 @@ export default function UploadPage() {
         headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
                   body: new URLSearchParams({
             title: 'Generated Caption',
-            prompt: imageType === 'drone_image' ? 'DEFAULT_DRONE_IMAGE' : 'DEFAULT_CRISIS_MAP',
+            // No prompt specified - backend will use active prompt for image type
             ...(modelName && { model_name: modelName }),
           }),
       });

py_backend/alembic/versions/0011_add_prompt_image_type_and_active.py

@@ -0,0 +1,59 @@
+"""Add image_type and is_active to prompts table
+
+Revision ID: 0011
+Revises: 0010
+Create Date: 2024-01-01 00:00:00.000000
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision = '0011'
+down_revision = '0010'
+branch_labels = None
+depends_on = None
+
+def upgrade():
+    # Add new columns to prompts table
+    op.add_column('prompts', sa.Column('image_type', sa.String(), nullable=True))
+    op.add_column('prompts', sa.Column('is_active', sa.Boolean(), server_default=sa.text('false'), nullable=False))
+    
+    # Update existing prompts to set image_type and is_active
+    op.execute("""
+        UPDATE prompts 
+        SET image_type = 'crisis_map', is_active = true 
+        WHERE p_code = 'DEFAULT_CRISIS_MAP'
+    """)
+    
+    op.execute("""
+        UPDATE prompts 
+        SET image_type = 'drone_image', is_active = true 
+        WHERE p_code = 'DEFAULT_DRONE_IMAGE'
+    """)
+    
+    # Make image_type NOT NULL after setting values
+    op.alter_column('prompts', 'image_type', nullable=False)
+    
+    # Create foreign key constraint
+    op.create_foreign_key(
+        'fk_prompts_image_type', 
+        'prompts', 
+        'image_types', 
+        ['image_type'], 
+        ['image_type']
+    )
+    
+    # Create index on image_type for performance
+    op.create_index('ix_prompts_image_type', 'prompts', ['image_type'])
+
+def downgrade():
+    # Remove the index
+    op.drop_index('ix_prompts_image_type', 'prompts')
+    
+    # Remove the foreign key constraint
+    op.drop_constraint('fk_prompts_image_type', 'prompts', type_='foreignkey')
+    
+    # Remove the columns
+    op.drop_column('prompts', 'is_active')
+    op.drop_column('prompts', 'image_type')

py_backend/alembic/versions/0012_fix_prompt_constraints.py

@@ -0,0 +1,34 @@
+"""Fix prompt constraints to allow multiple inactive prompts per image type
+
+Revision ID: 0012
+Revises: 0011
+Create Date: 2024-01-01 00:00:00.000000
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision = '0012'
+down_revision = '0011'
+branch_labels = None
+depends_on = None
+
+def upgrade():
+    # Drop the incorrect unique constraint
+    op.drop_constraint('uq_prompts_image_type_active', 'prompts', type_='unique')
+    
+    # Create a partial unique constraint that only applies when is_active = true
+    # This allows multiple inactive prompts per image type, but only one active
+    op.execute("""
+        CREATE UNIQUE INDEX uq_prompts_active_per_type 
+        ON prompts (image_type) 
+        WHERE is_active = true
+    """)
+
+def downgrade():
+    # Drop the partial unique constraint
+    op.execute("DROP INDEX IF EXISTS uq_prompts_active_per_type")
+    
+    # Recreate the original constraint
+    op.create_unique_constraint('uq_prompts_image_type_active', 'prompts', ['image_type', 'is_active'])

py_backend/app/crud.py

@@ -142,13 +142,111 @@ def get_prompt_by_label(db: Session, label: str):
     """Get a specific prompt by label text"""
     return db.query(models.Prompts).filter(models.Prompts.label == label).first()
 
+def get_active_prompt_by_image_type(db: Session, image_type: str):
+    """Get the active prompt for a specific image type"""
+    return db.query(models.Prompts).filter(
+        models.Prompts.image_type == image_type,
+        models.Prompts.is_active == True
+    ).first()
+
+def toggle_prompt_active_status(db: Session, p_code: str, image_type: str):
+    """Toggle the active status of a prompt for a specific image type"""
+    # Validate that the image_type exists
+    image_type_obj = db.query(models.ImageTypes).filter(models.ImageTypes.image_type == image_type).first()
+    if not image_type_obj:
+        raise ValueError(f"Invalid image_type: {image_type}")
+    
+    # Get the prompt to toggle
+    prompt = db.query(models.Prompts).filter(models.Prompts.p_code == p_code).first()
+    if not prompt:
+        return None
+    
+    # If the prompt is already active, deactivate it
+    if prompt.is_active:
+        prompt.is_active = False
+        db.commit()
+        db.refresh(prompt)
+        return prompt
+    
+    # If the prompt is not active, first deactivate the currently active prompt
+    # then activate this one
+    current_active = db.query(models.Prompts).filter(
+        models.Prompts.image_type == image_type,
+        models.Prompts.is_active == True
+    ).first()
+    
+    if current_active:
+        current_active.is_active = False
+        # Commit the deactivation first to avoid constraint violation
+        db.commit()
+    
+    prompt.is_active = True
+    db.commit()
+    db.refresh(prompt)
+    return prompt
+
+def create_prompt(db: Session, prompt_data: schemas.PromptCreate):
+    """Create a new prompt"""
+    # Validate that the image_type exists
+    image_type_obj = db.query(models.ImageTypes).filter(models.ImageTypes.image_type == prompt_data.image_type).first()
+    if not image_type_obj:
+        raise ValueError(f"Invalid image_type: {prompt_data.image_type}")
+    
+    # Check if prompt code already exists
+    existing_prompt = db.query(models.Prompts).filter(models.Prompts.p_code == prompt_data.p_code).first()
+    if existing_prompt:
+        raise ValueError(f"Prompt with code '{prompt_data.p_code}' already exists")
+    
+    # If this prompt is set as active, deactivate the currently active prompt for this image type
+    if prompt_data.is_active:
+        current_active = db.query(models.Prompts).filter(
+            models.Prompts.image_type == prompt_data.image_type,
+            models.Prompts.is_active == True
+        ).first()
+        
+        if current_active:
+            current_active.is_active = False
+            # Commit the deactivation first to avoid constraint violation
+            db.commit()
+    
+    # Create the new prompt
+    new_prompt = models.Prompts(
+        p_code=prompt_data.p_code,
+        label=prompt_data.label,
+        metadata_instructions=prompt_data.metadata_instructions,
+        image_type=prompt_data.image_type,
+        is_active=prompt_data.is_active
+    )
+    
+    db.add(new_prompt)
+    db.commit()
+    db.refresh(new_prompt)
+    return new_prompt
+
 def update_prompt(db: Session, p_code: str, prompt_update: schemas.PromptUpdate):
     """Update a specific prompt by code"""
     prompt = db.query(models.Prompts).filter(models.Prompts.p_code == p_code).first()
     if not prompt:
         return None
     
-    for field, value in prompt_update.dict(exclude_unset=True).items():
+    # Handle is_active field specially to maintain unique constraint
+    update_data = prompt_update.dict(exclude_unset=True)
+    
+    # If we're setting this prompt as active, deactivate other prompts for this image type
+    if 'is_active' in update_data and update_data['is_active']:
+        current_active = db.query(models.Prompts).filter(
+            models.Prompts.image_type == prompt.image_type,
+            models.Prompts.is_active == True,
+            models.Prompts.p_code != p_code  # Exclude current prompt
+        ).first()
+        
+        if current_active:
+            current_active.is_active = False
+            # Commit the deactivation first to avoid constraint violation
+            db.commit()
+    
+    # Update all fields
+    for field, value in update_data.items():
         setattr(prompt, field, value)
     
     db.commit()

py_backend/app/main.py

@@ -19,7 +19,6 @@ from app.routers.schemas import router as schemas_router
 
 app = FastAPI(title="PromptAid Vision")
 
-# Add request logging middleware
 @app.middleware("http")
 async def log_requests(request, call_next):
     print(f"DEBUG: {request.method} {request.url.path}")
@@ -36,7 +35,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# API routes - must come BEFORE static file mounting
 app.include_router(caption.router,     prefix="/api",            tags=["captions"])
 app.include_router(metadata.router,    prefix="/api",            tags=["metadata"])
 app.include_router(models.router,      prefix="/api",            tags=["models"])
@@ -58,17 +56,7 @@ async def list_images_no_slash():
     finally:
         db.close()
 
-@app.get("/api/prompts", include_in_schema=False)
-async def list_prompts_no_slash():
-    """Handle /api/prompts without trailing slash to prevent 307 redirect"""
-    from app.routers.prompts import get_prompts
-    from app.database import SessionLocal
-    
-    db = SessionLocal()
-    try:
-        return get_prompts(db)
-    finally:
-        db.close()
+
 
 @app.get("/health", include_in_schema=False, response_class=JSONResponse)
 async def health():
@@ -82,17 +70,16 @@ def root():
 <p>OK</p>
 <p><a href="/app/">Open UI</a> • <a href="/docs">API Docs</a></p>"""
 
-# Static file serving - must come AFTER API routes
 if os.path.exists("/app"):
     STATIC_DIR = "/app/static"
 else:
-    STATIC_DIR = "static"  # Use relative path to py_backend/static
+    STATIC_DIR = "static"  
 
 print(f"Looking for static files in: {STATIC_DIR}")
 
 if os.path.isdir(STATIC_DIR):
     print(f"Static directory found: {STATIC_DIR}")
-    # Mount static files at /app to avoid conflicts with /api routes
+
     app.mount("/app", StaticFiles(directory=STATIC_DIR, html=True), name="static")
     print(f"Static files mounted at /app from {STATIC_DIR}")
 else:
@@ -101,7 +88,6 @@ else:
     print(f"Parent directory contents: {os.listdir(os.path.dirname(os.path.dirname(__file__)))}")
     print(f"Attempting to find static directory...")
     
-    # Try to find static directory
     possible_paths = [
         "static",
         "../static", 
@@ -119,7 +105,7 @@ else:
     else:
         print("Could not find static directory - static file serving disabled")
 
-# SPA fallback - must come AFTER static file mounting
+
 @app.get("/app/{full_path:path}", include_in_schema=False)
 def spa_fallback(full_path: str):
     index = os.path.join(STATIC_DIR, "index.html")
@@ -127,7 +113,7 @@ def spa_fallback(full_path: str):
         return FileResponse(index)
     raise HTTPException(status_code=404, detail="Not Found")
 
-# Debug route to show all registered routes
+
 @app.get("/debug-routes", include_in_schema=False)
 async def debug_routes():
     """Show all registered routes for debugging"""

py_backend/app/models.py

@@ -62,6 +62,11 @@ class Prompts(Base):
     p_code = Column(String, primary_key=True)
     label = Column(Text, nullable=False)
     metadata_instructions = Column(Text, nullable=True)
+    image_type = Column(String, ForeignKey("image_types.image_type"), nullable=False)
+    is_active = Column(Boolean, default=False, nullable=False)
+    
+    # Relationship to image_types table
+    image_type_r = relationship("ImageTypes")
 
 class Models(Base):
     __tablename__ = "models"

py_backend/app/routers/caption.py

@@ -87,7 +87,7 @@ def get_db():
 async def create_caption(
     image_id: str,
     title: str = Form(...),
-    prompt: str = Form(...),
+    prompt: str = Form(None),  # Made optional since we'll use active prompts
     model_name: str | None = Form(None),
     db: Session = Depends(get_db),
 ):
@@ -97,14 +97,22 @@ async def create_caption(
     if not img:
         raise HTTPException(404, "image not found")
 
-
-    print(f"Looking for prompt: '{prompt}' (type: {type(prompt)})")
-
-    prompt_obj = crud.get_prompt(db, prompt)
-    
-    if not prompt_obj:
-        print(f"Prompt not found by code, trying to find by label...")
-        prompt_obj = crud.get_prompt_by_label(db, prompt)
+    # Get the active prompt for this image type
+    if prompt:
+        # If prompt is provided, use it (for backward compatibility)
+        print(f"Looking for prompt: '{prompt}' (type: {type(prompt)})")
+        prompt_obj = crud.get_prompt(db, prompt)
+        
+        if not prompt_obj:
+            print(f"Prompt not found by code, trying to find by label...")
+            prompt_obj = crud.get_prompt_by_label(db, prompt)
+    else:
+        # Use the active prompt for the image type
+        print(f"Looking for active prompt for image type: {img.image_type}")
+        prompt_obj = crud.get_active_prompt_by_image_type(db, img.image_type)
+        
+        if not prompt_obj:
+            raise HTTPException(400, f"No active prompt found for image type '{img.image_type}'")
     
     print(f"Prompt lookup result: {prompt_obj}")
     if not prompt_obj:

py_backend/app/routers/prompts.py

@@ -34,3 +34,35 @@ def update_prompt(p_code: str, prompt_update: schemas.PromptUpdate, db: Session
         from fastapi import HTTPException
         raise HTTPException(404, "Prompt not found")
     return prompt
+
+@router.post("/{p_code}/toggle-active", response_model=schemas.PromptOut)
+def toggle_prompt_active(p_code: str, image_type: str, db: Session = Depends(get_db)):
+    """Toggle the active status of a prompt for a specific image type"""
+    try:
+        prompt = crud.toggle_prompt_active_status(db, p_code, image_type)
+        if not prompt:
+            from fastapi import HTTPException
+            raise HTTPException(404, "Prompt not found")
+        return prompt
+    except ValueError as e:
+        from fastapi import HTTPException
+        raise HTTPException(400, str(e))
+
+@router.get("/active/{image_type}", response_model=schemas.PromptOut)
+def get_active_prompt(image_type: str, db: Session = Depends(get_db)):
+    """Get the active prompt for a specific image type"""
+    prompt = crud.get_active_prompt_by_image_type(db, image_type)
+    if not prompt:
+        from fastapi import HTTPException
+        raise HTTPException(404, "No active prompt found for this image type")
+    return prompt
+
+@router.post("/", response_model=schemas.PromptOut)
+def create_prompt(prompt_data: schemas.PromptCreate, db: Session = Depends(get_db)):
+    """Create a new prompt"""
+    try:
+        prompt = crud.create_prompt(db, prompt_data)
+        return prompt
+    except ValueError as e:
+        from fastapi import HTTPException
+        raise HTTPException(400, str(e))

py_backend/app/schemas.py

@@ -98,13 +98,24 @@ class PromptOut(BaseModel):
     p_code: str
     label: str
     metadata_instructions: str | None = None
+    image_type: str
+    is_active: bool
     
     class Config:
         from_attributes = True
 
+class PromptCreate(BaseModel):
+    p_code: str
+    label: str
+    metadata_instructions: str | None = None
+    image_type: str
+    is_active: bool
+
 class PromptUpdate(BaseModel):
     label: str
     metadata_instructions: str | None = None
+    image_type: str
+    is_active: bool
 
 class SourceOut(BaseModel):
     s_code: str