preliminary usable features

frontend/src/components/RunButton.tsx

@@ -1,8 +1,10 @@
-import React, {FC} from 'react';
+import React, {FC, MouseEventHandler} from 'react';
 import commonStore, {ModelStatus} from '../stores/commonStore';
 import {StartServer} from '../../wailsjs/go/backend_golang/App';
 import {Button} from '@fluentui/react-components';
 import {observer} from 'mobx-react-lite';
+import {exit, readRoot, switchModel, updateConfig} from '../apis';
+import {toast} from 'react-toastify';
 
 const mainButtonText = {
   [ModelStatus.Offline]: 'Run',
@@ -12,28 +14,42 @@ const mainButtonText = {
 };
 
 const onClickMainButton = async () => {
+  const modelConfig = commonStore.getCurrentModelConfig();
+  const port = modelConfig.apiParameters.apiPort;
+
   if (commonStore.modelStatus === ModelStatus.Offline) {
     commonStore.setModelStatus(ModelStatus.Starting);
-    StartServer(commonStore.getStrategy(), `models\\${commonStore.getCurrentModelConfig().modelParameters.modelName}`);
+    StartServer(port);
 
-    let timeoutCount = 5;
+    let timeoutCount = 6;
     let loading = false;
     const intervalId = setInterval(() => {
-      fetch('http://127.0.0.1:8000')
+      readRoot()
         .then(r => {
           if (r.ok && !loading) {
             clearInterval(intervalId);
             commonStore.setModelStatus(ModelStatus.Loading);
             loading = true;
-            fetch('http://127.0.0.1:8000/update-config', {
-              method: 'POST',
-              headers: {
-                'Content-Type': 'application/json'
-              },
-              body: JSON.stringify({})
-            }).then(async (r) => {
-              if (r.ok)
+            toast('Loading Model', {type: 'info'});
+            updateConfig({
+              max_tokens: modelConfig.apiParameters.maxResponseToken,
+              temperature: modelConfig.apiParameters.temperature,
+              top_p: modelConfig.apiParameters.topP,
+              presence_penalty: modelConfig.apiParameters.presencePenalty,
+              frequency_penalty: modelConfig.apiParameters.frequencyPenalty
+            });
+            switchModel({
+              model: `models\\${modelConfig.modelParameters.modelName}`,
+              strategy: commonStore.getStrategy(modelConfig)
+            }).then((r) => {
+              if (r.ok) {
                 commonStore.setModelStatus(ModelStatus.Working);
+              } else if (r.status === 304) {
+                toast('Loading Model', {type: 'info'});
+              } else {
+                commonStore.setModelStatus(ModelStatus.Offline);
+                toast('Failed to switch model', {type: 'error'});
+              }
             });
           }
         }).catch(() => {
@@ -46,15 +62,18 @@ const onClickMainButton = async () => {
       timeoutCount--;
     }, 1000);
   } else {
-    commonStore.setModelStatus(ModelStatus.Offline);
-    fetch('http://127.0.0.1:8000/exit', {method: 'POST'});
+    exit();
   }
 };
 
-export const RunButton: FC = observer(() => {
+export const RunButton: FC<{ onClickRun?: MouseEventHandler }> = observer(({onClickRun}) => {
   return (
     <Button disabled={commonStore.modelStatus === ModelStatus.Starting} appearance="primary" size="large"
-            onClick={onClickMainButton}>
+            onClick={async (e) => {
+              if (commonStore.modelStatus === ModelStatus.Offline)
+                await onClickRun?.(e);
+              await onClickMainButton();
+            }}>
       {mainButtonText[commonStore.modelStatus]}
     </Button>
   );