Allow defining prompts at the category level

Author: tomasr8

ai_summary/indico_ai_summary/client/CategoryManagePrompts.jsx

@@ -5,12 +5,45 @@
 // them and/or modify them under the terms of the MIT License;
 // see the LICENSE file for more details.
 
+import _ from 'lodash';
 import React from 'react';
 import ReactDOM from 'react-dom';
+import {Form as FinalForm} from 'react-final-form';
 import {Form} from 'semantic-ui-react';
 
-export default function CategoryManagePrompts() {
-  return <Form>TODO: Manage Prompts</Form>;
+import {indicoAxios} from 'indico/utils/axios';
+import {FinalSubmitButton, handleSubmitError} from 'indico/react/forms';
+
+import {FinalPromptManagerField} from './components/PromptManagerField';
+
+export default function CategoryManagePrompts({categoryId, prompts: predefinedPrompts}) {
+  const onSubmit = async ({prompts}, form) => {
+    // TODO: url import
+    const url = `/plugin/ai-summary/manage-category-prompts/${categoryId}`;
+    try {
+      await indicoAxios.post(url, {prompts});
+      form.initialize({prompts});
+    } catch (error) {
+      handleSubmitError(error);
+    }
+  };
+
+  const submitBtn = <FinalSubmitButton label="Save Changes" />;
+
+  return (
+    <FinalForm
+      onSubmit={onSubmit}
+      initialValues={{prompts: predefinedPrompts}}
+      initialValuesEqual={_.isEqual}
+      subscription={{}}
+    >
+      {fprops => (
+        <Form onSubmit={fprops.handleSubmit}>
+          <FinalPromptManagerField submitBtn={submitBtn} />
+        </Form>
+      )}
+    </FinalForm>
+  );
 }
 
 window.setupCategoryManagePrompts = function setupCategoryManagePrompts() {

ai_summary/indico_ai_summary/client/components/PromptManagerField.jsx

@@ -5,11 +5,25 @@
 // them and/or modify them under the terms of the MIT License;
 // see the LICENSE file for more details.
 
+import _ from 'lodash';
 import React, {useCallback, useMemo, useState} from 'react';
 import ReactDOM from 'react-dom';
 import {Form, TextArea, Button, Input, Card, Icon} from 'semantic-ui-react';
 
-export default function PromptManagerField({value, onChange}) {
+import {FinalField} from 'indico/react/forms';
+
+export function FinalPromptManagerField({submitBtn}) {
+  return (
+    <FinalField
+      name="prompts"
+      component={PromptManagerField}
+      submitBtn={submitBtn}
+      isEqual={_.isEqual}
+    />
+  );
+}
+
+export default function PromptManagerField({value, onChange, submitBtn}) {
   const addPrompt = () => {
     onChange([...value, {name: '', text: ''}]);
   };
@@ -27,7 +41,7 @@ export default function PromptManagerField({value, onChange}) {
   };
 
   return (
-    <Form>
+    <>
       {value.map((prompt, idx) => (
         <PromptField
           key={idx}
@@ -43,8 +57,9 @@ export default function PromptManagerField({value, onChange}) {
           Add Prompt
           <Icon name="plus" />
         </Button>
+        {submitBtn}
       </div>
-    </Form>
+    </>
   );
 }
 
@@ -81,7 +96,7 @@ function PromptField({name, text, onChangeName, onChangeText, onRemove}) {
 
 export function WTFPromptManagerField({fieldId}) {
   const field = useMemo(() => document.getElementById(`${fieldId}-data`), [fieldId]);
-  const [prompts, setPrompts] = useState(JSON.parse(field.value) || [{name: 'Title', text: ''}]);
+  const [prompts, setPrompts] = useState(JSON.parse(field.value));
 
   const onChange = useCallback(
     v => {
@@ -92,7 +107,11 @@ export function WTFPromptManagerField({fieldId}) {
     [field]
   );
 
-  return <PromptManagerField value={prompts} onChange={onChange} />;
+  return (
+    <Form>
+      <PromptManagerField value={prompts} onChange={onChange} />
+    </Form>
+  );
 }
 
 window.setupPromptManagerFieldWidget = function setupPromptManagerFieldWidget({fieldId}) {

ai_summary/indico_ai_summary/controllers.py

@@ -10,10 +10,13 @@
 from webargs import fields
 from webargs.flaskparser import use_kwargs
 
+from indico.core.db import db
 from indico.modules.categories.controllers.base import RHManageCategoryBase
 from indico.modules.events.management.controllers.base import RHManageEventBase
 from indico.modules.events.notes.util import get_scheduled_notes
 
+from indico_ai_summary.models.prompt import Prompt
+from indico_ai_summary.schemas import PromptSchema
 from indico_ai_summary.utils import cern_qwen, chunk_text, html_to_markdown, markdown_to_html
 from indico_ai_summary.views import WPCategoryManagePrompts
 
@@ -27,9 +30,19 @@ def _process_args(self):
         RHManageCategoryBase._process_args(self)
 
     def _process_GET(self):
+        prompts = Prompt.query.with_parent(self.category).all()
+        serialized_prompts = PromptSchema(many=True).dump(prompts)
         return WPCategoryManagePrompts.render_template('manage_category_prompts.html', category=self.category,
                                                        active_menu_item=CATEGORY_SIDEMENU_ITEM,
-                                                       stored_prompts=current_plugin.settings.get('prompts'))
+                                                       stored_prompts=serialized_prompts)
+
+    @use_kwargs({'prompts': fields.List(fields.Nested(PromptSchema), required=True)}, location='json')
+    def _process_POST(self, prompts):
+        Prompt.query.with_parent(self.category).delete()
+        for prompt_data in prompts:
+            prompt = Prompt(category_id=self.category.id, name=prompt_data['name'], text=prompt_data['text'])
+            db.session.add(prompt)
+        return '', 204
 
 
 class SummarizeEvent(RHManageEventBase):

ai_summary/indico_ai_summary/migrations/.no-header

@@ -0,0 +1,34 @@
+"""Add table for predefined prompts for categories
+
+Revision ID: e2bf94dfa64a
+Revises:
+Create Date: 2025-09-04 12:08:06.061611
+"""
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.sql.ddl import CreateSchema, DropSchema
+
+
+# revision identifiers, used by Alembic.
+revision = 'e2bf94dfa64a'
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.execute(CreateSchema('plugin_ai_summary'))
+    op.create_table('predefined_prompts',
+                    sa.Column('id', sa.Integer(), nullable=False),
+                    sa.Column('category_id', sa.Integer(), nullable=False, index=True),
+                    sa.Column('name', sa.Text(), nullable=False),
+                    sa.Column('text', sa.Text(), nullable=False),
+                    sa.ForeignKeyConstraint(['category_id'], ['categories.categories.id']),
+                    sa.PrimaryKeyConstraint('id'),
+                    schema='plugin_ai_summary')
+
+
+def downgrade():
+    op.drop_table('predefined_prompts', schema='plugin_ai_summary')
+    op.execute(DropSchema('plugin_ai_summary'))

ai_summary/indico_ai_summary/migrations/20250904_1208_e2bf94dfa64a_.py

@@ -0,0 +1,52 @@
+# This file is part of the CERN Indico plugins.
+# Copyright (C) 2014 - 2025 CERN
+#
+# The CERN Indico plugins are free software; you can redistribute
+# them and/or modify them under the terms of the MIT License; see
+# the LICENSE file for more details.
+
+from indico.core.db.sqlalchemy import db
+from indico.util.string import format_repr
+
+
+class Prompt(db.Model):
+    """Predefined LLM prompts"""
+
+    __tablename__ = 'predefined_prompts'
+    __table_args__ = {'schema': 'plugin_ai_summary'}
+
+    id = db.Column(
+        db.Integer,
+        primary_key=True
+    )
+    #: The ID of the category this prompt is defined for
+    category_id = db.Column(
+        db.Integer,
+        db.ForeignKey('categories.categories.id'),
+        index=True,
+        nullable=False
+    )
+    #: Name of the prompt
+    name = db.Column(
+        db.String,
+        nullable=False
+    )
+    #: Text of the prompt
+    text = db.Column(
+        db.String,
+        nullable=False
+    )
+
+    #: The Category this prompt is defined for
+    category = db.relationship(
+        'Category',
+        lazy=True,
+        backref=db.backref(
+            'predefined_prompts',
+            cascade='all, delete-orphan',
+            lazy='dynamic'
+        )
+    )
+
+    def __repr__(self):
+        return format_repr(self, 'id', 'category_id', 'name')

ai_summary/indico_ai_summary/models/__init__.py

@@ -21,6 +21,8 @@
 
 from indico_ai_summary.blueprint import blueprint
 from indico_ai_summary.controllers import CATEGORY_SIDEMENU_ITEM
+from indico_ai_summary.models.prompt import Prompt
+from indico_ai_summary.schemas import PromptSchema
 from indico_ai_summary.views import WPCategoryManagePrompts
 
 
@@ -61,8 +63,11 @@ def get_blueprints(self):
         return blueprint
 
     def _render_summarize_button(self, event):
+        global_prompts = self.settings.get('prompts')
+        category_prompts = Prompt.query.with_parent(event.category).all()
+        all_prompts = global_prompts + PromptSchema(many=True).dump(category_prompts)
         return render_plugin_template('summarize_button.html', category=event.category,
-                                      event=event, stored_prompts=self.settings.get('prompts'))
+                                      event=event, stored_prompts=all_prompts)
 
     def _extend_category_menu(self, sender, category, **kwargs):
         return SideMenuItem(CATEGORY_SIDEMENU_ITEM, _('Prompts'),

ai_summary/indico_ai_summary/models/prompt.py

@@ -0,0 +1,8 @@
+from webargs import fields
+
+from indico.core.marshmallow import mm
+
+
+class PromptSchema(mm.Schema):
+    name = fields.Str(required=True)
+    text = fields.Str(required=True)

ai_summary/indico_ai_summary/plugin.py

@@ -1,7 +1,7 @@
 {% extends 'categories/management/base.html' %}
 
 {% block title %}
-    {% trans %}Prompts{% endtrans %}
+    {% trans %}Predefined Prompts{% endtrans %}
 {% endblock %}
 
 {% block content %}