app.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
QLib因子分析与回测可视化平台
主应用入口
"""

import streamlit as st
import sys
import os
from pathlib import Path

# 页面配置
st.set_page_config(
    page_title="QLib因子分析平台",
    page_icon="📊",
    layout="wide",
    initial_sidebar_state="expanded"
)

# 自定义CSS样式
st.markdown("""
<style>
    .main-header {
        font-size: 2.5rem;
        color: #1f77b4;
        text-align: center;
        margin-bottom: 2rem;
    }
    .section-header {
        font-size: 1.5rem;
        color: #2c3e50;
        margin-top: 2rem;
        margin-bottom: 1rem;
    }
    .metric-card {
        background-color: #f8f9fa;
        padding: 1rem;
        border-radius: 0.5rem;
        border-left: 4px solid #1f77b4;
    }
    .success-message {
        background-color: #d4edda;
        color: #155724;
        padding: 1rem;
        border-radius: 0.25rem;
        border: 1px solid #c3e6cb;
    }
    .warning-message {
        background-color: #fff3cd;
        color: #856404;
        padding: 1rem;
        border-radius: 0.25rem;
        border: 1px solid #ffeaa7;
    }
    .error-message {
        background-color: #f8d7da;
        color: #721c24;
        padding: 1rem;
        border-radius: 0.25rem;
        border: 1px solid #f5c6cb;
    }
