Fix for InAnYan#118

JabRef · Aug 7, 2024 · a38c439 · a38c439
1 parent 8b7fff1
commit a38c439
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 1 deletion.
diff --git a/src/main/java/org/jabref/gui/ai/components/aichat/AiChatComponent.fxml b/src/main/java/org/jabref/gui/ai/components/aichat/AiChatComponent.fxml
@@ -26,7 +26,7 @@
               </ScrollPane>
           </children>
       </StackPane>
-      <HBox alignment="CENTER" spacing="10.0">
+      <HBox fx:id="promptHBox" alignment="CENTER" spacing="10.0">
          <children>
             <ExpandingTextArea fx:id="userPromptTextArea" HBox.hgrow="ALWAYS" />
             <Button fx:id="submitButton" mnemonicParsing="false" onAction="#onSendMessage" text="%Submit" />

diff --git a/src/main/java/org/jabref/gui/ai/components/aichat/AiChatComponent.java b/src/main/java/org/jabref/gui/ai/components/aichat/AiChatComponent.java
@@ -7,6 +7,7 @@
 import javafx.scene.control.ScrollPane;
 import javafx.scene.input.KeyCode;
 import javafx.scene.layout.BorderPane;
+import javafx.scene.layout.HBox;
 import javafx.scene.layout.StackPane;
 import javafx.scene.layout.VBox;
 import javafx.scene.text.Text;
@@ -39,6 +40,7 @@ public class AiChatComponent extends VBox {
 
     @FXML private ScrollPane scrollPane;
     @FXML private VBox chatVBox;
+    @FXML private HBox promptHBox;
     @FXML private ExpandingTextArea userPromptTextArea;
     @FXML private Button submitButton;
     @FXML private StackPane stackPane;
@@ -114,6 +116,8 @@ private void onSendMessage() {
                                 } else {
                                     addError(e.getMessage());
                                 }
+
+                                switchToErrorState(userPrompt);
                             });
 
             task.titleProperty().set(Localization.lang("Waiting for AI reply for %0...", citationKey));
@@ -122,6 +126,41 @@ private void onSendMessage() {
         }
     }
 
+    private void switchToErrorState(String userMessage) {
+        promptHBox.getChildren().clear();
+
+        Button retryButton = new Button(Localization.lang("Retry"));
+
+        retryButton.setOnAction(event -> {
+            userPromptTextArea.setText(userMessage);
+
+            chatVBox.getChildren().removeLast();
+            chatVBox.getChildren().removeLast();
+
+            switchToNormalState();
+
+            onSendMessage();
+        });
+
+        Button cancelButton = new Button(Localization.lang("Cancel"));
+
+        cancelButton.setOnAction(event -> {
+            chatVBox.getChildren().removeLast();
+            chatVBox.getChildren().removeLast();
+
+            switchToNormalState();
+        });
+
+        promptHBox.getChildren().add(retryButton);
+        promptHBox.getChildren().add(cancelButton);
+    }
+
+    private void switchToNormalState() {
+        promptHBox.getChildren().clear();
+        promptHBox.getChildren().add(userPromptTextArea);
+        promptHBox.getChildren().add(submitButton);
+    }
+
     private void setLoading(boolean loading) {
         userPromptTextArea.setDisable(loading);
         submitButton.setDisable(loading);