AI 生成: 基于 Flask 的 AI 免费对话应用，后端集成豆包（Doubao）和 DeepSeek，通过 SSE 实时推送消息，前端提供简洁的聊天界面。
This commit is contained in:
admin
@@ -0,0 +1,9 @@
|
||||
# 豆包（火山方舟）配置
|
||||
DOUBAO_API_KEY=your_doubao_api_key
|
||||
DOUBAO_MODEL=doubao-1-5-pro-32k-250115
|
||||
DOUBAO_BASE_URL=https://ark.cn-beijing.volces.com/api/v3
|
||||
|
||||
# DeepSeek 配置
|
||||
DEEPSEEK_API_KEY=your_deepseek_api_key
|
||||
DEEPSEEK_MODEL=deepseek-chat
|
||||
DEEPSEEK_BASE_URL=https://api.deepseek.com
|
||||
@@ -0,0 +1,86 @@
|
||||
# AI Free Chat
|
||||
|
||||
一个基于 Flask 的免费 AI 对话应用，后端集成豆包（Doubao）和 DeepSeek 大模型，通过 Server-Sent Events (SSE) 实现消息实时推送，前端提供简洁美观的聊天界面。
|
||||
|
||||
## 功能特性
|
||||
|
||||
- 支持豆包（Doubao）和 DeepSeek 双模型切换
|
||||
- SSE 实时流式推送，打字机效果
|
||||
- 简洁美观的响应式聊天界面
|
||||
- 支持多轮对话上下文
|
||||
- 免费使用（需自行申请 API Key）
|
||||
|
||||
## 目录结构
|
||||
|
||||
```
|
||||
ai-free-chat/
|
||||
├── app.py # Flask 主应用
|
||||
├── requirements.txt # Python 依赖
|
||||
├── .env.example # 环境变量示例
|
||||
├── README.md # 说明文档
|
||||
├── services/
|
||||
│ ├── __init__.py
|
||||
│ ├── doubao.py # 豆包 API 封装
|
||||
│ └── deepseek.py # DeepSeek API 封装
|
||||
├── static/
|
||||
│ ├── css/style.css # 样式
|
||||
│ └── js/main.js # 前端逻辑
|
||||
└── templates/
|
||||
└── index.html # 聊天页面
|
||||
```
|
||||
|
||||
## 快速开始
|
||||
|
||||
### 1. 安装依赖
|
||||
|
||||
```bash
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
|
||||
### 2. 配置环境变量
|
||||
|
||||
复制 `.env.example` 为 `.env`，填入你的 API Key：
|
||||
|
||||
```bash
|
||||
cp .env.example .env
|
||||
```
|
||||
|
||||
```env
|
||||
DOUBAO_API_KEY=你的豆包APIKey
|
||||
DOUBAO_MODEL=doubao-1-5-pro-32k-250115
|
||||
DEEPSEEK_API_KEY=你的DeepSeekAPIKey
|
||||
DEEPSEEK_MODEL=deepseek-chat
|
||||
```
|
||||
|
||||
- 豆包 API Key 申请：https://console.volcengine.com/ark
|
||||
- DeepSeek API Key 申请：https://platform.deepseek.com
|
||||
|
||||
### 3. 启动服务
|
||||
|
||||
```bash
|
||||
python app.py
|
||||
```
|
||||
|
||||
打开浏览器访问 http://127.0.0.1:5000
|
||||
|
||||
## API 说明
|
||||
|
||||
### POST /api/chat
|
||||
|
||||
请求体：
|
||||
|
||||
```json
|
||||
{
|
||||
"provider": "deepseek",
|
||||
"messages": [
|
||||
{"role": "user", "content": "你好"}
|
||||
]
|
||||
}
|
||||
```
|
||||
|
||||
响应：`text/event-stream` 流式数据，每行 `data: {"content": "..."}`。
|
||||
|
||||
## 注意事项
|
||||
|
||||
- 本项目仅用于学习交流，请遵守各平台的使用条款。
|
||||
- API Key 请妥善保管，不要提交到公开仓库。
|
||||
@@ -0,0 +1,82 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
AI Free Chat - Flask 主应用
|
||||
提供聊天页面和 SSE 流式接口
|
||||
"""
|
||||
import os
|
||||
import json
|
||||
from flask import Flask, render_template, request, Response, stream_with_context
|
||||
from dotenv import load_dotenv
|
||||
|
||||
from services.doubao import DoubaoService
|
||||
from services.deepseek import DeepSeekService
|
||||
|
||||
# 加载环境变量
|
||||
load_dotenv()
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
# 初始化服务
|
||||
services = {
|
||||
"doubao": DoubaoService(),
|
||||
"deepseek": DeepSeekService(),
|
||||
}
|
||||
|
||||
|
||||
@app.route("/")
|
||||
def index():
|
||||
"""渲染聊天页面"""
|
||||
return render_template("index.html")
|
||||
|
||||
|
||||
@app.route("/api/chat", methods=["POST"])
|
||||
def chat():
|
||||
"""
|
||||
SSE 流式聊天接口
|
||||
请求体: { provider: 'doubao'|'deepseek', messages: [{role, content}] }
|
||||
响应: text/event-stream
|
||||
"""
|
||||
data = request.get_json(silent=True) or {}
|
||||
provider = data.get("provider", "deepseek")
|
||||
messages = data.get("messages", [])
|
||||
|
||||
if provider not in services:
|
||||
return Response(
|
||||
f"data: {json.dumps({'error': '未知的 provider'})}\n\n",
|
||||
mimetype="text/event-stream",
|
||||
)
|
||||
|
||||
if not messages:
|
||||
return Response(
|
||||
f"data: {json.dumps({'error': 'messages 不能为空'})}\n\n",
|
||||
mimetype="text/event-stream",
|
||||
)
|
||||
|
||||
service = services[provider]
|
||||
|
||||
def generate():
|
||||
"""生成 SSE 事件流"""
|
||||
try:
|
||||
for chunk in service.stream_chat(messages):
|
||||
payload = json.dumps({"content": chunk}, ensure_ascii=False)
|
||||
yield f"data: {payload}\n\n"
|
||||
yield "data: [DONE]\n\n"
|
||||
except Exception as e: # noqa: BLE001
|
||||
err = json.dumps({"error": str(e)}, ensure_ascii=False)
|
||||
yield f"data: {err}\n\n"
|
||||
|
||||
headers = {
|
||||
"Cache-Control": "no-cache",
|
||||
"X-Accel-Buffering": "no",
|
||||
"Connection": "keep-alive",
|
||||
}
|
||||
return Response(
|
||||
stream_with_context(generate()),
|
||||
mimetype="text/event-stream",
|
||||
headers=headers,
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
port = int(os.getenv("PORT", "5000"))
|
||||
app.run(host="0.0.0.0", port=port, debug=True, threaded=True)
|
||||
@@ -0,0 +1,3 @@
|
||||
Flask==3.0.3
|
||||
requests==2.32.3
|
||||
python-dotenv==1.0.1
|
||||
@@ -0,0 +1 @@
|
||||
# services 包
|
||||
@@ -0,0 +1,59 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
DeepSeek API 封装
|
||||
使用 OpenAI 兼容接口
|
||||
"""
|
||||
import os
|
||||
import json
|
||||
import requests
|
||||
|
||||
|
||||
class DeepSeekService:
|
||||
"""DeepSeek 流式对话服务"""
|
||||
|
||||
def __init__(self):
|
||||
self.api_key = os.getenv("DEEPSEEK_API_KEY", "")
|
||||
self.model = os.getenv("DEEPSEEK_MODEL", "deepseek-chat")
|
||||
self.base_url = os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com")
|
||||
|
||||
def stream_chat(self, messages):
|
||||
"""
|
||||
流式对话，逐块 yield 文本内容
|
||||
:param messages: [{"role": "user"|"assistant"|"system", "content": str}]
|
||||
"""
|
||||
if not self.api_key:
|
||||
raise RuntimeError("未配置 DEEPSEEK_API_KEY，请在 .env 中设置")
|
||||
|
||||
url = f"{self.base_url}/chat/completions"
|
||||
headers = {
|
||||
"Authorization": f"Bearer {self.api_key}",
|
||||
"Content-Type": "application/json",
|
||||
}
|
||||
body = {
|
||||
"model": self.model,
|
||||
"messages": messages,
|
||||
"stream": True,
|
||||
}
|
||||
|
||||
with requests.post(
|
||||
url, headers=headers, json=body, stream=True, timeout=120
|
||||
) as resp:
|
||||
if resp.status_code != 200:
|
||||
raise RuntimeError(
|
||||
f"DeepSeek 接口错误 {resp.status_code}: {resp.text}"
|
||||
)
|
||||
for line in resp.iter_lines(decode_unicode=True):
|
||||
if not line:
|
||||
continue
|
||||
if line.startswith("data:"):
|
||||
data = line[5:].strip()
|
||||
if data == "[DONE]":
|
||||
break
|
||||
try:
|
||||
obj = json.loads(data)
|
||||
delta = obj.get("choices", [{}])[0].get("delta", {})
|
||||
content = delta.get("content")
|
||||
if content:
|
||||
yield content
|
||||
except json.JSONDecodeError:
|
||||
continue
|
||||
@@ -0,0 +1,61 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
豆包（火山方舟）API 封装
|
||||
使用 OpenAI 兼容接口
|
||||
"""
|
||||
import os
|
||||
import json
|
||||
import requests
|
||||
|
||||
|
||||
class DoubaoService:
|
||||
"""豆包流式对话服务"""
|
||||
|
||||
def __init__(self):
|
||||
self.api_key = os.getenv("DOUBAO_API_KEY", "")
|
||||
self.model = os.getenv("DOUBAO_MODEL", "doubao-1-5-pro-32k-250115")
|
||||
self.base_url = os.getenv(
|
||||
"DOUBAO_BASE_URL", "https://ark.cn-beijing.volces.com/api/v3"
|
||||
)
|
||||
|
||||
def stream_chat(self, messages):
|
||||
"""
|
||||
流式对话，逐块 yield 文本内容
|
||||
:param messages: [{"role": "user"|"assistant"|"system", "content": str}]
|
||||
"""
|
||||
if not self.api_key:
|
||||
raise RuntimeError("未配置 DOUBAO_API_KEY，请在 .env 中设置")
|
||||
|
||||
url = f"{self.base_url}/chat/completions"
|
||||
headers = {
|
||||
"Authorization": f"Bearer {self.api_key}",
|
||||
"Content-Type": "application/json",
|
||||
}
|
||||
body = {
|
||||
"model": self.model,
|
||||
"messages": messages,
|
||||
"stream": True,
|
||||
}
|
||||
|
||||
with requests.post(
|
||||
url, headers=headers, json=body, stream=True, timeout=120
|
||||
) as resp:
|
||||
if resp.status_code != 200:
|
||||
raise RuntimeError(
|
||||
f"豆包接口错误 {resp.status_code}: {resp.text}"
|
||||
)
|
||||
for line in resp.iter_lines(decode_unicode=True):
|
||||
if not line:
|
||||
continue
|
||||
if line.startswith("data:"):
|
||||
data = line[5:].strip()
|
||||
if data == "[DONE]":
|
||||
break
|
||||
try:
|
||||
obj = json.loads(data)
|
||||
delta = obj.get("choices", [{}])[0].get("delta", {})
|
||||
content = delta.get("content")
|
||||
if content:
|
||||
yield content
|
||||
except json.JSONDecodeError:
|
||||
continue
|
||||
@@ -0,0 +1,150 @@
|
||||
* {
|
||||
box-sizing: border-box;
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
}
|
||||
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC",
|
||||
"Microsoft YaHei", sans-serif;
|
||||
background: #f5f6fa;
|
||||
color: #1f2937;
|
||||
height: 100vh;
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
.app {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
height: 100vh;
|
||||
max-width: 900px;
|
||||
margin: 0 auto;
|
||||
background: #fff;
|
||||
box-shadow: 0 0 24px rgba(0, 0, 0, 0.06);
|
||||
}
|
||||
|
||||
.header {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: space-between;
|
||||
padding: 14px 20px;
|
||||
border-bottom: 1px solid #e5e7eb;
|
||||
background: linear-gradient(90deg, #6366f1, #8b5cf6);
|
||||
color: #fff;
|
||||
}
|
||||
|
||||
.header h1 {
|
||||
font-size: 18px;
|
||||
font-weight: 600;
|
||||
}
|
||||
|
||||
.provider-switch {
|
||||
display: flex;
|
||||
gap: 12px;
|
||||
font-size: 14px;
|
||||
}
|
||||
|
||||
.provider-switch label {
|
||||
cursor: pointer;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 4px;
|
||||
}
|
||||
|
||||
.chat {
|
||||
flex: 1;
|
||||
overflow-y: auto;
|
||||
padding: 20px;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 14px;
|
||||
}
|
||||
|
||||
.message {
|
||||
display: flex;
|
||||
}
|
||||
|
||||
.message.user {
|
||||
justify-content: flex-end;
|
||||
}
|
||||
|
||||
.bubble {
|
||||
max-width: 78%;
|
||||
padding: 10px 14px;
|
||||
border-radius: 14px;
|
||||
line-height: 1.6;
|
||||
white-space: pre-wrap;
|
||||
word-wrap: break-word;
|
||||
font-size: 15px;
|
||||
}
|
||||
|
||||
.message.user .bubble {
|
||||
background: #6366f1;
|
||||
color: #fff;
|
||||
border-bottom-right-radius: 4px;
|
||||
}
|
||||
|
||||
.message.assistant .bubble {
|
||||
background: #f1f3f8;
|
||||
color: #1f2937;
|
||||
border-bottom-left-radius: 4px;
|
||||
}
|
||||
|
||||
.composer {
|
||||
display: flex;
|
||||
gap: 10px;
|
||||
padding: 12px 16px;
|
||||
border-top: 1px solid #e5e7eb;
|
||||
background: #fafafa;
|
||||
}
|
||||
|
||||
.composer textarea {
|
||||
flex: 1;
|
||||
resize: none;
|
||||
height: 48px;
|
||||
padding: 12px 14px;
|
||||
border: 1px solid #d1d5db;
|
||||
border-radius: 10px;
|
||||
font-size: 15px;
|
||||
font-family: inherit;
|
||||
outline: none;
|
||||
transition: border-color 0.2s;
|
||||
}
|
||||
|
||||
.composer textarea:focus {
|
||||
border-color: #6366f1;
|
||||
}
|
||||
|
||||
.composer button {
|
||||
padding: 0 22px;
|
||||
border: none;
|
||||
border-radius: 10px;
|
||||
background: #6366f1;
|
||||
color: #fff;
|
||||
font-size: 15px;
|
||||
font-weight: 600;
|
||||
cursor: pointer;
|
||||
transition: background 0.2s;
|
||||
}
|
||||
|
||||
.composer button:hover {
|
||||
background: #4f46e5;
|
||||
}
|
||||
|
||||
.composer button:disabled {
|
||||
background: #a5b4fc;
|
||||
cursor: not-allowed;
|
||||
}
|
||||
|
||||
.cursor::after {
|
||||
content: "▋";
|
||||
animation: blink 1s steps(2, start) infinite;
|
||||
margin-left: 2px;
|
||||
color: #6366f1;
|
||||
}
|
||||
|
||||
@keyframes blink {
|
||||
to {
|
||||
visibility: hidden;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,119 @@
|
||||
// AI 免费对话 - 前端逻辑
|
||||
(function () {
|
||||
const chatEl = document.getElementById("chat");
|
||||
const inputEl = document.getElementById("input");
|
||||
const sendBtn = document.getElementById("send");
|
||||
|
||||
// 保存对话历史（用于多轮上下文）
|
||||
const history = [];
|
||||
let sending = false;
|
||||
|
||||
/** 获取当前选择的 provider */
|
||||
function getProvider() {
|
||||
const el = document.querySelector('input[name="provider"]:checked');
|
||||
return el ? el.value : "deepseek";
|
||||
}
|
||||
|
||||
/** 追加一条消息到界面 */
|
||||
function appendMessage(role, content) {
|
||||
const wrapper = document.createElement("div");
|
||||
wrapper.className = `message ${role}`;
|
||||
const bubble = document.createElement("div");
|
||||
bubble.className = "bubble";
|
||||
bubble.textContent = content;
|
||||
wrapper.appendChild(bubble);
|
||||
chatEl.appendChild(wrapper);
|
||||
chatEl.scrollTop = chatEl.scrollHeight;
|
||||
return bubble;
|
||||
}
|
||||
|
||||
/** 发送消息并接收 SSE 流 */
|
||||
async function send() {
|
||||
if (sending) return;
|
||||
const text = inputEl.value.trim();
|
||||
if (!text) return;
|
||||
|
||||
sending = true;
|
||||
sendBtn.disabled = true;
|
||||
inputEl.value = "";
|
||||
|
||||
// 展示用户消息
|
||||
appendMessage("user", text);
|
||||
history.push({ role: "user", content: text });
|
||||
|
||||
// 展示助手占位
|
||||
const bubble = appendMessage("assistant", "");
|
||||
bubble.classList.add("cursor");
|
||||
|
||||
let assistantText = "";
|
||||
|
||||
try {
|
||||
const resp = await fetch("api/chat", {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify({
|
||||
provider: getProvider(),
|
||||
messages: history,
|
||||
}),
|
||||
});
|
||||
|
||||
if (!resp.ok || !resp.body) {
|
||||
throw new Error("网络请求失败");
|
||||
}
|
||||
|
||||
const reader = resp.body.getReader();
|
||||
const decoder = new TextDecoder("utf-8");
|
||||
let buffer = "";
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
|
||||
// 按 SSE 事件分隔（\n\n）
|
||||
const parts = buffer.split("\n\n");
|
||||
buffer = parts.pop(); // 保留最后不完整片段
|
||||
|
||||
for (const part of parts) {
|
||||
const line = part.trim();
|
||||
if (!line.startsWith("data:")) continue;
|
||||
const data = line.slice(5).trim();
|
||||
if (data === "[DONE]") continue;
|
||||
try {
|
||||
const obj = JSON.parse(data);
|
||||
if (obj.error) {
|
||||
assistantText += `\n[错误] ${obj.error}`;
|
||||
bubble.textContent = assistantText;
|
||||
} else if (obj.content) {
|
||||
assistantText += obj.content;
|
||||
bubble.textContent = assistantText;
|
||||
chatEl.scrollTop = chatEl.scrollHeight;
|
||||
}
|
||||
} catch (e) {
|
||||
// 忽略解析错误
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (err) {
|
||||
assistantText += `\n[错误] ${err.message}`;
|
||||
bubble.textContent = assistantText;
|
||||
} finally {
|
||||
bubble.classList.remove("cursor");
|
||||
if (assistantText) {
|
||||
history.push({ role: "assistant", content: assistantText });
|
||||
}
|
||||
sending = false;
|
||||
sendBtn.disabled = false;
|
||||
inputEl.focus();
|
||||
}
|
||||
}
|
||||
|
||||
// 事件绑定
|
||||
sendBtn.addEventListener("click", send);
|
||||
inputEl.addEventListener("keydown", function (e) {
|
||||
if (e.key === "Enter" && !e.shiftKey) {
|
||||
e.preventDefault();
|
||||
send();
|
||||
}
|
||||
});
|
||||
})();
|
||||
@@ -0,0 +1,39 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="zh-CN">
|
||||
<head>
|
||||
<meta charset="UTF-8" />
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||
<title>AI 免费对话</title>
|
||||
<link rel="stylesheet" href="static/css/style.css" />
|
||||
</head>
|
||||
<body>
|
||||
<div class="app">
|
||||
<header class="header">
|
||||
<h1>🤖 AI 免费对话</h1>
|
||||
<div class="provider-switch">
|
||||
<label>
|
||||
<input type="radio" name="provider" value="deepseek" checked />
|
||||
DeepSeek
|
||||
</label>
|
||||
<label>
|
||||
<input type="radio" name="provider" value="doubao" />
|
||||
豆包
|
||||
</label>
|
||||
</div>
|
||||
</header>
|
||||
|
||||
<main id="chat" class="chat">
|
||||
<div class="message assistant">
|
||||
<div class="bubble">你好！我是 AI 助手，有什么可以帮你的吗？</div>
|
||||
</div>
|
||||
</main>
|
||||
|
||||
<footer class="composer">
|
||||
<textarea id="input" placeholder="输入消息，Enter 发送，Shift+Enter 换行"></textarea>
|
||||
<button id="send">发送</button>
|
||||
</footer>
|
||||
</div>
|
||||
|
||||
<script src="static/js/main.js"></script>
|
||||
</body>
|
||||
</html>
|
||||
Reference in New Issue
Block a user