Telegram_bot.R

#file.edit(path.expand(file.path("~", ".Renviron")))
library(telegram.bot)
library(stringr)
# Initiate the bot session using the token from the enviroment variable.
bot = Bot(token = bot_token('your_bot'))
usr_list <- c(12344566, 12345566)
create_image <- function(prompt, chat_id) {
  if (chat_id %in% usr_list) {
    if (stringr::str_detect(prompt,"seed=|seed:")) {
      seed <- as.numeric(
        str_extract(prompt, "[^=:]*$")
      )
      prompt <- prompt |> str_remove("[^=:]*$") |> str_remove("(seed:)|(seed=)")

    } else {
      seed <- sample.int(1e+9, 1)
      prompt <- prompt
    }
    sendtopy <- paste0(
      'conda run -n ldm python scripts/txt2img.py --prompt "', prompt,
      '" --plms --skip_grid --n_samples 1 --seed ', seed
    )
    system(sendtopy)
    df <- file.info(list.files("~/Projects/stable-diffusion/outputs/txt2img-samples/samples", full.names = T))
    most_recent <- rownames(df)[which.max(df$mtime)]
    bot$sendPhoto(chat_id = chat_id, photo = most_recent)
    message_to_bot <- paste0(
      "The above image was generated by stable diffusion using prompt: ", prompt,
      " and seed:", seed
    )
    bot$sendMessage(chat_id = chat_id, text = message_to_bot)
    
  } 
  else {
    bot$sendMessage(chat_id = chat_id, text = "You are not allowed to use this bot!")
  }
  
}

updater <- Updater(token = bot_token('your_bot'))
start <- function(bot, update){
  bot$sendMessage(chat_id = update$message$chat_id,
                  text = sprintf("Hello %s!", update$message$from$first_name))
}
start_handler <- CommandHandler("start", start)
updater <- updater + start_handler

prompt <- function(bot, update){
    create_image(prompt = stringr::str_remove(update$message$text, "/prompt"),
                 chat_id = update$message$chat_id)
}

updater <- updater + CommandHandler("prompt", prompt)
unknown <- function(bot, update){
  bot$sendMessage(chat_id = update$message$chat_id,
                  text = "Sorry, I didn't understand that command.")
}

updater <- updater + MessageHandler(unknown, MessageFilters$command)
echo <- function(bot, update){
  bot$sendMessage(chat_id = update$message$chat_id, text = update$message$text)
}

updater <- updater + MessageHandler(echo, MessageFilters$text)

updater$start_polling()

运行和停止脚本

nohup Rscript telegram_bot.R & &> ./nohup.out
ps -ax | grep telegram #查找PID
kill #对应的PID

定时清空log和图

log_clean.sh

#!/bin/bash

time=$(date "+%Y%m%d:%H-%M-%S")
echo "${time}"
echo "--我要开始清理日志啦-----"
echo '' > /home/wyih/Projects/stable-diffusion/sysLog.txt
find /home/wyih/Projects/stable-diffusion/outputs/txt2img-samples/samples/ -type f | xargs rm -rf
echo "---已经清理完毕---"

运行效果