Issue 371

src/main/java/com/devoxx/genie/action/CalcTokensForDirectoryAction.java

@@ -1,8 +1,10 @@
 package com.devoxx.genie.action;
 
+import com.devoxx.genie.controller.listener.TokenCalculationListener;
 import com.devoxx.genie.model.enumarations.ModelProvider;
 import com.devoxx.genie.service.TokenCalculationService;
 import com.devoxx.genie.ui.settings.DevoxxGenieStateService;
+import com.devoxx.genie.ui.util.NotificationUtil;
 import com.intellij.openapi.actionSystem.ActionUpdateThread;
 import com.intellij.openapi.actionSystem.AnActionEvent;
 import com.intellij.openapi.actionSystem.CommonDataKeys;
@@ -11,11 +13,12 @@
 import com.intellij.openapi.vfs.VirtualFile;
 import org.jetbrains.annotations.NotNull;
 
-public class CalcTokensForDirectoryAction extends DumbAwareAction {
+public class CalcTokensForDirectoryAction extends DumbAwareAction implements TokenCalculationListener {
+    private Project project;
 
     @Override
     public void actionPerformed(@NotNull AnActionEvent e) {
-        Project project = e.getProject();
+        this.project = e.getProject();
         VirtualFile selectedDir = e.getData(CommonDataKeys.VIRTUAL_FILE);
 
         if (project == null || selectedDir == null || !selectedDir.isDirectory()) {
@@ -28,7 +31,7 @@ public void actionPerformed(@NotNull AnActionEvent e) {
         int maxTokens = stateService.getDefaultWindowContext();
 
         new TokenCalculationService()
-            .calculateTokensAndCost(project, selectedDir, maxTokens, selectedProvider, null, false);
+            .calculateTokensAndCost(project, selectedDir, maxTokens, selectedProvider, null, false, this);
     }
 
 
@@ -47,4 +50,9 @@ public void update(@NotNull AnActionEvent e) {
     public boolean isDumbAware() {
         return true;
     }
+
+    @Override
+    public void onTokenCalculationComplete(String message) {
+        NotificationUtil.sendNotification(project, message);
+    }
 }

src/main/java/com/devoxx/genie/controller/ActionButtonsPanelController.java

@@ -0,0 +1,168 @@
+package com.devoxx.genie.controller;
+
+import com.devoxx.genie.chatmodel.ChatModelProvider;
+import com.devoxx.genie.controller.listener.PromptExecutionListener;
+import com.devoxx.genie.model.LanguageModel;
+import com.devoxx.genie.model.enumarations.ModelProvider;
+import com.devoxx.genie.model.request.ChatMessageContext;
+import com.devoxx.genie.service.DevoxxGenieSettingsService;
+import com.devoxx.genie.ui.EditorFileButtonManager;
+import com.devoxx.genie.ui.component.input.PromptInputArea;
+import com.devoxx.genie.ui.panel.ActionButtonsPanel;
+import com.devoxx.genie.ui.panel.PromptOutputPanel;
+import com.devoxx.genie.ui.settings.DevoxxGenieStateService;
+import com.devoxx.genie.ui.util.NotificationUtil;
+import com.devoxx.genie.util.ChatMessageContextUtil;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.ComboBox;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import static com.devoxx.genie.model.enumarations.ModelProvider.*;
+
+public class ActionButtonsPanelController implements PromptExecutionListener {
+
+    private final Project project;
+    private final EditorFileButtonManager editorFileButtonManager;
+    private final PromptInputArea promptInputArea;
+    private final ComboBox<ModelProvider> modelProviderComboBox;
+    private final ComboBox<LanguageModel> modelNameComboBox;
+    private final ChatModelProvider chatModelProvider = new ChatModelProvider();
+    private final ActionButtonsPanel actionButtonsPanel;
+    private final PromptExecutionController promptExecutionController;
+    private final ProjectContextController projectContextController;
+    private final TokenCalculationController tokenCalculationController;
+
+    public ActionButtonsPanelController(Project project,
+                                        PromptInputArea promptInputArea,
+                                        PromptOutputPanel promptOutputPanel,
+                                        ComboBox<ModelProvider> modelProviderComboBox,
+                                        ComboBox<LanguageModel> modelNameComboBox,
+                                        ActionButtonsPanel actionButtonsPanel) {
+
+        this.project = project;
+        this.promptInputArea = promptInputArea;
+        this.editorFileButtonManager = new EditorFileButtonManager(project, null);
+        this.modelProviderComboBox = modelProviderComboBox;
+        this.modelNameComboBox = modelNameComboBox;
+        this.actionButtonsPanel = actionButtonsPanel;
+        this.promptExecutionController = new PromptExecutionController(project, promptInputArea, promptOutputPanel, actionButtonsPanel);
+        this.projectContextController = new ProjectContextController(project, modelProviderComboBox, modelNameComboBox, actionButtonsPanel);
+        this.tokenCalculationController = new TokenCalculationController(project, modelProviderComboBox, modelNameComboBox, actionButtonsPanel);
+    }
+
+    public boolean isPromptRunning() {
+        return promptExecutionController.isPromptRunning();
+    }
+
+    public boolean handlePromptSubmission(String actionCommand,
+                                          boolean isProjectContextAdded,
+                                          String projectContext) {
+
+        String userPromptText = getUserPromptText();
+
+        if (userPromptText == null) {
+            return false;
+        }
+
+        ChatMessageContext currentChatMessageContext =
+                ChatMessageContextUtil.createContext(project,
+                        userPromptText,
+                        getSelectedLanguageModel(),
+                        chatModelProvider,
+                        actionCommand,
+                        editorFileButtonManager,
+                        projectContext,
+                        isProjectContextAdded);
+
+        return promptExecutionController.handlePromptSubmission(currentChatMessageContext);
+    }
+
+    /**
+     * Stop the prompt execution.
+     */
+    @Override
+    public void stopPromptExecution() {
+        promptExecutionController.stopPromptExecution();
+    }
+
+    @Override
+    public void startPromptExecution() {
+        promptExecutionController.startPromptExecution();
+    }
+
+    @Override
+    public void endPromptExecution() {
+        promptExecutionController.endPromptExecution();
+    }
+
+    private LanguageModel getSelectedLanguageModel() {
+        DevoxxGenieStateService stateService = DevoxxGenieStateService.getInstance();
+        LanguageModel selectedLanguageModel = (LanguageModel) modelNameComboBox.getSelectedItem();
+
+        // If selectedLanguageModel is null, create a default one
+        if (selectedLanguageModel == null) {
+            selectedLanguageModel = createDefaultLanguageModel(stateService);
+        }
+        return selectedLanguageModel;
+    }
+
+    /**
+     * get the user prompt text.
+     */
+    private @Nullable String getUserPromptText() {
+        String userPromptText = promptInputArea.getText();
+        if (userPromptText.isEmpty()) {
+            NotificationUtil.sendNotification(project, "Please enter a prompt.");
+            return null;
+        }
+        return userPromptText;
+    }
+
+    /**
+     * Create a default language model.
+     *
+     * @param stateService the state service
+     * @return the default language model
+     */
+    private LanguageModel createDefaultLanguageModel(@NotNull DevoxxGenieSettingsService stateService) {
+        ModelProvider selectedProvider = (ModelProvider) modelProviderComboBox.getSelectedItem();
+        if (selectedProvider != null &&
+                (selectedProvider.equals(LMStudio) ||
+                        selectedProvider.equals(GPT4All) ||
+                        selectedProvider.equals(Jlama) ||
+                        selectedProvider.equals(LLaMA))) {
+            return LanguageModel.builder()
+                    .provider(selectedProvider)
+                    .apiKeyUsed(false)
+                    .inputCost(0)
+                    .outputCost(0)
+                    .contextWindow(4096)
+                    .build();
+        } else {
+            String modelName = stateService.getSelectedLanguageModel(project.getLocationHash());
+            return LanguageModel.builder()
+                    .provider(selectedProvider != null ? selectedProvider : OpenAI)
+                    .modelName(modelName)
+                    .apiKeyUsed(false)
+                    .inputCost(0)
+                    .outputCost(0)
+                    .contextWindow(128_000)
+                    .build();
+        }
+    }
+
+    /**
+     * Initiates the calculation of tokens and cost based on the selected model and provider.
+     * It delegates the actual calculation to the TokenCalculationController.
+     */
+    public void calculateTokensAndCost() {
+        tokenCalculationController.calculateTokensAndCost();
+    }
+
+    public void updateButtonVisibility() {
+        boolean isSupported = projectContextController.isProjectContextSupportedProvider();
+        actionButtonsPanel.setCalcTokenCostButtonVisible(isSupported);
+        actionButtonsPanel.setAddProjectButtonVisible(isSupported);
+    }
+}