よくわからない現代魔法 RubyなTwitter bot作成メモ

/ 2024/04/25 (Thu) / 編集

[PR]上記の広告は3ヶ月以上新規記事投稿のないブログに表示されています。新しい記事を書く事で広告が消えます。

Libro-Twitter-Bot / 2010/06/12 (Sat) / 編集

というわけで、お返事内容を考えるためのクラス作り。

bot（人口無能）には相手の発言をごにょごにょするマルコフ型や、過去ログを元に返事を考えるログ型などがあるけれど、マルコフ型だと自然な文章は作りづらいし、ログ型だとログがたまらないとダメダメなので今回は辞書型で行きます。
最終的には学習システムを実装して「自分で辞書を拡張する辞書型人口無能」を目指すつもり。

今回の実装では「文章のみから返事を考える」ことを目的としたいのでTwitterAPIは使いません。与えられる情報はあくまで「届いたメッセージ」と「送信者の名前」のみ！なのでTwitter以外のbotへも流用可能です。

実装したい事

テンプレを使える事
コマンドが使えること
言い換えに対応してること
自分宛の発言と全員宛の発言を切り分けできること
独り言をしゃべること

テンプレに対応する

まずはテンプレに対応する事、「%user-likeFood%が好き」とかすると「沖縄そばが好き」にしてくれるような関数を作ります。

データベースには「ユーザ名」「ステータス(likeFoodとか)」「内容（沖縄そば）」を入れておくようにする。なのでカラムは「name」「key」「value」と識別用の「id」を入れて４つ。

