import fastf1
import pandas as pd
from fastapi import FastAPI
import os





app = FastAPI()


@app.get("/")
async def root():
    return {"message": "Hello World"}

@app.get("/{year}/{race}/{session}", response_model=None)
async def get_data(year: int, race: int, session: str) -> any:    
    
    f1session = fastf1.get_session(year, race, session)
    f1session.load(telemetry=False, weather=False, messages=False)
    # Load all laps with telemetry
    laps = f1session.laps
    laps['Sector1Time'] = laps['Sector1Time'].dt.total_seconds()
    laps['Sector2Time'] = laps['Sector2Time'].dt.total_seconds()
    laps['Sector3Time'] = laps['Sector3Time'].dt.total_seconds()
    laps['LapTime_in_seconds'] = laps['LapTime'].dt.total_seconds()
    laps['laptime_sum_sectortimes'] = laps.Sector1Time + laps.Sector2Time + laps.Sector3Time 
    laps.fillna("")

    
    
    return {"laps": laps.to_dict(orient='records')}