Create main.py

main.py

@@ -0,0 +1,84 @@
+import gradio as gr
+import pandas as pd
+import numpy as np
+from prophet import Prophet
+import random
+df = pd.read_csv('df60')
+place_df = pd.read_csv('place.csv')
+Station = pd.DataFrame(np.load('Station.npy',allow_pickle=True))
+def findProvince(Province):
+    if Province == 'Bangkok':
+        name = list(Station[Station[6]=='กรุงเทพมหานคร'][0])[0]
+    return name
+
+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)
+    
+    
+    if float(forecast['yhat'][0]) >= 5:
+        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}'
+    
+
+iface = gr.Interface(
+    fn = get_weather_forecast,
+    inputs = [
+        
+        gr.Dropdown(
+            ["Bangkok"], label="Province", info="Will add more later!"
+        ),
+
+        gr.Dropdown(
+            ["Indoor","Outdoor"], label="Activity"
+        ),
+
+        gr.Dropdown(
+            ["Entertainment","Culture",'Shop','Science'], label="Purpose"
+        ),
+
+        gr.Dropdown(
+            [2023], 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"
+        )
+
+    ],
+    
+    outputs=gr.outputs.Textbox(label="Prediction"),
+    live=True,
+    title="Weather Forecast",
+    description="Get the weather forecast for a city.",
+    theme="default",
+)
+
+iface.launch()