删除 app.py
This commit is contained in:
parent
239f923074
commit
e3c1bed334
196
app.py
196
app.py
@ -1,196 +0,0 @@
|
||||
import os
|
||||
from flask import Flask, render_template, request, jsonify
|
||||
from dotenv import load_dotenv
|
||||
import dashscope
|
||||
from dashscope import Generation
|
||||
from datetime import datetime
|
||||
import json
|
||||
|
||||
load_dotenv()
|
||||
dashscope.api_key = os.getenv("DASHSCOPE_API_KEY")
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
# --------------------------
|
||||
# 配置
|
||||
# --------------------------
|
||||
SCENE_PROMPTS = {
|
||||
"general": "这是通用写作场景，语言自然清晰。",
|
||||
"media": "这是自媒体写作，语言有吸引力，适合传播。",
|
||||
"script": "这是剧本写作，语言有画面感和情绪张力。",
|
||||
"essay": "这是学生作文写作，结构清晰、表达规范。",
|
||||
"academic_scene": "这是学术写作，语言严谨、逻辑清楚。"
|
||||
}
|
||||
|
||||
POLISH_STYLES = {
|
||||
"formal": "正式、书面",
|
||||
"literary": "文学性强、有文采",
|
||||
"spoken": "口语化、自然",
|
||||
"business": "商务、专业",
|
||||
"internet": "网络风格、轻松"
|
||||
}
|
||||
|
||||
history_records = []
|
||||
MAX_HISTORY = 10
|
||||
|
||||
# --------------------------
|
||||
# AI 调用
|
||||
# --------------------------
|
||||
def call_ai(prompt):
|
||||
resp = Generation.call(model="qwen-turbo", prompt=prompt)
|
||||
if resp.status_code == 200:
|
||||
return resp.output.text
|
||||
return "❌ AI 调用失败"
|
||||
|
||||
# --------------------------
|
||||
# 多版本润色
|
||||
# --------------------------
|
||||
def multi_version_rewrite(text, style):
|
||||
prompt = f"""
|
||||
请对以下文本进行润色，生成 3 个不同版本：
|
||||
要求：
|
||||
1. 保持原意
|
||||
2. 风格：{style}
|
||||
3. 输出 3 个版本，分别标记 版本1/版本2/版本3
|
||||
原文：
|
||||
{text}
|
||||
"""
|
||||
return call_ai(prompt)
|
||||
|
||||
# --------------------------
|
||||
# 扩写
|
||||
# --------------------------
|
||||
def expand_text(text, target_length):
|
||||
"""
|
||||
扩写文本到指定长度（字数/字符）
|
||||
"""
|
||||
prompt = f"""
|
||||
请将以下文本扩写，保持原意，但丰富细节和表达，使字数达到 {target_length} 字左右。
|
||||
可以生成 3 个不同版本，分别标记 版本1/版本2/版本3。
|
||||
原文：
|
||||
{text}
|
||||
"""
|
||||
return call_ai(prompt)
|
||||
|
||||
# --------------------------
|
||||
# 修改原因生成
|
||||
# --------------------------
|
||||
def generate_revision_reason(original, revised):
|
||||
prompt = f"""
|
||||
请对下面两段文本进行对比，说明修改的原因。
|
||||
仅输出文字说明，不要高亮或格式化。
|
||||
|
||||
原文：
|
||||
{original}
|
||||
|
||||
改写后：
|
||||
{revised}
|
||||
|
||||
请详细说明改动原因。
|
||||
"""
|
||||
return call_ai(prompt)
|
||||
|
||||
# --------------------------
|
||||
# 文学参考
|
||||
# --------------------------
|
||||
def generate_literary_reference(text):
|
||||
prompt = f"""
|
||||
请理解下面这句话的含义，然后给出 3 条
|
||||
“文学作品风格中的相似表达示例”。
|
||||
|
||||
要求：
|
||||
1. 不是原文引用
|
||||
2. 给出【风格参考作者】和【作品名】
|
||||
3. 仅作为文学风格参考
|
||||
|
||||
输出格式：
|
||||
1. 句子内容
|
||||
—— 风格参考：作者《作品名》
|
||||
|
||||
原句：
|
||||
{text}
|
||||
"""
|
||||
return call_ai(prompt)
|
||||
|
||||
# --------------------------
|
||||
# 历史记录
|
||||
# --------------------------
|
||||
def save_history(data, result):
|
||||
history_records.insert(0, {
|
||||
"time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||
"mode": data.get("mode"),
|
||||
"scene": data.get("scene"),
|
||||
"input": data.get("text", "")[:100],
|
||||
"output": result[:200]
|
||||
})
|
||||
if len(history_records) > MAX_HISTORY:
|
||||
history_records.pop()
|
||||
|
||||
# --------------------------
|
||||
# 情感分析
|
||||
# --------------------------
|
||||
def sentiment_analysis(text):
|
||||
prompt = f"""
|
||||
请分析下面文本的情感倾向，仅返回严格 JSON 格式：
|
||||
{{"positive": %, "neutral": %, "negative": %}}
|
||||
文本：
|
||||
{text}
|
||||
"""
|
||||
result = call_ai(prompt)
|
||||
try:
|
||||
parsed = json.loads(result.replace("：", ":").replace("%",""))
|
||||
return parsed
|
||||
except:
|
||||
return {"positive":0, "neutral":0, "negative":0}
|
||||
|
||||
# --------------------------
|
||||
# 路由
|
||||
# --------------------------
|
||||
@app.route("/")
|
||||
def index():
|
||||
return render_template("index.html")
|
||||
|
||||
@app.route("/rewrite", methods=["POST"])
|
||||
def rewrite():
|
||||
data = request.json
|
||||
text = data.get("text","")
|
||||
mode = data.get("mode")
|
||||
scene = data.get("scene","general")
|
||||
|
||||
# 根据模式生成文本
|
||||
if mode=="polish":
|
||||
style = data.get("style","formal")
|
||||
result = multi_version_rewrite(text, style)
|
||||
elif mode=="expand":
|
||||
target_length = data.get("target_length", 300)
|
||||
result = expand_text(text, target_length)
|
||||
else:
|
||||
result = f"普通生成模式：{text}" # 保留原功能
|
||||
|
||||
save_history(data, result)
|
||||
return jsonify({"result": result})
|
||||
|
||||
@app.route("/literary", methods=["POST"])
|
||||
def literary():
|
||||
text = request.json.get("text","")
|
||||
return jsonify({"reference": generate_literary_reference(text)})
|
||||
|
||||
@app.route("/history")
|
||||
def history():
|
||||
return jsonify(history_records)
|
||||
|
||||
@app.route("/revision_reason", methods=["POST"])
|
||||
def revision_reason():
|
||||
data = request.json
|
||||
original = data.get("original","")
|
||||
revised = data.get("revised","")
|
||||
reason = generate_revision_reason(original, revised)
|
||||
return jsonify({"reason": reason})
|
||||
|
||||
@app.route("/sentiment", methods=["POST"])
|
||||
def sentiment():
|
||||
text = request.json.get("text","")
|
||||
return jsonify({"result": sentiment_analysis(text)})
|
||||
|
||||
if __name__=="__main__":
|
||||
app.run(debug=True)
|
||||
Loading…
Reference in New Issue
Block a user