feat: data

src/streamlit_app.py

@@ -1,40 +1,97 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from pathlib import Path
+import tomli
+import plotly.graph_objects as go
+from typing import List, Dict
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# === 配置路径 ===
+DATA_ROOT = Path(__file__).parent.parent / "data"
+
+# === 页面设置 ===
+st.set_page_config(page_title="Flowillower 🌸", layout="wide")
+st.markdown("<h1 style='text-align: center; color: #7b68ee;'>Flowillower 🌸</h1>", unsafe_allow_html=True)
+st.markdown("<p style='text-align: center;'>Elegant and Transparent Experiment Tracking</p>", unsafe_allow_html=True)
+st.divider()
+
+# === 加载所有 Study ===
+def get_all_studies(data_root: Path) -> List[Path]:
+    return [p for p in data_root.iterdir() if p.is_dir()]
+
+studies = get_all_studies(DATA_ROOT)
+study_names = []
+
+for s in studies:
+    meta_file = s / ".study_meta.toml"
+    if meta_file.exists():
+        try:
+            meta = tomli.loads(meta_file.read_text())
+            name = meta.get("display_name", s.name)
+        except:
+            name = s.name
+    else:
+        name = s.name
+    study_names.append((name, s))
+
+# === Study 选择器 ===
+study_name_list = [name for name, _ in study_names]
+selected_study_name = st.sidebar.selectbox("Select a Study", study_name_list)
+selected_study_path = dict(study_names)[selected_study_name]
+
+# === Study 概览区 ===
+st.subheader(f"📁 Study Overview: {selected_study_name}")
+st.markdown("Metadata, description, config summary can be shown here (to be implemented).")
+st.divider()
+
+# === 加载所有 Trial ===
+def get_all_trials(study_path: Path) -> List[Path]:
+    return [p for p in study_path.iterdir() if p.is_dir()]
+
+trials = get_all_trials(selected_study_path)
+trial_names = [t.name for t in trials]
+selected_trial_name = st.sidebar.selectbox("🧪 Select Trial", trial_names)
+selected_trial_path = selected_study_path / selected_trial_name
+
+# === 加载 track 的 metric 数据 ===
+def load_track_data(trial_path: Path) -> Dict[str, List[Dict]]:
+    track_dir = trial_path / "logs/scalar"
+    result = {}
+    if not track_dir.exists(): return result
+
+    for f in track_dir.glob("metrics_*.toml"):
+        track_name = f.stem.replace("metrics_", "")
+        with open(f, "rb") as fp:
+            try:
+                data = tomli.load(fp).get("metrics", [])
+                result[track_name] = data
+            except:
+                continue
+    return result
+
+# === 展示 Trial 详细信息 ===
+st.subheader(f"📊 Trial: {selected_trial_name}")
+track_data = load_track_data(selected_trial_path)
+
+if not track_data:
+    st.warning("No metrics data found for this trial.")
+else:
+    for track_name, records in track_data.items():
+        st.markdown(f"### Track: {track_name}")
+        if not records:
+            st.info("No data in this track.")
+            continue
+
+        # 自动获取所有标量键（除 global_step）
+        keys = [k for k in records[0].keys() if k != "global_step"]
+        for key in keys:
+            fig = go.Figure()
+            fig.add_trace(go.Scatter(
+                x=[r["global_step"] for r in records],
+                y=[r.get(key, None) for r in records],
+                mode='lines+markers',
+                name=key
+            ))
+            fig.update_layout(title=f"{track_name.upper()} - {key}", template="plotly_dark", height=350)
+            st.plotly_chart(fig, use_container_width=True)
+
+st.divider()
+st.markdown("*Flowillower 🌸 — Minimalist, Poetic and Open Source.*")