Add vertical/horizontal line to a px.line figure about vizro HOT 2 CLOSED

import pandas as pd

import vizro.models as vm
import vizro.plotly.express as px
from vizro import Vizro
from vizro.models.types import capture


def aggregate_data():
    df_gapminder = px.data.gapminder()
    df_gapminder_agg = px.data.gapminder()
    df_gapminder_agg["lifeExp"] = df_gapminder_agg.groupby(by=["continent", "year"])["lifeExp"].transform("mean")
    df_gapminder_agg["gdpPercap"] = df_gapminder_agg.groupby(by=["continent", "year"])["gdpPercap"].transform("mean")
    df_gapminder_agg["pop"] = df_gapminder_agg.groupby(by=["continent", "year"])["pop"].transform("sum")
    df_gapminder["data"] = "Country"
    df_gapminder_agg["data"] = "Continent"
    df_gapminder = pd.concat([df_gapminder_agg, df_gapminder], ignore_index=True)
    return df_gapminder

df_gapminder = aggregate_data()


@capture("graph")  # Note change here
def my_figure(data_frame):
    fig = px.line(
        data_frame=data_frame,
        x="year",
        y="gdpPercap",
        color="data",
        labels={"year": "Year", "data": "Data", "gdpPercap": "GDP per capita"},
        color_discrete_map={"Country": "#afe7f9", "Continent": "#003875"},
        markers=True,
    )
    fig.add_vline(x=1970, line_color="green", line_dash="dash")
    return fig


page = vm.Page(
    title="Country Analysis",
    layout=vm.Layout(grid=[[0, 1]]),
    components=[
        vm.Graph(
            id="bar_country",
            figure=px.bar(
                df_gapminder,
                x="year",
                y="pop",
                color="data",
                barmode="group",
                labels={"year": "Year", "data": "Data", "pop": "Population"},
                color_discrete_map={"Country": "#afe7f9", "Continent": "#003875"},
            ),
        ),
        vm.Graph(id="line_country", 
                         figure=my_figure(data_frame=df_gapminder)   # Add your custom chart as usually
                         ),
    ],
    controls=[
        vm.Filter(column="country", selector=vm.Dropdown(value="India", multi=False, title="Select country")),
        vm.Filter(column="year", selector=vm.RangeSlider(title="Select timeframe")),
    ],
)

dashboard = vm.Dashboard(pages=[page])
Vizro().build(dashboard).run()