Update app.py

app.py

@@ -8,6 +8,22 @@ import json
 import random
 from prophet.serialize import model_from_json
 
+def get_data():
+    _conn = connector.connect(
+        host='110.238.111.32',
+        user = 'outsider',
+        password='Hack2024',
+        database = 'TheSimp'
+    )
+    _cursor = _conn.cursor()
+    _query = "SELECT name , province ,a_type ,genre ,close ,hour ,link FROM places"
+    _cursor.execute(_query)
+    _data = _cursor.fetchall()
+    _cursor.close()
+    _conn.close()
+    _df = pd.DataFrame(_data,columns=['name','province','a_type','genre','close','hour','link'])
+    return _df
+
 province_mapping = {
     'Bangkok': 'กรุงเทพฯ',
     'Nakohn Pathom': 'นครปฐม',
@@ -33,14 +49,14 @@ def get_advice(province, activity, purpose, year, month, date):
     
     province = province_mapping[province]
     
-    places = pd.read_csv('Places.csv')
+    places = get_data()
     places.replace({'indoor ': 'indoor', 'outdoor ': 'outdoor'}, inplace=True)
-    places = places[(places['จังหวัด'] == province) & (places['indoor/outdoor'] == activity) & (places['หมวดหมู่'] == purpose.lower())]
+    places = places[(places['province'] == province) & (places['a_type'] == activity) & (places['genre'] == purpose.lower())]
     
     random_idx = np.random.randint(0, len(places))
-    place_name = places.iloc[random_idx]['ที่เที่ยว']
-    close_day = places.iloc[random_idx]['ปิดวัน']
-    open_hour = places.iloc[random_idx]['เวลา']
+    place_name = places.iloc[random_idx]['name']
+    close_day = places.iloc[random_idx]['close']
+    open_hour = places.iloc[random_idx]['hour']
     
     advice = f"It might rain on {year}-{month}-{date}, so we suggest you to go indoor place such as {place_name}." if is_rain else f"It might not rain on {year}-{month}-{date}, so we suggest you to go to place such as {place_name}."