143 lines
3.8 KiB
Python
143 lines
3.8 KiB
Python
|
"""
|
|||
|
|
公共 UI 组件与工具函数，供各 tab_*.py 复用。
|
|||
|
|
|
|||
|
|
- 纯函数（如 sanitize_filename）不依赖 streamlit，避免循环导入。
|
|||
|
|
- 渲染组件（如 render_section_header、render_download_button）依赖 streamlit。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
from typing import Callable, Optional, List, Any
|
|||
|
|
|
|||
|
|
import streamlit as st
|
|||
|
|
|
|||
|
|
# 文件名非法字符，与 geo_tool 主入口规则一致
|
|||
|
|
INVALID_FS_CHARS = r'<>:"/\\|?*\n\r\t'
|
|||
|
|
|
|||
|
|
|
|||
|
|
def sanitize_filename(name: str, max_len: int = 80) -> str:
|
|||
|
|
"""将字符串清理为安全文件名（用于下载等）。各 Tab 统一由此处提供，避免重复实现。"""
|
|||
|
|
if not name:
|
|||
|
|
return "untitled"
|
|||
|
|
name = name.strip()
|
|||
|
|
name = re.sub(rf"[{re.escape(INVALID_FS_CHARS)}]", "_", name)
|
|||
|
|
name = re.sub(r"_+", "_", name).strip("_")
|
|||
|
|
return name[:max_len] if len(name) > max_len else name
|
|||
|
|
|
|||
|
|
|
|||
|
|
def extract_json_array(text: str) -> Optional[List[Any]]:
|
|||
|
|
"""从模型输出中抽取 JSON 数组（JsonOutputParser 失败时兜底）。"""
|
|||
|
|
if not text:
|
|||
|
|
return None
|
|||
|
|
m = re.search(r"\[[\s\S]*\]", text)
|
|||
|
|
if not m:
|
|||
|
|
return None
|
|||
|
|
try:
|
|||
|
|
return json.loads(m.group(0))
|
|||
|
|
except Exception:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def safe_decode_uploaded(uploaded) -> str:
|
|||
|
|
"""安全解码上传的文件内容"""
|
|||
|
|
if not uploaded:
|
|||
|
|
return ""
|
|||
|
|
b = uploaded.getvalue()
|
|||
|
|
for enc in ("utf-8-sig", "utf-8", "gb18030"):
|
|||
|
|
try:
|
|||
|
|
return b.decode(enc)
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
return b.decode("utf-8", errors="replace")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def render_section_header(
|
|||
|
|
title: str,
|
|||
|
|
caption: Optional[str] = None,
|
|||
|
|
level: int = 3,
|
|||
|
|
) -> None:
|
|||
|
|
"""渲染区块标题与可选说明。level=3 为 ###，4 为 ####。"""
|
|||
|
|
prefix = "#" * level
|
|||
|
|
st.markdown(f"{prefix} {title}")
|
|||
|
|
if caption:
|
|||
|
|
st.caption(caption)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def render_download_button(
|
|||
|
|
label: str,
|
|||
|
|
data: str | bytes,
|
|||
|
|
filename: str,
|
|||
|
|
mime: str,
|
|||
|
|
key: str,
|
|||
|
|
use_container_width: bool = True,
|
|||
|
|
) -> None:
|
|||
|
|
"""统一风格的下载按钮。"""
|
|||
|
|
st.download_button(
|
|||
|
|
label,
|
|||
|
|
data=data,
|
|||
|
|
file_name=filename,
|
|||
|
|
mime=mime,
|
|||
|
|
key=key,
|
|||
|
|
use_container_width=use_container_width,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def render_tab_top_with_clear(
|
|||
|
|
title: str,
|
|||
|
|
caption: Optional[str] = None,
|
|||
|
|
clear_key: str = "tab_clear",
|
|||
|
|
clear_label: str = "清空本模块结果",
|
|||
|
|
on_clear: Optional[Callable[[], None]] = None,
|
|||
|
|
) -> None:
|
|||
|
|
"""
|
|||
|
|
Tab 顶部：左侧标题+说明，右侧「清空本模块结果」按钮。
|
|||
|
|
on_clear 若提供，会在点击清空时调用（可在此内清空 session_state 并 st.toast）。
|
|||
|
|
"""
|
|||
|
|
col_left, col_right = st.columns([3, 1])
|
|||
|
|
with col_left:
|
|||
|
|
st.markdown(f"**{title}**")
|
|||
|
|
if caption:
|
|||
|
|
st.caption(caption)
|
|||
|
|
with col_right:
|
|||
|
|
if st.button(clear_label, use_container_width=True, key=clear_key):
|
|||
|
|
if on_clear:
|
|||
|
|
on_clear()
|
|||
|
|
st.toast("已清空。")
|
|||
|
|
st.rerun()
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ========== 统一的消息提示组件 ==========
|
|||
|
|
|
|||
|
|
def show_success(message: str, toast: bool = True) -> None:
|
|||
|
|
"""显示成功消息"""
|
|||
|
|
st.success(message)
|
|||
|
|
if toast:
|
|||
|
|
st.toast(message, icon="✅")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def show_error(message: str, toast: bool = True) -> None:
|
|||
|
|
"""显示错误消息"""
|
|||
|
|
st.error(message)
|
|||
|
|
if toast:
|
|||
|
|
st.toast(message, icon="❌")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def show_warning(message: str, toast: bool = True) -> None:
|
|||
|
|
"""显示警告消息"""
|
|||
|
|
st.warning(message)
|
|||
|
|
if toast:
|
|||
|
|
st.toast(message, icon="⚠️")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def show_info(message: str, toast: bool = True) -> None:
|
|||
|
|
"""显示信息消息"""
|
|||
|
|
st.info(message)
|
|||
|
|
if toast:
|
|||
|
|
st.toast(message, icon="ℹ️")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def show_loading(func, message: str = "加载中..."):
|
|||
|
|
"""统一的加载状态包装器"""
|
|||
|
|
with st.spinner(message):
|
|||
|
|
return func()
|