Python+ Streamlit + Ollama 实现一个网页版聊天机器人

Python + Streamlit + Ollama 实现一个网页版聊天机器人

一、python 安装 streamlit 和 ollama

pip install streamlit
pip install  install

二、完整代码

import  streamlit as st

import ollama
# 本地启动的 ollama
client = ollama.Client(host='http://localhost:11434')

if "messages" not in st.session_state:
    st.session_state['messages'] = []

# 标题
st.title("Streamlit + Ollama 实现一个聊天机器人")

# 分隔符
st.divider()

# 输入问题
prompt = st.chat_input("请输入你的问题")

if prompt:
    # role : user 、 assistant  ai human
    st.session_state['messages'].append({'role' : 'user' , 'content' : prompt})
    for message in st.session_state['messages']:
        st.chat_message(message['role']).markdown(message['content'])

    with st.spinner("🤔思考中。。。。"):
        response = client.chat(
            model='deepseek-r1:1.5b',
            messages=[{'role' : 'user' , 'content' : prompt}]
        )
        st.session_state['messages'].append({'role' : 'assistant' , 'content' : response['message']['content']})
        st.chat_message('assistant').markdown(response['message']['content'])

三、启动命令

    streamlit run streamlit_ollama.py

四、效果