Update app.py

app.py

@@ -3,54 +3,51 @@ import pandas as pd
 import numpy as np
 from prophet import Prophet
 from mysql import connector
+import json
+from prophet.serialize import model_from_json
 
+province_dict = {
+    "Bangkok":"กรุงเทพฯ",
+    'Nakohn Pathom':'นครปฐม',
+    'Pathum Thani':'ปทุมธานี',
+    'Nakohn Nayok':'นครนายก',
+    'Nonthaburi':'นนทบุรี',
+    'Samut Songkhram':'สมุทรสงคราม'
+}
 
 
 
-def findProvince(Province):
-    if Province == 'Bangkok':
-        name = list(Station[Station[6]=='กรุงเทพมหานคร'][0])[0]
-    return name
+def weather_forcast(year,month,date):
+    _date = pd.to_datetime(f'{year}-{month}-{date}')
+        
+    _df = pd.DataFrame({'ds':[_date]})
+
+    _prediction = _model.predict(_df)
+    _prediction = _prediction['yhat']
 
-def get_weather_forecast(Province,Activity,Purpose,Year,Month,Date):
-    name = findProvince(Province)
-    p_df = df.rename(columns={'Date_time':'ds',str(name):'y'})
-    rain_df =  p_df[p_df['mode']=='Rain'][['ds','y']]
-    rain_model = Prophet(interval_width=0.95)
-    rain_model.fit(rain_df)
-    date_time = pd.Series(pd.to_datetime(f'{Year}-{Month}-{Date}',format = '%Y-%m-%d'),name='ds')
-    date_time = pd.DataFrame(date_time)
-    forecast = rain_model.predict(date_time)
+    _israin = False if(_prediction<0.5) else True
+    return _israin
     
+def get_advice(province,activity,purpose,year,month,date):
+    _province = province_dict[province]
+    with open('prophet_model.json', 'r') as fin:
+        _model = model_from_json(json.load(fin)
+    _purpose = purpose.lower()
+    _day = pd.to_datetime(f'{year}-{month}-{date}').day_name()
+    _activity = 'indoor' if(weather_forcast(year,month,date)) else activity.lower()
+
     
-    if float(forecast['yhat'][0]) >= 6:
-        WeatherCon = 'SUNNY'
-        txt = 'น้อย'
-    else:
-        WeatherCon = 'RAIN'
-        txt = 'สูง'
-    if Activity == 'Indoor':
-        IsIndoor = 'Y'
-    else:
-        IsIndoor = 'N'
-        
-    place = place_df[place_df['Purpose']==Purpose]
-    place = place[place['IsIndoor']==IsIndoor]
-    place = place[place['Province']==Province]
-    if WeatherCon != 'SUNNY':
-        place = place[place['WeatherCon']==WeatherCon]
-    place = place['LocationName']
-    #random = random.randrange(len(place))
-    place = list(place)[0]
-    return f'มีโอกาสฝนตก{txt} สถานที่ที่แนะนำคือ: {place}'
     
+    return f'มีโอกาสฝนตก{txt} สถานที่ที่แนะนำคือ: {place}'
+
+
 
 iface = gr.Interface(
     fn = get_weather_forecast,
     inputs = [
         
         gr.Dropdown(
-            ["Bangkok"], label="Province", info="Will add more later!"
+            ["Bangkok",'Nakohn Pathom','Pathum Thani','Nakohn Nayok','Nonthaburi','Samut Songkhram'], label="Province", info="Will add more later!"
         ),
 
         gr.Dropdown(
@@ -58,11 +55,11 @@ iface = gr.Interface(
         ),
 
         gr.Dropdown(
-            ["Entertainment","Culture",'Shop','Science'], label="Purpose"
+            ["Shopping","Relax",'Education','Culture','Nature'], label="Purpose"
         ),
 
         gr.Dropdown(
-            [2023], label="Year", info="Will add more later!"
+            [2024], label="Year", info="Will add more later!"
         ),
 
         gr.Dropdown(