import streamlit as st
import altair as alt
import pandas as pd
import io

st.set_page_config(initial_sidebar_state="collapsed",
page_title="My App",
layout="wide")

df = pd.read_excel(r"C:\Users\toi.watanabe\Desktop\streamlit\data.xlsx")
df["取引日時"]=pd.to_datetime(df["取引日時"])

st.sidebar.header("フィルター")
category = st.sidebar.multiselect(
"カテゴリー",
options=df["カテゴリー"].unique(),
default=df["カテゴリー"].unique(),
label_visibility="visible"
)
start_date, end_date = st.sidebar.date_input(
"期間",
[df["取引日時"].min(), df["取引日時"].max()]
)
filtered_df = df[
(df["カテゴリー"].isin(category)) &
(df["取引日時"].between(pd.to_datetime(start_date), pd.to_datetime(end_date)))
]
st.dataframe(filtered_df, use_container_width=False)
#🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢
def create_excel(dataframe):
buffer = io.BytesIO()
with pd.ExcelWriter(buffer, engine="xlsxwriter") as writer:
dataframe.to_excel(writer, index=False, sheet_name="data")
return buffer.getvalue()

st.download_button(
label="Excelダウンロード",
data=create_excel(filtered_df), # ← 押されたときに初めて実行される
file_name="data.xlsx",
mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
)
#🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢🟢

chart_df = df[(df["取引日時"].between(pd.to_datetime(start_date), pd.to_datetime(end_date)))]

状態をセッションに保持

if "toggle_state" not in st.session_state:
st.session_state.toggle_state = False

押すたびに状態を反転させる関数

def toggle():
st.session_state.toggle_state = not st.session_state.toggle_state

ボタン

st.button("グラフ切替", on_click=toggle)

状態に応じて表示

if st.session_state.toggle_state:
st.write("折れ線グラフ")
# 線は単色
line_layer = (
alt.Chart(chart_df)
.mark_line(color="white")
.encode(
x="取引日時:T",
y="取引後残高:Q"
)
)
st.altair_chart(line_layer, use_container_width=True)

else:
st.write("ローソク足チャート")
# ポイントはカテゴリー別に色分け
line_layer = (
alt.Chart(chart_df)
.mark_line(color="white")
.encode(
x="取引日時:T",
y="取引後残高:Q"
)
)

point_layer = (
    alt.Chart(chart_df)
    .mark_point(filled=True, size=100)
    .encode(
        x=alt.X("取引日時:T",axis=alt.Axis(format="%m月%d日")),
        y="取引後残高:Q",
        color=alt.Color("カテゴリー:N",legend=None),
        tooltip=[
            alt.Tooltip("取引日時:T", format='%Y-%m-%d %H:%M:%S', title=None),
            "取引金額",
            "取引後残高",
            "カテゴリー"
        ]
    )
)

#縦線レイヤー
highlight_date = "2026-01-30 00:00:00"
vline = alt.Chart(pd.DataFrame({'検知日': [highlight_date]})).mark_rule(
    color='red',      # 線の色
    strokeDash=[5,5], # 破線
    size=4            # 線の太さ
).encode(
   x=alt.X("検知日:T",axis=alt.Axis(format="%m月%d日")),
    tooltip=[
        alt.Tooltip("検知日:T", format='%Y-%m-%d %H:%M:%S'),
    ]
)
chart = (line_layer + point_layer + vline).add_selection(alt.selection_interval(bind='scales',encodings=['x']))
st.altair_chart(chart, use_container_width=True)

def my_function():
st.write("関数が実行されました！")

st.button("実行", on_click=my_function)