"""Proteobench Streamlit-based web server."""
from pathlib import Path
import streamlit as st
from _base import StreamlitPage
from UI_utils import (
build_submissions_figure,
build_tool_pie_chart,
get_monthly_visitors,
get_n_modules,
get_n_modules_proposed,
get_n_submitted_points,
get_n_supported_tools,
stat_box,
)
@st.dialog("Tool Breakdown", width="large")
def _show_tool_breakdown(module_title, tool_counts):
pie_fig = build_tool_pie_chart(module_title, tool_counts)
st.plotly_chart(pie_fig, use_container_width=True)
# Path to the index.rst file
file_path = Path(__file__).resolve().parent.parent / "docs" / "index.rst"
fig_path = Path(__file__).resolve().parent.parent / "img" / "icons" / "png" # Adjusted to match the new structure
[docs]
class StreamlitPageHome(StreamlitPage):
"""
This class sets up the main page layout for the Streamlit application.
"""
def _main_page(self):
"""
Set up the main page layout for the Streamlit application.
"""
# Placeholders TODO: replace with actual data
n_modules_all = get_n_modules()
n_modules_proposed = get_n_modules_proposed(file_path.read_text(encoding="utf-8"))
n_tools_supported = get_n_supported_tools()
n_of_points_submitted = get_n_submitted_points() # This function should return the number of submitted points
if (
"tracking" in st.secrets
and "matomo_endpoint" in st.secrets["tracking"]
and "matomo_idsite" in st.secrets["tracking"]
and "matomo_token" in st.secrets["tracking"]
):
monthly_uniq_visitors = get_monthly_visitors(
st.secrets["tracking"]["matomo_endpoint"],
st.secrets["tracking"]["matomo_token"],
st.secrets["tracking"]["matomo_idsite"],
)
else:
monthly_uniq_visitors = "not configured"
st.header("ProteoBench Overview")
st.markdown(
"""
<style>
.stat-card-glass {
backdrop-filter: blur(10px);
background: rgba(255, 255, 255, 0.6);
border-radius: 12px;
border: 1px solid rgba(255, 255, 255, 0.3);
padding: 12px;
box-shadow: 0 8px 32px 0 rgba(31, 38, 135, 0.1);
width: 100%;
display: flex;
flex-direction: column;
justify-content: center;
text-align: center;
}
.stat-card-glass h3 {
font-size: 1rem;
font-weight: 600;
margin-bottom: 8px;
}
.stat-card-glass .metric {
font-size: 1.5rem;
font-weight: 700;
margin: 0;
}
.stat-card-glass .icon {
font-size: 1.8rem;
margin-bottom: 8px;
}
</style>
""",
unsafe_allow_html=True,
)
# First row
row1_col1, row1_col2 = st.columns(2)
with row1_col1:
st.markdown(
stat_box(
"Active modules",
n_modules_all,
fig_path / "module.png",
url="https://proteobench.readthedocs.io/en/stable/available-modules/",
),
unsafe_allow_html=True,
)
with row1_col2:
st.markdown(
stat_box(
"Proposed and in-development modules",
n_modules_proposed,
fig_path / "module-construction.png",
url="https://github.com/orgs/Proteobench/discussions",
),
unsafe_allow_html=True,
)
st.markdown("<br>", unsafe_allow_html=True)
# Second row
row2_col1, row2_col2 = st.columns(2)
with row2_col1:
st.markdown(
stat_box(
"Supported workflows and tools",
n_tools_supported,
fig_path / "workflow-run.png",
url="https://proteobench.readthedocs.io/en/stable/available-modules/12-parsed-parameters-for-public-submission/",
),
unsafe_allow_html=True,
)
with row2_col2:
st.markdown(
stat_box(
"Submitted points",
n_of_points_submitted,
fig_path / "scatter-plot.png",
url="https://proteobench.cubimed.rub.de/datasets/",
),
unsafe_allow_html=True,
)
st.markdown("<br>", unsafe_allow_html=True)
# Third row
row3 = st.columns(1)
with row3[0]:
st.markdown(
stat_box("Monthly unique visitors", monthly_uniq_visitors, fig_path / "user.png"),
unsafe_allow_html=True,
)
# Submissions per module barplot
st.markdown("<br>", unsafe_allow_html=True)
st.subheader("Submissions per Module")
bar_fig, tool_data = build_submissions_figure()
if bar_fig is not None:
event = st.plotly_chart(
bar_fig,
use_container_width=True,
on_select="rerun",
selection_mode="points",
key="submissions_chart",
)
# Show pie chart in a dialog popup when a bar is clicked
selection = event.get("selection", {}) if event else {}
points = selection.get("points", [])
if points:
module_title = points[0].get("x")
if module_title and module_title in tool_data and tool_data[module_title]:
_show_tool_breakdown(module_title, tool_data[module_title])
else:
st.info("Submission data is currently unavailable.")
if __name__ == "__main__":
StreamlitPageHome()
