Auto-Photoshop-StableDiffusion-Plugin/sdapi.js

//javascript plugin can't read images from local directory so we send a request to local server to read the image file and send it back to plugin as image string base64
async function getInitImage (init_image_name) {
  console.log('getInitImage(): get Init Image from the server :')
  payload = {
    init_image_name: init_image_name
  }

  const full_url = 'http://127.0.0.1:8000/getInitImage/'
  console.log(full_url)
  console.log('getInitImage payload:', payload)
  let request = await fetch(full_url, {
    method: 'POST',
    headers: {
      Accept: 'application/json',
      'Content-Type': 'application/json'
    },
    body: JSON.stringify(payload)
    // "body": payload
  })

  let json = await request.json()
  console.log('json:')
  console.dir(json)
  base64data = json.init_image_str
  image_src = `data:image/png;base64, ${base64data}`
  return image_src

  // console.log(img.src)

  // let img_blob =  await (await fetch(img.src)).blob()
  // console.log("img_blob:")
  // console.dir(img_blob)
}

async function requestTxt2Img (payload) {
  // const url = "http://127.0.0.1:7860"
  // const full_url =`${url}/sdapi/v1/txt2img`

  // payload = {
  //     "prompt": "puppy dog",
  //     "steps": 5
  // }

  //   payload = {
  //     "prompt": "cute cat, kitten",
  //     "steps": 10
  //   }

  // response = requests.post(, json=payload)

  console.log('requestTxt2Img(): about to send a fetch request')

  const full_url = 'http://127.0.0.1:8000/txt2img/'
  console.log(full_url)

  let request = await fetch(full_url, {
    method: 'POST',
    headers: {
      Accept: 'application/json',
      'Content-Type': 'application/json'
    },
    body: JSON.stringify(payload)
    // "body": payload
  })

  let json = await request.json()
  console.log('json:')
  console.dir(json)
  




  // base64data = json.images[0]
  // let img = document.getElementById('img1')
  // img.src = `data:image/png;base64, ${base64data}`
  // console.log(img.src)

  // let img_blob =  await (await fetch(img.src)).blob()
  // console.log("img_blob:")
  // console.dir(img_blob)

  // try {
  //   navigator.clipboard.write([
  //       new ClipboardItem({
  //           'image/png': img_blob
  //       })
  //   ]);
  // } catch (error) {
  //   console.error(error);
  // }

  //  request.data (data =>{
  //     console.log(data);
  // });
  return json
}

async function requestImg2Img (payload) {
  console.log('requestTxt2Img(): about to send a fetch request')

  const full_url = 'http://127.0.0.1:8000/img2img/'
  console.log(full_url)
  console.log('requestImg2Img payload is: ', payload)
  let request = await fetch(full_url, {
    method: 'POST',
    headers: {
      Accept: 'application/json',
      'Content-Type': 'application/json'
    },
    body: JSON.stringify(payload)
    // "body": payload
  })

  let json = await request.json()
  console.log('json:')
  console.dir(json)

  return json
}

async function requestProgress () {
  console.log('requestProgress: ')

  const full_url =
    'http://127.0.0.1:8000/sdapi/v1/progress?skip_current_image=false'
  let request = await fetch(full_url)
  let json = await request.json()
  console.log('progress json:')
  console.dir(json)

  return json
}

async function requestGetModels () {
  console.log('requestGetModels: ')
  let json = []
  try{

    const full_url = 'http://127.0.0.1:8000/sdapi/v1/sd-models'
    let request = await fetch(full_url)
    json = await request.json()
    console.log('models json:')
    console.dir(json)
    
  }catch(e){
    console.warn(`issues requesting from ${full_url}`,e)
  }
    return json
}

async function requestGetSamplers () {
  console.log('requestGetSamplers: ')

  const full_url = 'http://127.0.0.1:8000/sdapi/v1/samplers'
  let request = await fetch(full_url)
  let json = await request.json()
  console.log('samplers json:')
  console.dir(json)

  return json
}

async function requestSwapModel (model_title) {
  console.log('requestSwapModel: ')
  // const full_url = 'http://127.0.0.1:8000/swapModel'

  const full_url = 'http://127.0.0.1:8000/sdapi/v1/options'
  payload = {
    sd_model_checkpoint: model_title
  }
  let request = await fetch(full_url, {
    method: 'POST',
    headers: {
      Accept: 'application/json',
      'Content-Type': 'application/json'
    },
    body: JSON.stringify(payload)
    // "body": payload
  })

  let json = await request.json()

  console.log('models json:')
  console.dir(json)

  return json
}

