From 9597a7ea802cf1e3e9da45d700db65951ed0164f Mon Sep 17 00:00:00 2001
From: Ludovic Apvrille <ludovic.apvrille@telecom-paris.fr>
Date: Tue, 29 Aug 2023 17:55:42 +0200
Subject: [PATCH] Adding time to generate model with AI
---
src/main/java/ui/window/JFrameAI.java | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/src/main/java/ui/window/JFrameAI.java b/src/main/java/ui/window/JFrameAI.java
index bd16e8e1d6..df02c76b1d 100644
--- a/src/main/java/ui/window/JFrameAI.java
+++ b/src/main/java/ui/window/JFrameAI.java
@@ -123,6 +123,8 @@ public class JFrameAI extends JFrame implements ActionListener {
private HashMap<Integer, ImageIcon> rotatedI = new HashMap<>();
private JButton buttonClose, buttonStart, buttonApplyResponse;
+ private long startTime, endTime;
+
public JFrameAI(String title, MainGUI _mgui) {
super(title);
mgui = _mgui;
@@ -320,6 +322,7 @@ public class JFrameAI extends JFrame implements ActionListener {
private void start() {
TraceManager.addDev("Start in JFrameAI");
+ startTime = System.currentTimeMillis();
currentChatIndex = answerPane.getSelectedIndex();
ChatData selected = selectedChat();
@@ -395,6 +398,8 @@ public class JFrameAI extends JFrame implements ActionListener {
selected.aiInteract.makeRequest(question.getText());
+ //question.setText("Total time: " + (endTime - startTime) + " ms");
+
} else {
error("AI interface failed (no key has been set?)");
}
@@ -776,6 +781,9 @@ public class JFrameAI extends JFrame implements ActionListener {
lastAnswer = text;
//GraphicLib.appendToPane(selectedChat().answer, "\nAI:" + text + "\n", Color.red);
enableDisableActions();
+ endTime = System.currentTimeMillis();
+ GraphicLib.appendToPane(console, "Done. Total time: " + (endTime - startTime) + " ms\n", Color.black);
+
}
--
GitLab