RossmannProject/rossmann-bot-local.py at main · egoliveira1/RossmannProject · 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
import pandas as pd
import json
import requests

from flask import Flask, request, Response

# constants
TOKEN = '1752664004:AAGDlEf6zQrDcrAN09GRjfdfurkT_tQilPY'

# Info about the Bot
#https://api.telegram.org/bot1752664004:AAGDlEf6zQrDcrAN09GRjfdfurkT_tQilPY/getMe

# get updates
#https://api.telegram.org/bot1752664004:AAGDlEf6zQrDcrAN09GRjfdfurkT_tQilPY/getUpdates

# Webhook
#https://api.telegram.org/bot1752664004:AAGDlEf6zQrDcrAN09GRjfdfurkT_tQilPY/setWebhook?url=https://79de032cf774d0.localhost.run

# send messages
#https://api.telegram.org/bot1752664004:AAGDlEf6zQrDcrAN09GRjfdfurkT_tQilPY/sendMessage?chat_id=837459307&text=Hi,man!


def send_message(chat_id, text):
	url = 'https://api.telegram.org/bot{}/'.format(TOKEN)
	url = url + 'sendMessage?chat_id={}'.format( chat_id )

	r = requests.post( url, json = {'text' : text } )
	print( 'Status Code {}'.format( r.status_code ) )

	return None

def load_dataset(store_id):
	# loading datasets
	df10 = pd.read_csv('/home/eron/repos/RossmannProject/data/test.csv')
	df_store_raw = pd.read_csv('/home/eron/repos/RossmannProject/data/store.csv')

	# Merge test + store
	df_test = pd.merge (df10, df_store_raw, how = 'left', on = 'Store')

	# Choose store for prediction
	df_test = df_test[df_test['Store'] == store_id ]

	if not df_test.empty:
		# Remove closed days
		df_test = df_test[df_test['Open'] != 0]
		df_test = df_test[~df_test['Open'].isnull()]
		df_test = df_test.drop('Id', axis = 1)

		# Convert DataFrame in Json
		data = json.dumps(df_test.to_dict( orient = 'records'))
	else:
		data = 'error'

	return data

def predict( data ):
	# API Call
	url = 'https://rossmann-model-sales.herokuapp.com/rossmann/predict'
	header = {'Content-type' : 'application/json'}
	data = data

	r = requests.post(url, data = data, headers = header)
	print( 'Status Code {}'.format( r.status_code ) )

	d1 = pd.DataFrame (r.json(), columns = r.json()[0].keys())

	return d1

def parse_message( message ):
	chat_id = message['message']['chat']['id']
	store_id = message['message']['text']

	store_id = store_id.replace( '/', '')

	try:
		store_id = int( store_id )

	except ValueError:

		store_id = 'error'

	return chat_id, store_id

# API initialize
app = Flask( __name__ )

@app.route( '/', methods=['GET', 'POST'] )

def index():
	if request.method == 'POST':
		message = request.get_json()

		chat_id, store_id = parse_message ( message )

		if store_id != 'error':
			# loading data
			data = load_dataset(store_id)

			if data != 'error':
				# prediction
				d1 = predict(data)

				# calculation
				d2 = d1[['store', 'prediction']].groupby('store').sum().reset_index()

				# send message
				msg = 'Store number {} will sell US$ {:,.2f} in the next 6 weeks'.format (
						d2['store'].values[0],
						d2['prediction'].values[0] )

				send_message ( chat_id, msg )
				return Response ( 'Ok', status = 200 )

			else:
				send_message ( chat_id, 'Store not available')
				return Response ( 'Ok', status = 200 )

		else:
			send_message ( chat_id, 'Store ID is Wrong')
			return Response ( 'Ok', status = 200 )

	else:
		return '<h1> Rossmann Telegram BOT </h1>'

if __name__ == '__main__':
	app.run( host= '0.0.0.0', port = 5000, debug = True)