app.py

import streamlit as st
from openai import OpenAI

# 设置 API 参数
base_url = "https://internlm-chat.intern-ai.org.cn/puyu/api/v1/"
api_key = "sk-请填写准确的 token！"
model = "internlm2.5-latest"

# 创建 OpenAI 客户端
client = OpenAI(
    api_key=api_key,
    base_url=base_url,
)

# 设置页面标题
st.title("Chat with InternLM")

# 创建一个文本输入框供用户输入问题
user_input = st.text_input("请输入你的问题:")

# 按钮用于提交问题
if st.button("发送"):
    # 调用 OpenAI API 获取回复
    chat_rsp = client.chat.completions.create(
        model=model,
        messages=[{"role": "user", "content": user_input}],
    )

    # 显示回复
    for choice in chat_rsp.choices:
        st.write(f"回复: {choice.message.content}")