sql ='create table personal ('
sql+='id integer primary key'
sql+=',name text not null'
sql+=',key text not null'
sql+=',value text not null'
sql+=');'
@talkdb.execute(sql)
sql ='insert into personal values (:id,:name,:key,:value);'			@talkdb.execute(sql,:name=>'my',:key=>'name',:value=>'#{$config[:MY_NAME@talkdb.execute(sql,:name=>'my',:key=>'likeFood',:value=>'沖縄そば')
@talkdb.execute(sql,:name=>'testUser',:key=>'name',:value=>'テストなユーザ')

以下のようなレコード追加用関数も用意しとく、以降のテーブルにも同じようなのを用意しとく。

def append_personal(name='',key='',value='')
  sql ='insert into personal values (:id,:name,:key,:value);'
  begin
    return @talkdb.execute(sql,:name=>name,:key=>key,:value=>value.chomp) if(name!='' and key!='' and value!='') 
  rescue SQLite3::SQLException
    time = Time.now.strftime("%Y/%m/%d %H:%M:%S")
    puts "#{time}\t[ERROR:append_personal] database access error"
  end
  return nil
end
<textarea>
<br>
ここからが本題、今回は「%name_key%」という文字列を「value」に置き換える関数。ただし、keyがuserなら投稿者のユーザ名に置き換える、また「%user_name%」の時は例外的に”データベースに存在しないときは投稿者名を代入する”ようにする。<br>
<textarea class="code">
#あらかじめ設定されている文字列を置き換え
def replace(str,from_name='')
  #まずは「%name_key%」を探す
  source = str.scan(/%[a-zA-Z0-9]*_[a-zA-Z0-9]*%/)
  source.each{|command|
    #nameとkeyに分割する。
    keys = command.gsub("%",'').split('_')
    #keyがuserなら投稿者のユーザ名に置き換え
    keys[0] = from_name if (keys[0] == 'user')
    result = @talkdb.execute("select value from personal where name=='#{keys[0]}' and key=='#{keys[1]}';")
    result.flatten! if(result!=nil)
    str.sub!(command,result[0]) if(result[0]!=nil)

    str.sub!(command,from_name) if(result[0]==nil and keys[1]=='name')
  }
  return str
end

以下のようスクリプトを実行すると

require 'TalkClass'
talk = TalkClass.new
p talk.replace("test %my_likeFood% dayo",'testUser')
p talk.replace("test %user_name% dayo",'testUser')
p talk.replace("test %user_name% dayo",'othertUser')

データベース中で該当する者をさがしてきて置き換えてくれる。

"test 沖縄そば dayo"
"test テストなユーザ dayo"
"test othertUser dayo"

コマンド

お継ぎはコマンド。「＜コマンド名＞」という部分を置き換えてくれる関数。replaceとの違いは”データベースに登録しておけない情報”というところで、例えば現在の時刻(nowTime)などを置き換える。

def command(str)
  source = str.scan(/<[a-zA-Z0-9]*>/)
  source.each{|command|
    cmd = command.gsub("<",'').gsub(">",'')
    case cmd
    when "NowTime"
      answer = Time.now.strftime("%H:%M:%S")
    when "Daet"
      answer = Time.now.strftime("%Y/%m/%d")
    end
    str.sub!(command,answer) if(answer!=nil)
  }
  return str
end

scan関数大活躍w　replaceと違ってデータベースを増やして対応する訳ではないので、新しい処理を追加したくなったらプログラムを書き換えないといけない。

require 'TalkClass'
talk = TalkClass.new
p talk.command("<NowTime>")

実行すると

"21:56:32"

返事を考える＆言い換えの対応

まずはお返事テーブル(reply)を作り「こんにちわ」と言われたら「こんにちわ！」と返事するように作る、そして言い換えテーブル(trans)「こんちわ」も「こん」も「こんにちは」も、すべて「こんにちわ」と言われたものとして処理する、そうすればより多くの言葉に対応できる！

replyテーブルのカラムは「get」「say」「flag」の３つ、getと言われたらsayと返す、ただしflagを使って”自分宛のとき”と”全員宛のとき”で使うか使わないか選べるようにする。そうすることで、他の誰かに向かって「好きな食べ物は？」と聞いているのに自分が答えちゃうとかをなくします。

transテーブルは「key」と「value」を用意し、このvalueがreplyテーブルのgetと同じになるように作ってあります。
このsearch_choicesには句読点等で区切った文節が渡されます（後述）、返り値は返答の選択肢を入れた配列です。（sayには「|」区切りで複数の返答方法が入っています。）

def search_choices(str,personal_mode=true)
  result=nil
  begin
    #1)単語が言い換えテーブルにあるか調べる
    new_key = @talkdb.execute("select value from trans where key=='#{str}';")
    new_key.flatten! if(new_key!=nil)
    #2)あったら返事テーブルを調べる
    if (personal_mode) then
      choices = @talkdb.execute("select say from reply where get=='#{new_key[0]}';")
    else
      choices = @talkdb.execute("select say from reply where get=='#{new_key[0]}' and flag=='SAY_TO_US';")
    end
    #3)返事の種類を配列で返す
    if (choices!=nil) then
      choices.flatten!
      result = choices[0].split("|") if (choices[0]!=nil)
    end
  rescue SQLite3::SQLException
    time = Time.now.strftime("%Y/%m/%d %H:%M:%S")
    puts "#{time}\t[ERROR:search_choices] database access error"
  end
  return result
end

そしてこの関数を使って返事を考えるのが次の関数、区切り文字に従って文章を文節にわけ、search_choicesに渡すのが主な仕事。さきほど紹介したreplaceとcommandでテンプレ対応とコマンド対応もしとく。

#誰かから話しかけられたときの返事
def think_answer(from_name='',tweet='',mode='')
  return '' if (from_name=='' or tweet=='')
  message=''
  answer=''
  #(1) 区切り文字で分割
  phrase = tweet.split(/[,.!?、。！？\s\n]/)
  phrase.delete("")
  #(2)文節ごとに返答を考える
  phrase.each{|str|
    choices = search_choices(str,true) if (mode=='reply') #返事をする
    choices = search_choices(str,false) if (mode=='say') #話しかける
    message+=choices[rand(choices.size)] if(choices!=nil)
  }
  answer_tmp = replace(message,from_name) #テンプレ対応
  answer = command(answer_tmp) #コマンド対応
  return "@#{from_name} #{answer}" if(answer!='')
  return ''
end

ここまででで以下のようなことができるよ。

require 'TalkClass'
talk = TalkClass.new
p talk.think_answer('testUser',"こんにちは!今何時？",'reply')
p talk.think_answer('testUser',"こんにちは!今何時？",'say')

実行すると以下のようになります。「今何時？」はお返事(reply)の時にしか使わないようにしてあるので話しかける(say)ときは無視されます。

"@testUser こんにちは！22:44:57だよ。"
"@testUser こんちわ！"

しかしここで問題が一つ、ツィートかダイレクトメッセージかはAPIで取得する時点でわかるけど自分宛のツィート（@myname）か全員宛のツィートかは読んでみないとわからない！
そこでこうゆう関数を用意します。ダイレクトメッセージ(DM)かツィート(TW)かは取得時にわかるはずなので引数で。

#メッセージを分類する。
def parse(type='',from_name='',tweet='')
  reply_message=''
  case type
  when 'DM'
    reply_message='D #{from_name} お手紙ありがとう！後で返事書くからね'
  when 'TW'
    if (/@#{$config[:MY_NAME]}/ =~ tweet) then #自分宛のツィート
      puts "#{from_name}から話しかけられました"
      reply_message = think_answer(from_name,tweet,'reply') #返事をする
    else #それ以外
      puts "#{from_name}がなにか言ってます"
      reply_message = think_answer(from_name,tweet,'say') #話しかける
    end
  end
  return reply_message
end
<textarea>
ためしに以下のスクリプトを実行、ただし「#{$config[:MY_NAME]}」にはbotのユーザ名が入ってます。
<pre class="code">
load 'config.rb',true
require 'TalkClass'
p talk.parse('TW','testUser',"@#{$config[:MY_NAME]} こんにちは！今何時？")
p talk.parse('TW','testUser',"こんにちは！今何時？")
talk = TalkClass.new
</pre>
実行すると、以下のように自分宛のメッセージかどうか自動で判定します。
<pre class="cmd">
testUserから話しかけられました
"@testUser こんにちは！23:00:08だよ。"
testUserがなにか言ってます
"@testUser こんにちは！"
</pre>
さぁ！もう一息だお！！

<a name="message"><h2>独り言を言う</h2></a>
まぁTwitterなので、いつまでたってもだれも話しかけてくれないなんてこともあります。なので独り言を言うのは必要！寂しくなんかないよ！！<br>
<br>
今回は時間帯別にいろんなことをしゃべります。テーブル(message)にはメッセージ内容(message)と投稿する時間帯(time)が保存されています。<br>
timeは「morning」や「night」などの時間帯、もしくは「1830」などの２４時間表記の時間です。てことはなにが面倒か？時間帯の判定ですね。。。<br>
<textarea class="code">
#独り言をしゃべる
def think_tweet(nowH=0,nowM=0)
  #(1)現在時刻を調べます、ただし10分くぎりです。
  nowH = Time.now.strftime("%H").to_i if(nowH==0)
  nowM = Time.now.strftime("%M").to_i if(nowM==0)
  nowM = (nowM/10)*10 #10分区切りではかります。
  time = nowH*100+nowM
  time_name = ''

  #(2)時間帯を判定
  case time
  when 600 .. 1200
    time_name = 'morning'
  when 1430 .. 1530
    time_name = 'teaTime'
  when 1200 .. 1700
    time_name = 'afterNoon'
  when 1700 .. 2000
    time_name = 'evening'
  else
    time_name = 'night'
  end

  #(3)テーブルから引いてくる
  result=nil
  sql = "select message from speak where time=='every' or time=='#{time_name}' or time=='#{time}';"
  begin
    result = @talkdb.execute(sql);
  rescue SQLite3::SQLException
    error_time = Time.now.strftime("%Y/%m/%d %H:%M:%S")
    puts "#{error_time}\t[ERROR:think_tweet] database access error"
  end
  result.flatten! if(result!=nil)

  #選択肢の中から適当に１つ選ぶ
  message =  result[rand(result.size)]
  message_tmp = replace(message)#テンプレ対応
  return command(message_tmp)#コマンド対応
end

ちなみに引数を用意してあるのはテスト用で、別に引数なしでも動きます。

load 'config.rb',true
require 'TalkClass'

talk = TalkClass.new
p talk.think_tweet
p talk.think_tweet

今回は２３時に実行したので以下のようなものがでてきました。

"誰か構ってよ、、、"
"夜ですね、わかります。"

[0回]

お名前：
URL：		メール：
タイトル：		文字色：
コメント：
パス：