fixing copilot

apps/remix-ide/src/app/plugins/copilot/suggestion-service/copilot-suggestion.ts

@@ -20,8 +20,11 @@ export class CopilotSuggestion extends Plugin {
   ready: boolean
   constructor() {
     super(profile)
-    this.service = new SuggestionService()
     this.context = ''
+  }
+
+  onActivation(): void {
+    this.service = new SuggestionService()
     this.service.events.on('progress', (data) => {
       this.emit('loading', data)
     })

apps/remix-ide/src/app/plugins/copilot/suggestion-service/suggestion-service.ts

@@ -1,11 +1,13 @@
 import EventEmitter from 'events'
 
-export type SuggestOptions = { max_new_tokens: number, 
+export type SuggestOptions = {
+  max_new_tokens: number, 
   temperature: number,
   do_sample:boolean
   top_k: number,
   top_p:number, 
-                               stream_result:boolean}
+  stream_result:boolean
+}
 
 export class SuggestionService {
   worker: Worker
@@ -17,12 +19,12 @@ export class SuggestionService {
     this.worker = new Worker(new URL('./worker.js', import.meta.url), {
       type: 'module'
     });
-    this.init()
     this.events = new EventEmitter()
     this.responses = {}
     this.current
   }
 
+  //todo ask Yann if we should keep the model
   terminate(): void {
     this.worker.terminate()
     this.worker = new Worker(new URL('./worker.js', import.meta.url), {

apps/remix-ide/src/app/plugins/copilot/suggestion-service/worker.js

@@ -22,6 +22,7 @@ class CodeCompletionPipeline {
 
 // Listen for messages from the main thread
 self.addEventListener('message', async (event) => {
+  console.log("worker message ", event.data)
   const {
     id, model, text, max_new_tokens, cmd,
 

apps/remix-ide/src/app/tabs/settings-tab.tsx

@@ -60,6 +60,12 @@ module.exports = class SettingsTab extends ViewPlugin {
     this.renderComponent()
   }
 
+  onActivation(): void {
+    this.on('copilot-suggestion', 'loading', (data) => {
+      this.call('terminal', 'log', { type: 'typewritererror', value: `.` })
+      console.log("oninit")
+    })
+  }
   render() {
     return (
       <div id="settingsTab">

libs/remix-ui/settings/src/lib/remix-ui-settings.tsx

@@ -4,6 +4,12 @@ import React, {useState, useRef, useReducer, useEffect, useCallback} from 'react
 import {AppModal, AlertModal, ModalTypes} from '@remix-ui/app'
 import {labels, textDark, textSecondary} from './constants'
 
+enum CONSENT {
+  GIVEN = 0,
+  NOT_GIVEN,
+  NOT_ASKED
+}
+
 import './remix-ui-settings.css'
 import {
   generateContractMetadat,
@@ -56,6 +62,7 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
   const [ipfsProjectSecret, setipfsProjectSecret] = useState('')
   const copilotDownload = useRef(null)
 
+  let consentGivenForAI = CONSENT.NOT_ASKED
   const intl = useIntl()
   const initValue = () => {
     const metadataConfig = props.config.get('settings/generate-contract-metadata')
@@ -123,7 +130,12 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
   }, [props.useMatomoAnalytics])  
 
   useEffect(() => {
+    console.log("useEffect on useCopilot")
     if (props.useCopilot !== null) copilotActivate(props.config, props.useCopilot, dispatch)
+    if (props.useCopilot) {
+      const a = async () => await onchangeCopilotActivate()
+    }
+    console.log("useEffect on useCopilot finish")
   }, [props.useCopilot])
   
   const onchangeGenerateContractMetadata = (event) => {
@@ -134,9 +146,10 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
     textWrapEventAction(props.config, props.editor, event.target.checked, dispatch)
   }
 
-  const onchangeCopilotActivate = async (event) => {
-    if (!event.target.checked) {
-      copilotActivate(props.config, event.target.checked, dispatch)
+  const onchangeCopilotActivate = async () => {
+    console.log("onchangeCopilotActivate ", props.useCopilot)
+    if (!props.useCopilot) {
+      copilotActivate(props.config, props.useCopilot, dispatch)
       props.plugin.call('copilot-suggestion', 'uninstall')
       return
     } 
@@ -148,13 +161,8 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
       const dot = loaded.match(/(.*)\./g)
       copilotDownload.current.innerText = dot ? dot[0].replace('.', '') : loaded
     })
-    const modalActivate: AppModal = {
-      id: 'loadcopilotActivate',
-      title: 'Download Solidity copilot',
-      modalType: ModalTypes.default,
-      okLabel: 'Close',
-      message,
-      okFn: async() => {
+    const startCopilot = async () => {
+      await props.plugin.call('copilot-suggestion', 'init')
       props.plugin.off('copilot-suggestion', 'loading')
       if (await props.plugin.call('copilot-suggestion', 'status')) {
         copilotActivate(props.config, true, dispatch)          
@@ -162,16 +170,36 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
         props.plugin.call('copilot-suggestion', 'uninstall')
         copilotActivate(props.config, false, dispatch)
       }
+    }
+    const modalActivate: AppModal = {
+      id: 'loadcopilotActivate',
+      title: 'Download Solidity copilot',
+      modalType: ModalTypes.default,
+      okLabel: 'OK',
+      //cancelLabel: 'Cancel',
+      message,
+      okFn: async() => {
+        consentGivenForAI = CONSENT.GIVEN
+        startCopilot()
       },
       hideFn: async () => {
+        consentGivenForAI = CONSENT.NOT_GIVEN
         props.plugin.off('copilot-suggestion', 'loading')
-        if (await props.plugin.call('copilot-suggestion', 'status')) {
-          copilotActivate(props.config, true, dispatch)          
-        } else {
-          props.plugin.call('copilot-suggestion', 'uninstall')
-          copilotActivate(props.config, false, dispatch)
+        // if (await props.plugin.call('copilot-suggestion', 'status')) {
+        //   copilotActivate(props.config, true, dispatch)          
+        // } else {
+        //   props.plugin.call('copilot-suggestion', 'uninstall')
+        //   copilotActivate(props.config, false, dispatch)
+        // }
       }
     }
+    
+    if (consentGivenForAI === CONSENT.NOT_ASKED) {
+      props.plugin.call('notification', 'modal', modalActivate)
+    } else if (consentGivenForAI === CONSENT.GIVEN) {
+      startCopilot()
+    } else {
+      // NOT_GIVEN
     }
 
     if (await props.plugin.call('copilot-suggestion', 'status')) {
@@ -180,9 +208,6 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
       props.plugin.call('copilot-suggestion', 'uninstall')
       copilotActivate(props.config, false, dispatch)
     }
-
-    props.plugin.call('copilot-suggestion', 'init')
-    props.plugin.call('notification', 'modal', modalActivate)
  }
 
   const onchangeCopilotMaxNewToken = (event) => {
@@ -460,7 +485,7 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
     copilotTemperatureValue = 0.5
   }
 
-  if (isCopilotActivated) props.plugin.call('copilot-suggestion', 'init')
+  //if (isCopilotActivated) props.plugin.call('copilot-suggestion', 'init')
   const copilotSettings = () => (
     <div className="border-top">
       <div className="card-body pt-3 pb-2">
@@ -468,19 +493,6 @@ export const RemixUiSettings = (props: RemixUiSettingsProps) => {
           <FormattedMessage id="settings.copilot" />
         </h6>
 
-        <div className="pt-2 mb-0">
-          <div className="text-secondary mb-0 h6">
-            <div>
-              <div className="custom-control custom-checkbox mb-1">
-                <input onChange={onchangeCopilotActivate} id="copilot-activate" type="checkbox" className="custom-control-input" checked={isCopilotActivated} />
-                <label className={`form-check-label custom-control-label align-middle ${getTextClass('settings/copilot/suggest/activate')}`} htmlFor="copilot-activate">
-                  <FormattedMessage id="settings.copilot.activate" />
-                </label>
-              </div>
-            </div>
-          </div>
-        </div>
-
         <div className="pt-2 mb-0">
           <div className="text-secondary mb-0 h6">
             <div>

libs/remix-ui/tabs/src/lib/remix-ui-tabs.tsx

@@ -267,7 +267,7 @@ export const TabsUI = (props: TabsUIProps) => {
               className="btn ai-switch text-ai pl-2 pr-0 py-0 d-flex"
               disabled={!(tabsState.currentExt === 'sol' )}
               onClick={async () => {
-                await props.plugin.call('settings', 'updateCopilotChoice', ai_switch)
+                await props.plugin.call('settings', 'updateCopilotChoice', !ai_switch)
                 setAI_switch(!ai_switch)
               }}
             >