hsaest commited on
Commit
6159f52
1 Parent(s): b948647

Upload folder using huggingface_hub

Browse files
annotation/src/utils.py CHANGED
@@ -146,7 +146,7 @@ def get_city_list(days, deparure_city, destination):
146
  if days == 3:
147
  city_list.append(destination)
148
  else:
149
- city_set = open('../database/background/citySet_with_states.txt').read().split('\n')
150
  state_city_map = {}
151
  for unit in city_set:
152
  city, state = unit.split('\t')
 
146
  if days == 3:
147
  city_list.append(destination)
148
  else:
149
+ city_set = open('/home/user/app/database/background/citySet_with_states.txt').read().split('\n')
150
  state_city_map = {}
151
  for unit in city_set:
152
  city, state = unit.split('\t')
commonsenseConstraint.py CHANGED
@@ -21,7 +21,7 @@ restaurants = Restaurants()
21
  googleDistanceMatrix = GoogleDistanceMatrix()
22
  attractions = Attractions()
23
 
24
- city_state_set = open('../database/background/citySet_with_states.txt','r').read().split('\n')
25
  city_state_map = {x:y for x,y in [unit.split('\t') for unit in city_state_set]}
26
 
27
 
 
21
  googleDistanceMatrix = GoogleDistanceMatrix()
22
  attractions = Attractions()
23
 
24
+ city_state_set = open('/home/user/app/database/background/citySet_with_states.txt','r').read().split('\n')
25
  city_state_map = {x:y for x,y in [unit.split('\t') for unit in city_state_set]}
26
 
27
 
tools/attractions/test.py CHANGED
@@ -4,14 +4,14 @@ import sys
4
  import os
5
  sys.path.append(os.path.abspath(os.path.join(os.getcwd(), "..")))
6
  os.chdir(os.path.dirname(os.path.abspath(__file__)))
7
- unique_cities = Attractions(path="../../database/attractions/attractions.csv").data['City'].unique()
8
- df = Attractions(path="../../database/attractions/attractions.csv").data
9
  print(len(df))
10
- citySet = open('../../database/background/citySet.txt','r').read().split('\n')
11
  cnt = 0
12
  for city in unique_cities:
13
  if city not in citySet:
14
  df = df[df['City'] != city]
15
  print(len(df))
16
 
17
- df.to_csv('../../database/attractions/attractions2.csv', index=False)
 
4
  import os
5
  sys.path.append(os.path.abspath(os.path.join(os.getcwd(), "..")))
6
  os.chdir(os.path.dirname(os.path.abspath(__file__)))
7
+ unique_cities = Attractions(path="..//home/user/app/database/attractions/attractions.csv").data['City'].unique()
8
+ df = Attractions(path="..//home/user/app/database/attractions/attractions.csv").data
9
  print(len(df))
10
+ citySet = open('..//home/user/app/database/background/citySet.txt','r').read().split('\n')
11
  cnt = 0
12
  for city in unique_cities:
13
  if city not in citySet:
14
  df = df[df['City'] != city]
15
  print(len(df))
16
 
17
+ df.to_csv('..//home/user/app/database/attractions/attractions2.csv', index=False)
tools/flights/apis.py CHANGED
@@ -5,7 +5,7 @@ from annotation.src.utils import extract_before_parenthesis
5
 
6
  class Flights:
7
 
8
- def __init__(self, path="../database/flights/clean_Flights_2022.csv"):
9
  self.path = path
10
  self.data = None
11
 
 
5
 
6
  class Flights:
7
 
8
+ def __init__(self, path="/home/user/app/database/flights/clean_Flights_2022.csv"):
9
  self.path = path
10
  self.data = None
11
 
tools/flights/test.py CHANGED
@@ -18,12 +18,12 @@ import os
18
  # results = results[results["DestCityName"] == destination]
19
  sys.path.append(os.path.abspath(os.path.join(os.getcwd(), "..")))
20
  os.chdir(os.path.dirname(os.path.abspath(__file__)))
21
- unique_cities = list(Flights(path="../../database/flights/clean_Flights_2022.csv").data['OriginCityName'].unique()) + list(Flights(path="../../database/flights/clean_Flights_2022.csv").data['DestCityName'].unique())
22
- df = Flights(path="../../database/flights/clean_Flights_2022.csv").data
23
 
24
  print(len(df))
25
 
26
- citySet = open('../../database/background/citySet.txt','r').read().split('\n')
27
  cnt = 0
28
 
29
  for city in unique_cities:
@@ -32,4 +32,4 @@ for city in unique_cities:
32
 
33
  print(len(df))
34
 
35
- df.to_csv('../../database/flights/clean_Flights_2022.csv', index=False)
 
18
  # results = results[results["DestCityName"] == destination]
19
  sys.path.append(os.path.abspath(os.path.join(os.getcwd(), "..")))
20
  os.chdir(os.path.dirname(os.path.abspath(__file__)))
