Update app.py

app.py

@@ -1,97 +1,66 @@
 import gradio as gr
 import pandas as pd
 import numpy as np
-from prophet import Prophet
-from mysql import connector
-import json
-import random
 from prophet.serialize import model_from_json
 
-province_dict = {
-    "Bangkok":"กรุงเทพฯ",
-    'Nakohn Pathom':'นครปฐม',
-    'Pathum Thani':'ปทุมธานี',
-    'Nakohn Nayok':'นครนายก',
-    'Nonthaburi':'นนทบุรี',
-    'Samut Songkhram':'สมุทรสงคราม'
+province_mapping = {
+    'Bangkok': 'กรุงเทพฯ',
+    'Nakohn Pathom': 'นครปฐม',
+    'Pathum Thani': 'ปทุมธานี',
+    'Nakohn Nayok': 'นครนายก',
+    'Nonthaburi': 'นนทบุรี',
+    'Samut Songkhram': 'สมุทรสงคราม'
 }
 
 with open('prophet_model.json', 'r') as fin:
-        _model = model_from_json(json.load(fin))
+    prophet_model = model_from_json(json.load(fin))
 
-def weather_forcast(year,month,date):
+def will_rain(year, month, date):
     _date = pd.to_datetime(f'{year}-{month}-{date}')
-        
-    _df = pd.DataFrame({'ds':[_date]})
-
-    _prediction = _model.predict(_df)
+    _df = pd.DataFrame({'ds': [_date]})
+    _prediction = prophet_model.predict(_df)
     _prediction = float(_prediction['yhat'])
+    return _prediction >= 0.5
 
-    _israin = False if(_prediction<0.5) else True
-    return _israin
+def get_advice(province, activity, purpose, year, month, date):
+    is_rain = will_rain(year, month, date)
+    activity = 'indoor' if is_rain else activity.lower()
     
-def get_advice(province,activity,purpose,year,month,date):
-    _province = province_dict[province]
+    province = province_mapping[province]
     
-    _purpose = str(purpose).lower()
-    _day = pd.to_datetime(f'{year}-{month}-{date}').day_name()
-    _activity = 'indoor' if(weather_forcast(year,month,date)) else activity.lower()
-
-    _places = pd.read_csv('Places.csv')
-    _places = _places[_places['จังหวัด']==_province]
-    _places = _places[_places['indoor/outdoor']==_activity]
-    _places = _places[_places['หมวดหมู่']==_purpose]
-    _places = _places[_places['ปิดวัน']!=_day]
-
-    random_idx = random.randrange(0,len(_places))
-    _advice = f'It might rain on {year}-{month}-{date} so we suggest you to go indoor place such as...' if(weather_forcast(year,month,date)) else f'It might not rain on {year}-{month}-{date} so we suggest you to go to place such as...'
-    _place_name = _places['ที่เที่ยว'][random_idx]
-    _place_close = _places['ปิดวัน'][random_idx]
-    _place_hour = _places['เวลา'][random_idx]
-    return _advice,_place_name,_place_close,_place_hour
-
-
+    places = pd.read_csv('Places.csv')
+    places.replace({'indoor ': 'indoor', 'outdoor ': 'outdoor'}, inplace=True)
+    places = places[(places['จังหวัด'] == province) & (places['indoor/outdoor'] == activity) & (places['หมวดหมู่'] == 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]['เวลา']
+    
+    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}."
+    
+    return advice, place_name, close_day, open_hour
 
 iface = gr.Interface(
-    fn = get_advice,
-    inputs = [
-        
-        gr.Dropdown(
-            ["Bangkok",'Nakohn Pathom','Pathum Thani','Nakohn Nayok','Nonthaburi','Samut Songkhram'], label="Province", info="Will add more later!"
-        ),
-
-        gr.Dropdown(
-            ["Indoor","Outdoor"], label="Activity"
-        ),
-
-        gr.Dropdown(
-            ["Shopping","Relax",'Education','Culture','Nature'], label="Purpose"
-        ),
-
-        gr.Dropdown(
-            [2024], label="Year", info="Will add more later!"
-        ),
-
-        gr.Dropdown(
-            [1,2,3,4,5,6,7,8,9,10,11,12], label="Month"
-        ),
-
-        gr.Dropdown(
-            [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], label="Date"
-        )
-
+    fn=get_advice,
+    inputs=[
+        gr.Dropdown(["Bangkok", 'Nakohn Pathom', 'Pathum Thani', 'Nakohn Nayok', 'Nonthaburi', 'Samut Songkhram'], label="Province",),
+        gr.Dropdown(["indoor", "outdoor"], label="Activity"),
+        gr.Dropdown(["shopping", "relax", 'education', 'culture', 'nature'], label="Purpose"),
+        gr.Dropdown([2024], label="Year"),
+        gr.Dropdown([i for i in range(1, 13)], label="Month"),
+        gr.Dropdown([i for i in range(1, 32)], label="Date"),
     ],
-    
     outputs=[
-        gr.components.Textbox(label='Advice :'),
-        gr.components.Textbox(label="Place Name :"),
-        gr.components.Textbox(label="Close day :"),
-        gr.components.Textbox(label="Open hour :")
-        ],
+        gr.components.Textbox(label='Advice'),
+        gr.components.Textbox(label="Place Name"),
+        gr.components.Textbox(label="Close day"),
+        gr.components.Textbox(label="Open hour"),
+    ],
     live=True,
-    title="Right place , Right day",
+    title="Right place, Right day",
     description="Get the weather forecast for a place you would like to go.",
-    theme="default",
+    theme="default"
 )
 
-iface.launch()
+iface.launch()