AI数据工程是“炼油系统”，AI量化是“发动机”

import streamlit as st
import openai
from datetime import datetime

# --- 1. 配置页面 ---
st.set_page_config(page_title="AI周报助手", page_icon="📝")
st.title("📝 AI周报助手")
st.caption("输入你本周的零散工作记录，AI帮你整理成专业周报")

# --- 2. 侧边栏：API配置与个性化 ---
with st.sidebar:
    st.header("⚙️ 设置")
    api_key = st.text_input("输入你的OpenAI API Key", type="password")
    role = st.selectbox("你的角色", ["产品经理", "工程师", "设计师", "市场运营"])
    tone = st.selectbox("周报风格", ["简洁精炼", "详细充实", "数据驱动"])
    st.markdown("---")
    st.caption("数据仅存储在本地，不会上传")

# --- 3. 主功能区：用户输入 ---
raw_notes = st.text_area(
    "📌 粘贴你的工作记录（每行一条）",
    placeholder="例如：\n和研发对齐了需求优先级\n完成竞品分析文档\n用户反馈登录流程卡顿，已跟进",
    height=200
)

# --- 4. 一键生成，带防误触保护 ---
if st.button("🚀 生成周报", type="primary"):
    # 基础校验
    if not api_key:
        st.error("请先输入API Key")
    elif not raw_notes.strip():
        st.warning("工作记录不能为空")
    else:
        # 调用大模型（带错误处理）
        with st.spinner("AI正在整理中..."):
            try:
                client = openai.OpenAI(api_key=api_key)
                
                # 构建系统提示词
                system_prompt = f"""你是一位资深的{role}，擅长将零散的工作记录整理为结构清晰、重点突出的周报。
                风格要求：{tone}。
                请将用户输入的每一条记录归纳整合，输出包含【本周重点】、【具体进展】、【下周计划】三个板块的周报。
                语言专业但不生硬，可直接复制使用。"""
                
                response = client.chat.completions.create(
                    model="gpt-4o-mini",  # 成本友好
                    messages=[
                        {"role": "system", "content": system_prompt},
                        {"role": "user", "content": raw_notes}
                    ],
                    temperature=0.5  # 降低随机性，更稳定
                )
                
                report = response.choices[0].message.content
                
                # --- 5. 结果展示与交互设计 ---
                st.success("✅ 生成完成！")
                st.markdown("---")
                st.markdown(report)
                
                # 让用户能一键复制（Streamlit原生支持）
                st.download_button(
                    label="📥 下载为Markdown文件",
                    data=f"# {role}周报 - {datetime.now().strftime('%Y-%m-%d')}\n\n{report}",
                    file_name=f"周报_{datetime.now().strftime('%Y%m%d')}.md",
                    mime="text/markdown"
                )
                
                # 反馈闭环：简单的点赞/点踩
                col1, col2 = st.columns(2)
                with col1:
                    st.button("👍 有帮助", key="helpful")
                with col2:
                    st.button("👎 需改进", key="unhelpful")
                    
            except Exception as e:
                st.error(f"生成失败: {str(e)}")
                st.info("提示：请检查API Key是否正确，以及账户余额是否充足")

# --- 6. 使用示例（降低用户认知负担）---
with st.expander("💡 查看示例输入"):
    st.code("""
和前端确认了改版时间线
后台接口延迟问题，已提工单
周三开了需求评审会，结论OK
周五需要给老板汇报项目进度
    """)