| import time
|
| import base64
|
| import streamlit as st
|
| from agent.react_agent import ReactAgent
|
| import os
|
|
|
| current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
| st.title("智扫通机器人智能客服")
|
| st.divider()
|
|
|
| if "agent" not in st.session_state:
|
| st.session_state["agent"] = ReactAgent()
|
|
|
| if "message" not in st.session_state:
|
| st.session_state["message"] = []
|
|
|
| for message in st.session_state["message"]:
|
| st.chat_message(message["role"]).write(message["content"])
|
|
|
| prompt = st.chat_input()
|
|
|
|
|
| def set_background_with_opacity(image_path, opacity=0.5):
|
| """
|
| 设置带有透明度（半透明）的背景图片
|
| :param image_path: 本地图片路径
|
| :param opacity: 图片透明度，0为完全透明，1为完全不透明
|
| """
|
| with open(image_path, "rb") as f:
|
| img_data = f.read()
|
| encoded = base64.b64encode(img_data).decode()
|
| bg_url = f"data:image/jpg;base64,{encoded}"
|
|
|
| st.markdown(
|
| f"""
|
| <style>
|
| /* 主容器定位为相对位置，以便伪元素绝对定位 */
|
| .stApp {{
|
| position: relative;
|
| background: none; /* 移除原有背景 */
|
| }}
|
| /* 伪元素作为背景层 */
|
| .stApp::before {{
|
| content: "";
|
| position: absolute;
|
| top: 0;
|
| left: 0;
|
| width: 100%;
|
| height: 100%;
|
| background: url({bg_url}) no-repeat center center fixed;
|
| background-size: cover;
|
| opacity: {opacity}; /* 关键：设置图片透明度 */
|
| z-index: -1; /* 置于底层 */
|
| pointer-events: none; /* 让背景层不干扰鼠标事件 */
|
| }}
|
| /* 确保主要内容在背景层之上且有足够对比度，可选 */
|
| .stApp > div {{
|
| position: relative;
|
| z-index: 1;
|
| }}
|
| </style>
|
| """,
|
| unsafe_allow_html=True
|
| )
|
|
|
|
|
| set_background_with_opacity(os.path.join(current_dir, "iu.jpg"), opacity=0.5)
|
|
|
| if prompt:
|
| st.chat_message("user").write(prompt)
|
| st.session_state["message"].append({
|
| "role": "user", "content": prompt
|
| })
|
| response_messages = []
|
| with st.spinner("智能客服思考中..."):
|
| res_stream = st.session_state["agent"].execute_stream(prompt)
|
|
|
|
|
| def capture(generator, cache_list):
|
|
|
| for chunk in generator:
|
| cache_list.append(chunk)
|
| for char in chunk:
|
| time.sleep(0.01)
|
| yield char
|
|
|
|
|
| st.chat_message("assistant").write_stream(capture(res_stream, response_messages))
|
| st.session_state["message"].append({
|
| "role": "assistant", "content": response_messages[-1]
|
| })
|
|
|