</style>
""", unsafe_allow_html=True)

def main():
    """主应用入口"""
    
    # 侧边栏导航
    st.sidebar.markdown("# 📊 QLib因子分析平台")
    st.sidebar.markdown("---")
    
    # 导航菜单
    page_options = {
        "🏠 首页": "home",
        "📈 数据管理": "data",
        "🧮 因子管理": "factors", 
        "📊 因子分析": "analysis",
        "🎯 策略回测": "backtest",
        "⚙️ 系统设置": "settings"
    }
    
    selected_page = st.sidebar.selectbox(
        "选择功能模块",
        list(page_options.keys()),
        index=0
    )
    
    page_key = page_options[selected_page]
    
    # 页面内容
    if page_key == "home":
        show_home_page()
    elif page_key == "data":
        from qlib_factor_platform.ui.data_page import show_data_page
        show_data_page()
    elif page_key == "factors":
        from qlib_factor_platform.ui.factor_page import show_factor_page
        show_factor_page()
    elif page_key == "analysis":
        from qlib_factor_platform.ui.analysis_page import show_analysis_page
        show_analysis_page()
    elif page_key == "backtest":
        from qlib_factor_platform.ui.backtest_page import show_backtest_page
        show_backtest_page()
    elif page_key == "settings":
        show_settings_page()

def show_home_page():
    """显示首页"""
    st.markdown('<h1 class="main-header">QLib因子分析与回测可视化平台</h1>', unsafe_allow_html=True)
    
    # 平台介绍
    col1, col2 = st.columns([2, 1])
    
    with col1:
        st.markdown("### 🎯 平台简介")
        st.markdown("""
        本平台基于微软QLib框架构建，提供一站式的量化因子研究与策略回测解决方案：
        
        - **📈 数据管理**：集成akshare数据源，支持A股实时行情获取与本地缓存
        - **🧮 因子管理**：内置Alpha158/360等经典因子，支持自定义因子编辑
        - **📊 因子分析**：IC分析、分层回测、相关性分析等完整因子评价体系
        - **🎯 策略回测**：可视化策略配置、绩效分析、风险评估
        - **📱 用户友好**：Streamlit界面，操作简单，结果直观
        """)
        
        st.markdown("### 🚀 快速开始")
        st.markdown("""
        1. **数据准备**：在"数据管理"中配置股票代码和时间范围
        2. **因子选择**：在"因子管理"中选择或创建因子
        3. **因子分析**：在"因子分析"中评估因子有效性
        4. **策略回测**：在"策略回测"中配置参数并运行回测
        """)
    
    with col2:
        st.markdown("### 📋 系统状态")
        
        # 检查QLib状态
        try:
            import qlib
            qlib.init()
            qlib_status = "✅ 已安装"
            qlib_version = qlib.__version__
        except:
            qlib_status = "❌ 未安装"
            qlib_version = "N/A"
        
        st.metric("QLib状态", qlib_status)
        st.metric("QLib版本", qlib_version)
        
        # 检查数据状态
        data_path = "~/.qlib/qlib_data/cn_data"
        if os.path.exists(os.path.expanduser(data_path)):
            data_status = "✅ 已就绪"
        else:
            data_status = "⚠️ 需要初始化"
        
        st.metric("数据状态", data_status)
        
        # 检查依赖包
        required_packages = ["streamlit", "plotly", "pandas", "numpy"]
        missing_packages = []
        
        for pkg in required_packages:
            try:
                __import__(pkg)
            except ImportError:
                missing_packages.append(pkg)
        
        if not missing_packages:
            deps_status = "✅ 完整"
        else:
            deps_status = f"⚠️ 缺失: {', '.join(missing_packages)}"
        
        st.metric("依赖状态", deps_status)
    
    # 功能特性展示
    st.markdown("---")
    st.markdown("### ✨ 核心功能")
    
    feature_cols = st.columns(4)
    
    features = [
        ("📊", "数据可视化", "K线图、因子分布、绩效曲线"),
        ("🧮", "因子计算", "158+预置因子，支持自定义"),
        ("📈", "IC分析", "信息系数、IR比率、分层收益"),
        ("🎯", "策略回测", "多策略支持，详细绩效报告"),
        ("💾", "数据缓存", "本地存储，避免重复下载"),
        ("🔧", "参数调优", "可视化参数配置与优化"),
        ("📱", "界面友好", "响应式设计，操作简单"),
        ("🚀", "高性能", "并行计算，快速响应")
    ]
    
    for i, (icon, title, desc) in enumerate(features):
        with feature_cols[i % 4]:
            st.markdown(f"""
            <div class="metric-card">
                <h3>{icon} {title}</h3>
                <p>{desc}</p>
            </div>
            """, unsafe_allow_html=True)
    
    # 使用指南
    st.markdown("---")
    st.markdown("### 📖 使用指南")
    
    with st.expander("🔧 环境配置指南"):
        st.markdown("""
        **1. 安装依赖包**
        ```bash
        pip install qlib streamlit plotly pandas numpy akshare
        ```
        
        **2. 初始化QLib数据**
        ```python
        import qlib
        from qlib.constant import REG_CN
        qlib.init(provider_uri="~/.qlib/qlib_data/cn_data", region=REG_CN)
        ```
        
        **3. 启动应用**
        ```bash
        streamlit run app.py
        ```
        """)
    
    with st.expander("📚 因子开发指南"):
        st.markdown("""
        **自定义因子示例**
        ```python
        # 动量因子
        momentum = "Ref($close, -20) / $close - 1"
        
        # 波动率因子  
        volatility = "Std($close, 20) / Mean($close, 20)"
        
        # 成交量因子
        volume_factor = "Mean($volume, 5) / Mean($volume, 20)"
        ```
        """)
    
    with st.expander("🎯 策略配置指南"):
        st.markdown("""
        **策略参数说明**
        - **topk**: 选择因子值最高的前N只股票
        - **n_drop**: 每次调仓时随机剔除的股票数量
        - **调仓频率**: 支持1d、1w、1m等周期
        - **交易成本**: 佣金、印花税等费用设置
        """)

def show_settings_page():
    """显示设置页面"""
    st.markdown('<h1 class="main-header">⚙️ 系统设置</h1>', unsafe_allow_html=True)
    
    st.markdown("### 📊 基础配置")
    
    # 数据路径配置
    with st.expander("📁 数据路径配置"):
        provider_uri = st.text_input(
            "QLib数据路径",
            value="~/.qlib/qlib_data/cn_data",
            help="QLib数据存储路径，支持相对路径和绝对路径"
        )
        
        cache_dir = st.text_input(
            "缓存目录",
            value="./cache",
            help="应用缓存数据存储目录"
        )
        
        if st.button("保存路径配置"):
            # TODO: 实现配置保存逻辑
            st.success("路径配置已保存")
    
    # 性能配置
    with st.expander("⚡ 性能配置"):
        n_jobs = st.slider(
            "并行进程数",
            min_value=1,
            max_value=8,
            value=4,
            help="数据计算时使用的并行进程数"
        )
        
        chunk_size = st.slider(
            "数据块大小",
            min_value=100,
            max_value=5000,
            value=1000,
            step=100,
            help="数据处理时的块大小，影响内存使用"
        )
        
        cache_expire_days = st.slider(
            "缓存过期天数",
            min_value=1,
            max_value=30,
            value=7,
            help="数据缓存的有效期（天）"
        )
    
    # 显示配置
    with st.expander("🖥️ 界面配置"):
        theme = st.selectbox(
            "界面主题",
            ["Light", "Dark"],
            index=0
        )
        
        chart_style = st.selectbox(
            "图表样式",
            ["Plotly", "Matplotlib", "Altair"],
            index=0
        )
        
        auto_refresh = st.checkbox(
            "自动刷新数据",
            value=False,
            help="是否自动检查并刷新过期数据"
        )
    
    # 系统信息
    st.markdown("### 📋 系统信息")
    
    col1, col2 = st.columns(2)
    
    with col1:
        st.markdown("**环境信息**")
        st.text(f"Python版本: {sys.version}")
        st.text(f"工作目录: {os.getcwd()}")
        st.text(f"平台: {sys.platform}")
    
    with col2:
        st.markdown("**包版本信息**")
        try:
            import streamlit
            st.text(f"Streamlit: {streamlit.__version__}")
        except:
            st.text("Streamlit: 未安装")
            
        try:
            import pandas
            st.text(f"Pandas: {pandas.__version__}")
        except:
            st.text("Pandas: 未安装")
            
        try:
            import plotly
            st.text(f"Plotly: {plotly.__version__}")
        except:
            st.text("Plotly: 未安装")

if __name__ == "__main__":
    main()