async function requestInterrupt (model_title) {
  try {
    console.log('requestInterrupt: ')
    // const full_url = 'http://127.0.0.1:8000/swapModel'

    const full_url = 'http://127.0.0.1:8000/sdapi/v1/interrupt'
    // payload = {
    //   sd_model_checkpoint: model_title
    // }
    payload = ''
    let request = await fetch(full_url, {
      method: 'POST',
      headers: {
        Accept: 'application/json',
        'Content-Type': 'application/json'
      },
      // body: JSON.stringify(payload)
      // "body": payload
    })

    console.log("interrupt request:", request)
    let json = await request.json()

    console.log('interrupt json:')
    console.dir(json)

    return json
  } catch (e) {
    console.warn(e)
  }
}

async function getVersionRequest(){
  // version = "v0.0.0"
    console.log('requestGetSamplers: ')
  try {

    const full_url = 'http://127.0.0.1:8000/version'
    let request = await fetch(full_url)
    let json = await request.json()
    console.log('version json:',json)
    version = json['version']
    
    return version
  }catch(e){
    console.warn(e)
    version = "v0.0.0"

    return version
  }
  

}

async function changeSdUrl(new_sd_url){
  // version = "v0.0.0"
    console.log('changeSdUrl: new_sd_url:', new_sd_url)
  try {

    payload= {
      "sd_url": new_sd_url
    }

    const full_url = 'http://127.0.0.1:8000/sd_url/'
    console.log("changeSdUrl: payload: ", payload)
  let request = await fetch(full_url, {
    method: 'POST',
    headers: {
      Accept: 'application/json',
      'Content-Type': 'application/json'
    },
    body: JSON.stringify(payload)
    
  })


    
    // let json = await request.json()
    // console.log('changeSdUrl:',json)
    console.log('changeSdUrl: request: ',request)
    
    
    
  }catch(e){
    console.warn(e)
    
  }
  

}

// function printTheJSONInPrettyFormat(json) {
//   // var badJSON = document.getElementById('prettyJSONFormat').value;
//   // var parseJSON = JSON.parse(badJSON);
//   // var JSONInPrettyFormat = JSON.stringify(json, undefined, 4);
//   // return 
// }
async function loadHistory (uniqueDocumentId) {
  
  let json = {}
  try {
    payload = {
      "uniqueDocumentId":uniqueDocumentId
    }

    const full_url = 'http://127.0.0.1:8000/history/load'

    let request = await fetch(full_url, {
      method: 'POST',
      headers: {
        Accept: 'application/json',
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(payload)
    })

    json = await request.json()
    console.log('loadHistory:', json)
    // console.log('loadPromptShortcut: request: ',request)
  } catch (e) {
    console.warn(e)
  }

  return [json['image_paths'],json['metadata_jsons']]
}
async function loadPromptShortcut () {
  // console.log('loadPromptShortcut:')
  let json = {}
  try {
    payload = {}

    const full_url = 'http://127.0.0.1:8000/prompt_shortcut/load'

    let request = await fetch(full_url, {
      method: 'POST',
      headers: {
        Accept: 'application/json',
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(payload)
    })

    json = await request.json()
    console.log('loadPromptShortcut:', json)
    // console.log('loadPromptShortcut: request: ',request)
  } catch (e) {
    console.warn(e)
  }

  return json['prompt_shortcut']
}

async function savePromptShortcut (prompt_shortcut) {
  // console.log('loadPromptShortcut:')
  let json = {}
  try {
    payload = {"prompt_shortcut":prompt_shortcut}

    const full_url = 'http://127.0.0.1:8000/prompt_shortcut/save'

    let request = await fetch(full_url, {
      method: 'POST',
      headers: {
        Accept: 'application/json',
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(payload)
    })

    json = await request.json()
    console.log('savePromptShortcut:', json)
    // console.log('loadPromptShortcut: request: ',request)
  } catch (e) {
    console.warn(e)
  }

  return json['prompt_shortcut']
}

module.exports = {
  requestTxt2Img,
  requestImg2Img,
  getInitImage,
  requestProgress,
  requestGetModels,
  requestSwapModel,
  requestInterrupt,
  requestGetSamplers,
  getVersionRequest,
  changeSdUrl,
  loadPromptShortcut,
  savePromptShortcut,
  loadHistory
}