DataCollectTeleBot/bot.py at master · SamsonChoo/DataCollectTeleBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
from telegram.ext import Updater
from telegram.ext import MessageHandler, Filters
from telegram.ext import CommandHandler
import logging
import gspread
from oauth2client.service_account import ServiceAccountCredentials
import config

global rowCount
rowCount = 15
updater = Updater(token=config.Token, use_context=True)
dispatcher = updater.dispatcher

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                     level=logging.INFO)
def start(update, context):
    context.bot.send_message(chat_id=update.message.chat_id, text="""Hi, I am a data collection bot for ValteRego.
Please key in your training statement in the following format:
    [text] *space* [emotion number from 1-8] *space* [intensity number from 1-3]
To view the full list of emotions and the corresponding intensity, please use '/list'.
To view the picture of Plutchik's wheel of emotion, please use '/pic'.
To see an example of text input, please use '/eg'.
""")

start_handler = CommandHandler('start', start)
dispatcher.add_handler(start_handler)

emotions={1:{1:"serenity",2:"joy",3:"ecstasy"},
            2:{1:"pensiveness",2:"sadness",3:"grief"},
            3:{1:"acceptance",2:"trust",3:"admiration"},
            4:{1:"boredom",2:"disgust",3:"loathing"},
            5:{1:"apprehension",2:"fear",3:"terror"},
            6:{1:"annoyance",2:"anger",3:"rage"},
            7:{1:"distraction",2:"surprise",3:"amazement"},
            8:{1:"interest",2:"anticipation",3:"vigilance"}}

#emotions={1:"serenity",2:"joy",3:"ecstasy",4:"pensiveness",5:"sadness",6:"grief",7:"acceptance",8:"trust",9:"admiration",10:"boredom",11:"disgust",12:"loathing",13:"apprehension",14:"fear",15:"terror",16:"annoyance",17:"anger",18:"rage",19:"distraction",20:"surprise",21:"amazement",22:"interest",23:"anticipation",24:"vigilance"}

def list(update, context):
    for item, amount in emotions.items():
        context.bot.send_message(chat_id=update.message.chat_id, text="{} ({})".format(item, amount))

list_handler = CommandHandler('list', list)
dispatcher.add_handler(list_handler)


def pic(update, context):
    context.bot.send_photo(chat_id=update.message.chat_id, photo="https://upload.wikimedia.org/wikipedia/commons/thumb/c/ce/Plutchik-wheel.svg/2000px-Plutchik-wheel.svg.png")

pic_handler = CommandHandler('pic', pic)
dispatcher.add_handler(pic_handler)

def eg(update, context):
    context.bot.send_message(chat_id=update.message.chat_id, text="""Example statement 1:
    Hi Valte, do you want some investment? 1 3
Mapped emotion will be emotion 1 with intensity 3: ecstasy.

Example statement 2:
    Samson 3 3
Mapped emotion will be emotion 3 with intensity 3: admiration.""")

eg_handler = CommandHandler('eg', eg)
dispatcher.add_handler(eg_handler)


# responsetext1 = """Thank you for your input. Among the 8 emotions below, which response would you expect? Please key in a number. Alternatively, you may use /list command to find out the full list of emotions.
# 1. Joy
# 2. Sadness
# 3. Trust
# 4. Disgust
# 5. Fear
# 6. Anger
# 7. Surprise
# 8. Anticipation
# """

# global response1
# response1 = 1

# responsetext3 = """Thank you for your input!"""

# def trigger(update, context):
#     global count
#     global response1
#     if (count%3==1):
#         context.bot.send_message(chat_id=update.message.chat_id, text=responsetext1)
#         sheet.update_cell((count//3)+1, 1, update.message.text)
#     elif (count%3==2):
#         response1 = int(update.message.text)

#         responsetext2 = ("""From a scale of 1-3, 3 being the most severe, how would you rate the intensity of the emotion?
# 1. %s
# 2. %s
# 3. %s
# """ % (emotions[response1][1],emotions[response1][2],emotions[response1][3]))

#         context.bot.send_message(chat_id=update.message.chat_id, text=responsetext2)
#         sheet.update_cell((count//3)+1, 2, update.message.text)
#     else:
#         response2 = int(update.message.text)
#         context.bot.send_message(chat_id=update.message.chat_id, text=responsetext3)
#         sheet.update_cell((count//3), 3, update.message.text)
#         sheet.update_cell((count//3), 4, emotions[response1][response2])
#     count+=1

def trigger(update,context):
    global rowCount
    context.bot.send_message(chat_id=update.message.chat_id,text="Thank you for your input!")
    inputs = update.message.text
    sheet.update_cell(rowCount,1,inputs[0:-4])
    sheet.update_cell(rowCount,2,inputs[-3])
    sheet.update_cell(rowCount,3,inputs[-1])
    sheet.update_cell(rowCount,4,emotions[int(inputs[-3])][int(inputs[-1])])
    rowCount += 1

trigger_handler = MessageHandler(Filters.all,trigger)
dispatcher.add_handler(trigger_handler)

# use creds to create a client to interact with the Google Drive API
scope = ['https://spreadsheets.google.com/feeds',
         'https://www.googleapis.com/auth/drive']
creds = ServiceAccountCredentials.from_json_keyfile_name('client_secret.json', scope)
client = gspread.authorize(creds)

# Find a workbook by name and open the first sheet
# Make sure you use the right name here.
sheet = client.open("Telebot Data").sheet1


updater.start_polling()