Update main.py

main.py

@@ -169,6 +169,32 @@ def compute_accelerations(telemetry):
 
     return np.round(lon_acc,2), np.round(lat_acc,2)
 
+@st.cache_data
+@app.get("/wdc", response_model=None)
+def driver_standings() -> any:
+    YEAR = datetime.datetime.now().year
+    df = pd.DataFrame(
+        pd.read_html(f"https://www.formula1.com/en/results.html/{YEAR}/drivers.html")[0]
+    )
+    df = df[["Driver", "PTS", "Car"]]
+    # reverse the order
+    df = df.sort_values(by="PTS", ascending=False)
+
+    # in Driver column only keep the last 3 characters
+    df["Driver"] = df["Driver"].str[:-3]
+    
+    # add colors to the dataframe
+    car_colors = available_data.team_colors(YEAR)
+    df["fill"] = df["Car"].map(car_colors)
+
+    
+    # remove rows where points is 0
+    df = df[df["PTS"] != 0]
+    df.reset_index(inplace=True, drop=True)
+    df.rename(columns={"PTS": "Points"}, inplace=True)
+
+    return  {"WDC":df.to_dict("records")}
+
 @st.cache_data
 @app.get("/", response_model=None)
 async def root():