/
E3_demoBot_Twitter_logreg.py
134 lines (98 loc) · 4.05 KB
/
E3_demoBot_Twitter_logreg.py
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
131
132
# -*- coding: utf-8 -*-
"""
Created on Tue Apr 20 19:06:13 2021
@author: Erik
"""
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
#from helper import gsheet_helper
import wolframalpha
#import tkinter as Tk
#from tkinter import *
import pandas as pd
# Global for twitter
import validators
from tweet_processing_logreg import tweet_prediction
#%%
def start(update, context): # Saving user information in excel
user_name = update.message.from_user['first_name']
user_lastname = update.message.from_user['last_name']
user_username = update.message.from_user['username']
user_id = update.message.from_user['id']
user_info = [user_name, user_lastname, user_username, user_id]
Info = pd.DataFrame(user_info , index = ['User Name','User Lastname','Username','User ID'])
Info = Info.transpose()
data = pd.read_excel('Info.xlsx' , engine='openpyxl')
frames = [data, Info]
data = pd.concat(frames)
data.to_excel('Info.xlsx', index = False, header = True)
print(data)
update.message.reply_text('Hello {}, I am E3 demo bot' .format(update.message.from_user['first_name']))
def user_info(update, context): # For reading user info and displaying
data = pd.read_excel('Info.xlsx' , engine='openpyxl')
data_show = data.tail(5)
print(data_show)
update.message.reply_text(str(data_show))
def query_info(update, context): # For reading queries and displaying
data = pd.read_excel('Query.xlsx' , engine='openpyxl')
data_show = data.tail(5)
print(data_show)
update.message.reply_text(str(data_show))
def query(update, res):
if res['@success'] == False:
none = 'Question cannot be resolved, it might be mistyped.'
#print('Question cannot be resolved')
return none
else:
# pod is the whole answers
pod = res['pod']
# pod[0] is the question
pod0 = res['pod'][0]
# pod[1] may contains the answer
pod1 = res['pod'][1]
result = pod1['subpod']['plaintext']
print(result)
update.message.reply_text(result)
# find the wiki answer
for sub in pod:
subpod = sub['@title']
if subpod == 'Wikipedia summary':
result_wiki = sub['subpod']['plaintext']
result_link = sub['subpod']['infos']['info']['link']['@url']
print(result_wiki)
update.message.reply_text(result_wiki)
def save_query(update, user_query): # Saving query in excel
user_id = update.message.from_user['id']
info_query = [user_id, user_query]
Query = pd.DataFrame(info_query , index = ['User ID','Query'])
Query = Query.transpose()
data = pd.read_excel('Query.xlsx' , engine='openpyxl')
frames = [data, Query]
data = pd.concat(frames)
data.to_excel('Query.xlsx', index = False, header = True)
def reply(update, context):
app_id = ' '
client = wolframalpha.Client(app_id)
user_query = update.message.text
save_query(update, user_query)
valid = validators.url(user_query) # Checking whether is a valid link or not
if valid==True:
tweet_id = user_query.split('/')[-1]
print("Url is valid from twitter with ID:", tweet_id)
pred = tweet_prediction(update,tweet_id)
update.message.reply_text('We estimate the current tweet is ' + str(round(pred, 2)) + '% a real emergency')
else: # In case of normal text query
res = client.query(user_query)
query(update,res)
def main():
updater = Updater(" ", use_context=True)
updater.dispatcher.add_handler(CommandHandler('start' , start))
updater.dispatcher.add_handler(CommandHandler('user_info' , user_info))
updater.dispatcher.add_handler(CommandHandler('query_info' , query_info))
updater.dispatcher.add_handler(MessageHandler(Filters.text, reply))
# Start
updater.start_polling()
print("I am alive now")
# Stay waiting
updater.idle()
if __name__ == "__main__":
main()