ChickenTinder/python_backend.py at master · Codeology/ChickenTinder · 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
import pandas as pd
import numpy as np
from flask import Flask, request
from flask_cors import CORS
import requests
import json

restaurants = pd.read_csv('Dummy Restaurant Dataset - Sheet1.csv')
zones = restaurants.ZONE.unique()
#users = pd.DataFrame(columns=['username', 'password', 'location', 'startUp'])

url = "http://localhost:3000"


heuristic_values = []
maxReviewCount = restaurants['REVIEW_COUNT'].max()
def get_heuristic(rating, review_count): # Calculates heuristic value from rating and numReviews value in a dataframe
    rating_value = (rating / 5) * 50
    count_value = (review_count / maxReviewCount) * 50
    return rating_value + count_value
for i in restaurants.index:
    h_val = get_heuristic(restaurants['YELP_RATING'][i], restaurants['REVIEW_COUNT'][i])
    heuristic_values.append(h_val)

restaurants['HEURISTIC'] = heuristic_values
southSide = restaurants.loc[restaurants['ZONE']=='South Side']
southSide = southSide.sort_values(by=['HEURISTIC'], ascending=False)
northSide = restaurants.loc[restaurants['ZONE']=='North Side']
northSide = northSide.sort_values(by=['HEURISTIC'], ascending=False)
elmwood = restaurants.loc[restaurants['ZONE']=='Elmwood']
elmwood = elmwood.sort_values(by=['HEURISTIC'], ascending=False)
downtown = restaurants.loc[restaurants['ZONE']=='Downtown']
downtown = downtown.sort_values(by=['HEURISTIC'], ascending=False)

app = Flask(__name__)
cors = CORS(app)
@app.route("/", methods = ['GET', 'POST'])                                                  # this info will contain a user name, password, and location preference
def preference():
    if request.method == 'POST':
        if 'name' in request.json:
            return restaurants.iloc[0].to_json(orient = 'index')
        else:
            if request.json['pref'] == 'noPref':
                return restaurants.iloc[request.json['count']].to_json(orient = 'index')
            else:
                preferred_location = restaurants.loc[restaurants['ZONE']== request.json['pref']]
                preferred_location = preferred_location.sort_values(by='HEURISTIC', ascending=False)
                return preferred_location.iloc[request.json['count']].to_json(orient = 'index')

# #listen for which area user prefers
# preferred_zone = restaurants

# app = Flask(__name__)
# @app.route("/", methods = ['GET', 'POST'])                      # this info will contain a user name, password, and location preference
# def preference():
#     if request.method == 'POST' and request.json['userName']:
#         if request.json['location'] == 'south side':
#             preferred_zone = southSide
#         elif request.json['location'] =='north side':
#             preferred_zone = northSide
#         elif request.json['location'] == 'elmwood':
#             preferred_zone = elmwood
#         elif request.json['location'] == 'downtown':
#             preferred_zone = downtown
#         requests.post(url, restaurant_info_dict[0])


# num_users = 1

# # preferred_zone = downtown #FOR EXAMPLE

# restaurant_info_dict = preferred_zone.to_dict(orient='records')

# i = 1 # which restaurant we are on

# num_preferred_restaurants = len(restaurant_info_dict)

# # send back restaurant data

# restaurant_to_send = restaurant_info_dict
# choices = []

# @app.route("/", methods = ['GET', 'POST'])
# def swipe():
#     if request.method == 'POST':
#         restaurant_choice = request.json['choice']
#         choices.append(restaurant_choice)
#         if i < num_preferred_restaurants:
#             if len(choices) == num_users:
#                 restaurant_info_dict[i]['swipes'] = sum(choices)
#                 if sum(choices) == num_users:
#                     restaurant_info_dict[i]['match'] = 1
#                     requests.post(url, data=restaurant_info_dict[i])
#                 else:
#                     if (i+1) == num_preferred_restaurants:
#                         restaurant_info_dict[i]['match'] = 2
#                         requests.post(url, data=restaurant_info_dict[i])
#                     else:
#                         restaurant_info_dict[i+1]['match'] = 0
#                         requests.post(url, data=restaurant_info_dict[i + 1])


if __name__ == "__main__":
    app.run(port="5000")