21
+ unique_cities = list(Flights(path="..//home/user/app/database/flights/clean_Flights_2022.csv").data['OriginCityName'].unique()) + list(Flights(path="..//home/user/app/database/flights/clean_Flights_2022.csv").data['DestCityName'].unique())
22
+ df = Flights(path="..//home/user/app/database/flights/clean_Flights_2022.csv").data
23
 
24
  print(len(df))
25
 
26
+ citySet = open('..//home/user/app/database/background/citySet.txt','r').read().split('\n')
27
  cnt = 0
28
 
29
  for city in unique_cities:
 
32
 
33
  print(len(df))
34
 
35
+ df.to_csv('..//home/user/app/database/flights/clean_Flights_2022.csv', index=False)
tools/googleDistanceMatrix/apis.py CHANGED
@@ -14,7 +14,7 @@ os.environ["https_proxy"] = "http://127.0.0.1:7890"
14
  class GoogleDistanceMatrix:
15
  def __init__(self, subscription_key: str="AIzaSyClbhCKqk2QS97jadE0RQ3qYCAteY7sK4I") -> None:
16
  self.gplaces_api_key: str = subscription_key
17
- self.data = pd.read_csv('../database/googleDistanceMatrix/distance.csv')
18
  print("GoogleDistanceMatrix loaded.")
19
 
20
  def run(self, origin, destination, mode='driving'):
 
14
  class GoogleDistanceMatrix:
15
  def __init__(self, subscription_key: str="AIzaSyClbhCKqk2QS97jadE0RQ3qYCAteY7sK4I") -> None:
16
  self.gplaces_api_key: str = subscription_key
17
+ self.data = pd.read_csv('/home/user/app/database/googleDistanceMatrix/distance.csv')
18
  print("GoogleDistanceMatrix loaded.")
19
 
20
  def run(self, origin, destination, mode='driving'):
tools/restaurants/apis.py CHANGED
@@ -4,7 +4,7 @@ from typing import Optional
4
  from annotation.src.utils import extract_before_parenthesis
5
 
6
  class Restaurants:
7
- def __init__(self, path="../database/restaurants/clean_restaurant_2022.csv"):
8
  self.path = path
9
  self.data = pd.read_csv(self.path).dropna()[['Name','Average Cost','Cuisines','Aggregate Rating','City']]
10
  print("Restaurants loaded.")
 
4
  from annotation.src.utils import extract_before_parenthesis
5
 
6
  class Restaurants:
7
+ def __init__(self, path="/home/user/app/database/restaurants/clean_restaurant_2022.csv"):
8
  self.path = path
9
  self.data = pd.read_csv(self.path).dropna()[['Name','Average Cost','Cuisines','Aggregate Rating','City']]
10
  print("Restaurants loaded.")
utils/budget_estimation.py CHANGED
@@ -56,7 +56,7 @@ def budget_calc(org, dest, days, date:list , people_number=None, local_constrain
56
 
57
 
58
  elif grain == "state":
59
- city_set = open('../database/background/citySet_with_states.txt').read().strip().split('\n')
60
 
61
  all_hotel_data = []
62
  all_restaurant_data = []
 
56
 
57
 
58
  elif grain == "state":
59
+ city_set = open('/home/user/app/database/background/citySet_with_states.txt').read().strip().split('\n')
60
 
61
  all_hotel_data = []
62
  all_restaurant_data = []
utils/query_element_selection.py CHANGED
@@ -11,7 +11,7 @@ import numpy as np
11
 
12
  google_distance = GoogleDistanceMatrix()
13
 
14
- city_set = open('../database/background/citySet_with_states.txt').read().strip().split('\n')
15
 
16
  state_city_map = {}
17
 
@@ -49,7 +49,7 @@ def select_consecutive_dates(num_days, start_date=datetime(2022, 3, 1), end_date
49
 
50
  def get_org_dest(days:int):
51
  if days == 3:
52
- city_set = open('../database/background/citySet_with_states.txt').read().strip().split('\n')
53
 
54
  org = random.choice(city_set)
55
 
@@ -63,7 +63,7 @@ def get_org_dest(days:int):
63
 
64
  elif days in [5,7]:
65
 
66
- state_set = open('../database/background/citySet_with_states.txt').read().strip().split('\n')
67
  org = random.choice(state_set)
68
 
69
  while True:
 
11
 
12
  google_distance = GoogleDistanceMatrix()
13
 
14
+ city_set = open('/home/user/app/database/background/citySet_with_states.txt').read().strip().split('\n')
15
 
16
  state_city_map = {}
17
 
 
49
 
50
  def get_org_dest(days:int):
51
  if days == 3:
52
+ city_set = open('/home/user/app/database/background/citySet_with_states.txt').read().strip().split('\n')
53
 
54
  org = random.choice(city_set)
55
 
 
63
 
64
  elif days in [5,7]:
65
 
66
+ state_set = open('/home/user/app/database/background/citySet_with_states.txt').read().strip().split('\n')
67
  org = random.choice(state_set)
68
 
69
  while True: