iT邦幫忙

2021 iThome 鐵人賽

DAY 20
1
AI & Data

建立FF14資訊Discord chatbot系列 第 20

[DAY 20]用bot打出色色柴犬counter牌(更新句子相似度判斷)

  • 分享至 

  • xImage
  •  

原本以為色色柴犬counter功能不會再更新了

直到discord上來了這樣一段


哇他們這樣玩還沒觸發counter讓我很沒面子你們知道ㄇ

之前的判斷很簡單就是有"可以色色"跟"不可以色色"的訊息出現才會打出counter牌

這次我直接用TF來得出句子相似度,函式參考這篇

完整代碼如下,一樣也是用cog架構

import pandas as pd
from discord.ext import commands
from core.classes import Cog_Extension
import discord
from random import choice
from sklearn.feature_extraction.text import CountVectorizer
import numpy as np
from scipy.linalg import norm
class main(Cog_Extension):
    def __init__(self,*args,**kwargs):
        super().__init__(*args,**kwargs)
        self.table=[]
        self.nohokh = ["https://i.imgur.com/cZksr04.jpg","https://i.imgur.com/wH4CJLF.jpg","https://i.imgur.com/ag0DVMT.jpg"]
    
    def tf_similarity(self,s1, s2):
        def add_space(s):
            return ' '.join(list(s))

        # 將字中間加入空格
        s1, s2 = add_space(s1), add_space(s2)
        # 轉化為TF矩陣
        cv = CountVectorizer(tokenizer=lambda s: s.split())
        corpus = [s1, s2]
        vectors = cv.fit_transform(corpus).toarray()
        # 計算TF係數
        return np.dot(vectors[0], vectors[1]) / (norm(vectors[0]) * norm(vectors[1]))
    
    
    
    @commands.Cog.listener()
    #當有訊息時
    async def on_message(self,message):
        embed = discord.Embed()
        #排除自己的訊息,避免陷入無限循環
        if message.author == self.bot.user:
            return
        
        noh_sim = self.tf_similarity(message.content, "不可以色色")
        okh_sim = self.tf_similarity(message.content, "可以色色")
        
        #不可以色色排組對應圖片
        if noh_sim > 0.7 and noh_sim > okh_sim:
            if "可以色色" and "不可以色色" in self.table:
                self.table=[]
                await message.reply(choice(self.nohokh))#特招
            else:
                noh = ["https://imgur.dcard.tw/TUGMyF7h.jpg","https://imgur.dcard.tw/ZayW5My.jpg","https://i.imgur.com/WLbQBlT.jpg","https://i.imgur.com/nGI4pbO.jpg","https://i.imgur.com/jrOvlgH.jpg","https://i.imgur.com/68mqoce.jpg"]
                self.table.append("不可以色色")
                await message.reply(choice(noh))#不可以色色康特
            # await message.channel.send(embed=embed)
        elif okh_sim > 0.7 and noh_sim < okh_sim:
            if "可以色色" and "不可以色色" in self.table:
                self.table=[]
                await message.reply(choice(self.nohokh))#特招
            else:
                okh = ["https://i.imgur.com/UVZti2m.jpg","https://i.imgur.com/FoGDvP7.jpg","https://i.imgur.com/c2pDzLJ.jpg","https://i.imgur.com/0Ux7atA.jpg","https://i.imgur.com/hnnSpzr.jpg","https://i.imgur.com/F03Raad.jpg"]
                self.table.append("可以色色")
                await message.reply(choice(okh))#可以色色康特
        


def setup(bot):
    bot.add_cog(main(bot))

呈現結果如下:

已經把程式碼更新上去了,等著看好戲^^

至於頻道上提到的透過關鍵字觸發讓某人去其他頻道是可行的

看情況我會把這功能寫出來


上一篇
[DAY 19]bot在線狀態通知
下一篇
[DAY 21]糾團通知功能(1/3)
系列文
建立FF14資訊Discord chatbot30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

1 則留言

0
虎虎
iT邦研究生 4 級 ‧ 2021-10-05 08:38:59

哇!坐等後續,能這樣應用真的太有趣了!!!
話說這個系列不直接上 Github嗎 XD

我要留言

立即登入留言