Feature ai chat regenerate button

src/main/java/org/jabref/gui/ai/components/aichat/AiChatComponent.java

@@ -84,13 +84,19 @@ public AiChatComponent(AiService aiService,
         aiService.getIngestionService().ingest(name, ListUtil.getLinkedFiles(entries).toList(), bibDatabaseContext);
 
         ViewLoader.view(this)
-                .root(this)
-                .load();
+                  .root(this)
+                  .load();
     }
 
     @FXML
     public void initialize() {
         uiChatHistory.setItems(aiChatLogic.getChatHistory());
+        uiChatHistory.setRegenerateCallback(userPrompt -> {
+            deleteLastMessage();
+            deleteLastMessage();
+            chatPrompt.switchToNormalState();
+            onSendMessage(userPrompt);
+        });
         initializeChatPrompt();
         initializeNotice();
         initializeNotifications();
@@ -174,8 +180,8 @@ private List<Notification> updateNotificationsForEntry(BibEntry entry) {
         entry.getFiles().stream().map(file -> aiService.getIngestionService().ingest(file, bibDatabaseContext)).forEach(ingestionStatus -> {
             switch (ingestionStatus.getState()) {
                 case PROCESSING -> notifications.add(new Notification(
-                    Localization.lang("File %0 is currently being processed", ingestionStatus.getObject().getLink()),
-                    Localization.lang("After the file will be ingested, you will be able to chat with it.")
+                        Localization.lang("File %0 is currently being processed", ingestionStatus.getObject().getLink()),
+                        Localization.lang("After the file will be ingested, you will be able to chat with it.")
                 ));
 
                 case ERROR -> {

src/main/java/org/jabref/gui/ai/components/aichat/chathistory/ChatHistoryComponent.java

@@ -1,5 +1,7 @@
 package org.jabref.gui.ai.components.aichat.chathistory;
 
+import java.util.function.Consumer;
+
 import javafx.collections.ListChangeListener;
 import javafx.collections.ObservableList;
 import javafx.fxml.FXML;
@@ -10,11 +12,15 @@
 import org.jabref.gui.util.UiTaskExecutor;
 
 import com.airhacks.afterburner.views.ViewLoader;
+import dev.langchain4j.data.message.AiMessage;
 import dev.langchain4j.data.message.ChatMessage;
+import dev.langchain4j.data.message.UserMessage;
 
 public class ChatHistoryComponent extends ScrollPane {
     @FXML private VBox vBox;
 
+    private Consumer<String> regenerateCallback;
+
     public ChatHistoryComponent() {
         ViewLoader.view(this)
                   .root(this)
@@ -35,14 +41,34 @@ public void setItems(ObservableList<ChatMessage> items) {
         items.addListener((ListChangeListener<? super ChatMessage>) obs -> fill(items));
     }
 
+    public void setRegenerateCallback(Consumer<String> regenerateCallback) {
+        this.regenerateCallback = regenerateCallback;
+    }
+
     private void fill(ObservableList<ChatMessage> items) {
         UiTaskExecutor.runInJavaFXThread(() -> {
             vBox.getChildren().clear();
-            items.forEach(chatMessage ->
-                    vBox.getChildren().add(new ChatMessageComponent(chatMessage, chatMessageComponent -> {
-                        int index = vBox.getChildren().indexOf(chatMessageComponent);
-                        items.remove(index);
-                    })));
+            for (ChatMessage chatMessage : items) {
+                ChatMessageComponent component = new ChatMessageComponent(chatMessage, comp -> {
+                    items.remove(chatMessage);
+                });
+                if (chatMessage instanceof AiMessage) {
+                    component.setOnRegenerate(comp -> {
+                        if (!items.isEmpty() && items.getLast() == chatMessage && items.size() > 1) {
+                            ChatMessage previous = items.get(items.size() - 2);
+                            if (previous instanceof UserMessage message) {
+                                String userText = message.singleText();
+                                items.removeLast();
+                                items.removeLast();
+                                if (regenerateCallback != null) {
+                                    regenerateCallback.accept(userText);
+                                }
+                            }
+                        }
+                    });
+                }
+                vBox.getChildren().add(component);
+            }
         });
     }
 
@@ -51,3 +77,4 @@ public void scrollDown() {
         this.setVvalue(this.getVmax());
     }
 }
+

src/main/java/org/jabref/gui/ai/components/aichat/chatmessage/ChatMessageComponent.fxml

@@ -35,6 +35,15 @@
                     <Tooltip text="%Delete message from chat history" />
                 </tooltip>
             </Button>
+            <Button onAction="#onRegenerateClick" styleClass="icon-button,narrow" textAlignment="CENTER">
+                <graphic>
+                    <JabRefIconView glyph="REFRESH"/>
+                </graphic>
+                <tooltip>
+                    <Tooltip text="%Regenerate AI response" />
+                </tooltip>
+            </Button>
         </VBox>
     </HBox>
 </fx:root>
+

src/main/java/org/jabref/gui/ai/components/aichat/chatmessage/ChatMessageComponent.java

@@ -26,6 +26,7 @@ public class ChatMessageComponent extends HBox {
 
     private final ObjectProperty<ChatMessage> chatMessage = new SimpleObjectProperty<>();
     private final ObjectProperty<Consumer<ChatMessageComponent>> onDelete = new SimpleObjectProperty<>();
+    private final ObjectProperty<Consumer<ChatMessageComponent>> onRegenerate = new SimpleObjectProperty<>();
 
     @FXML private HBox wrapperHBox;
     @FXML private VBox vBox;
@@ -63,6 +64,10 @@ public void setOnDelete(Consumer<ChatMessageComponent> onDeleteCallback) {
         this.onDelete.set(onDeleteCallback);
     }
 
+    public void setOnRegenerate(Consumer<ChatMessageComponent> callback) {
+        this.onRegenerate.set(callback);
+    }
+
     private void loadChatMessage() {
         switch (chatMessage.get()) {
             case UserMessage userMessage -> {
@@ -87,7 +92,7 @@ private void loadChatMessage() {
             }
 
             default ->
-                LOGGER.error("ChatMessageComponent supports only user, AI, or error messages, but other type was passed: {}", chatMessage.get().type().name());
+                    LOGGER.error("ChatMessageComponent supports only user, AI, or error messages, but other type was passed: {}", chatMessage.get().type().name());
         }
     }
 
@@ -103,7 +108,15 @@ private void onDeleteClick() {
         }
     }
 
+    @FXML
+    private void onRegenerateClick() {
+        if (onRegenerate.get() != null) {
+            onRegenerate.get().accept(this);
+        }
+    }
+
     private void setColor(String fillColor, String borderColor) {
         vBox.setStyle("-fx-background-color: " + fillColor + "; -fx-border-radius: 10; -fx-background-radius: 10; -fx-border-color: " + borderColor + "; -fx-border-width: 3;");
     }
 }
+

src/main/resources/l10n/JabRef_de.properties

@@ -2644,6 +2644,7 @@ Additionally,\ we\ use\ Deep\ Java\ Library\ (DJL)\ embedding\ models\ for\ both
 An\ API\ key\ has\ to\ be\ provided=Ein API-Schlüssel muss bereitgestellt werden
 Current\ AI\ model\:\ %0.\ The\ AI\ may\ generate\ inaccurate\ or\ inappropriate\ responses.\ Please\ verify\ any\ information\ provided.=Aktuelles KI-Modell\: %0. Die KI kann ungenaue oder unangemessene Antworten generieren. Bitte überprüfen Sie die bereitgestellten Informationen.
 Delete\ message\ from\ chat\ history=Nachricht aus dem Chatverlauf löschen
+Regenerate\ AI\ response=KI-Antwort neu generieren
 Generated\ at\ %0\ by\ %1=Generiert am %0 von %1
 Retry=Erneut versuchen
 Updating\ local\ embedding\ model...=Aktualisiere lokales Einbettungsmodell...

src/main/resources/l10n/JabRef_en.properties

@@ -2649,6 +2649,7 @@ Additionally,\ we\ use\ Deep\ Java\ Library\ (DJL)\ embedding\ models\ for\ both
 An\ API\ key\ has\ to\ be\ provided=An API key has to be provided
 Current\ AI\ model\:\ %0.\ The\ AI\ may\ generate\ inaccurate\ or\ inappropriate\ responses.\ Please\ verify\ any\ information\ provided.=Current AI model: %0. The AI may generate inaccurate or inappropriate responses. Please verify any information provided.
 Delete\ message\ from\ chat\ history=Delete message from chat history
+Regenerate\ AI\ response=Regenerate AI response
 Generated\ at\ %0\ by\ %1=Generated at %0 by %1
 Retry=Retry
 Updating\ local\ embedding\ model...=Updating local embedding model...

src/main/resources/l10n/JabRef_fr.properties

@@ -2644,6 +2644,7 @@ Additionally,\ we\ use\ Deep\ Java\ Library\ (DJL)\ embedding\ models\ for\ both
 An\ API\ key\ has\ to\ be\ provided=Une clé API doit être fournie
 Current\ AI\ model\:\ %0.\ The\ AI\ may\ generate\ inaccurate\ or\ inappropriate\ responses.\ Please\ verify\ any\ information\ provided.=Modèle actuel de l'IA \: %0. L'IA peut générer des réponses inexactes ou inappropriées. Veuillez vérifier toutes les informations fournies.
 Delete\ message\ from\ chat\ history=Supprimer le message de l'historique du tchat
+Regenerate\ AI\ response=R�g�n�rer la r�ponse de l'IA
 Generated\ at\ %0\ by\ %1=Généré à %0 par %1
 Retry=Réessayer
 Updating\ local\ embedding\ model...=Mise à jour du modèle d'intégration local...

src/main/resources/l10n/JabRef_pt_BR.properties

@@ -2647,6 +2647,7 @@ Additionally,\ we\ use\ Deep\ Java\ Library\ (DJL)\ embedding\ models\ for\ both
 An\ API\ key\ has\ to\ be\ provided=Uma chave de API tem de ser fornecida
 Current\ AI\ model\:\ %0.\ The\ AI\ may\ generate\ inaccurate\ or\ inappropriate\ responses.\ Please\ verify\ any\ information\ provided.=Modelo de IA atual\: %0. A IA pode gerar respostas imprecisas ou inadequadas. Por favor, verifique todas as informações fornecidas.
 Delete\ message\ from\ chat\ history=Excluir mensagem do histórico de bate-papo
+Regenerate\ AI\ response=Regenerar resposta da IA
 Generated\ at\ %0\ by\ %1=Gerado a %0 por %1
 Retry=Tentar novamente
 Updating\ local\ embedding\ model...=Atualizando modelo local de incorporação...