Update app.py

app.py

@@ -9,6 +9,7 @@ import dateutil.parser as dp
 import pandas as pd
 from huggingface_hub import hf_hub_url, cached_download
 import time
+from datetime import datetime
 
 def get_row():
     response_tomtom = requests.get(
@@ -29,7 +30,7 @@ def get_row():
     json_response_smhi = json.loads(response_smhi.text) 
     
     # weather data manual https://opendata.smhi.se/apidocs/metanalys/parameters.html#parameter-wsymb
-    referenceTime = dp.parse(json_response_smhi["referenceTime"]).timestamp()
+    # referenceTime = dp.parse(json_response_smhi["referenceTime"]).timestamp()
     
     t             = json_response_smhi["timeSeries"][0]["parameters"][0]["values"][0] # Temperature
     ws            = json_response_smhi["timeSeries"][0]["parameters"][4]["values"][0] # Wind Speed
@@ -38,7 +39,7 @@ def get_row():
     vis           = json_response_smhi["timeSeries"][0]["parameters"][9]["values"][0] # Visibility
 
     # Use current time
-    referenceTime = time.time()
+    referenceTime = datetime.fromtimestamp(time.time())
     
     row           ={"referenceTime": referenceTime, 
                         "temperature": t, 
@@ -54,41 +55,58 @@ def get_row():
     return row
 
 model = joblib.load(cached_download(
-    hf_hub_url("tilos/Traffic_Prediction", "traffic_model.pkl")
+    hf_hub_url("Chenzhou/Traffic_Prediction", "traffic_model_adam.pkl")
 ))
 
 def infer(input_dataframe):
-  return pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
+    serie = input_dataframe["referenceTime"]
+    ts = dp.parse(serie.iloc[0]).timestamp()
+    input_dataframe["referenceTime"] = ts
+    res = pd.DataFrame(model.predict(input_dataframe)).clip(0, 1).iloc[0, 0]
+    if res > 0.8:
+        status = "Smooth Traffic on E4"
+    elif res > 0.5:
+        status = "Slight congestion on E4"
+    else:
+        status = "Total congestion on E4"
+    return pd.DataFrame({'Freeflow Level':[res], 'Status': [status]})
 
 title = "Stoclholm Highway E4 Real Time Traffic Prediction"
 description = "Stockholm E4 (59°23'44.7"" N 17°59'00.4""E) highway real time traffic prediction"
 
-inputs = [gr.Dataframe(row_count = (1, "fixed"), col_count=(7,"fixed"), 
-                       headers=["referenceTime", "t", "ws", "prec1h", "fesn1h", "vis", "confidence"], 
-                       # datatype=["timestamp", "float", "float", "float", "float", "float"],
-                       label="Input Data", interactive=1)]
+# inputs = [gr.Dataframe(row_count = (1, "fixed"), col_count=(7,"fixed"), 
+#                        headers=["referenceTime", "t", "ws", "prec1h", "fesn1h", "vis", "confidence"], 
+#                        # datatype=["timestamp", "float", "float", "float", "float", "float"],
+#                        label="Input Data", interactive=1)]
 
-outputs = [gr.Dataframe(row_count = (1, "fixed"), col_count=(1, "fixed"), label="Predictions", headers=["Congestion Level"])]
+# outputs = [gr.Dataframe(row_count = (1, "fixed"), col_count=(1, "fixed"), label="Predictions", headers=["Congestion Level"])]
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            gr.Dataframe(row_count = (1, "fixed"), col_count=(7,"fixed"), 
+            inputs = gr.Dataframe(row_count = (1, "fixed"), col_count=(7,"fixed"), 
                        headers=["referenceTime", "t", "ws", "prec1h", "fesn1h", "vis", "confidence"], 
                        # datatype=["timestamp", "float", "float", "float", "float", "float"],
                        label="Input Data", interactive=1)
         with gr.Column():
-            gr.Dataframe(row_count = (1, "fixed"), col_count=(1, "fixed"), label="Predictions", headers=["Congestion Level"])
-            
-    demo.load(get_row, every=10)
+            outputs = gr.Dataframe(row_count = (1, "fixed"), col_count=(2, "fixed"), label="Predictions", headers=["Freeflow Level", "Status"])
     
     with gr.Row():
         btn_sub = gr.Button(value="Submit")
+    with gr.Row():
+        btn_ref = gr.Button(value="Get real-time data")
+
+    
     
     btn_sub.click(infer, inputs = inputs, outputs = outputs)
+    btn_ref.click(get_row, inputs = None, outputs = inputs)
     
     #examples = gr.Examples(fn = infer, examples=[get_row()],inputs=inputs,outputs=outputs  ,cache_examples=True)
-    examples = gr.Examples(examples=[get_row()] ,inputs=inputs  ,cache_examples=False)
+    examples = gr.Examples(fn = infer, examples=[get_row()] ,inputs=inputs, outputs=outputs, cache_examples=False)
+
+    # demo.load(get_row, inputs = None, outputs = [inputs], every=10)
+    demo.load(get_row, inputs = None, outputs = [inputs])
+
         
 
     
@@ -96,4 +114,4 @@ with gr.Blocks() as demo:
 # interface.launch()
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()