285 lines
9.4 KiB
Python
285 lines
9.4 KiB
Python
from flask import Flask, render_template, request, jsonify, session
|
||
from services.deepseek_service import deepseek_service
|
||
from config import Config
|
||
import uuid
|
||
|
||
app = Flask(__name__)
|
||
app.secret_key = "interviewer-secret-key-change-in-production"
|
||
|
||
INTERVIEW_PHASES = {
|
||
"intro": "自我介绍",
|
||
"professional": "专业能力",
|
||
"scenario": "情景假设",
|
||
"career": "职业规划",
|
||
"closing": "面试结束"
|
||
}
|
||
|
||
QUESTION_COUNTS = {
|
||
"intro": 2,
|
||
"professional": 4,
|
||
"scenario": 2,
|
||
"career": 1
|
||
}
|
||
|
||
@app.route("/")
|
||
def index():
|
||
return render_template("index.html")
|
||
|
||
|
||
@app.route("/api/chat", methods=["POST"])
|
||
def chat():
|
||
data = request.json
|
||
user_input = data.get("message", "").strip()
|
||
system_type = data.get("system_type", "general_assistant")
|
||
conversation_key = data.get("conversation_key", "default")
|
||
|
||
if not user_input:
|
||
return jsonify({"error": "请输入内容"}), 400
|
||
|
||
if conversation_key not in session:
|
||
session[conversation_key] = []
|
||
|
||
conversation_history = session[conversation_key]
|
||
|
||
try:
|
||
result = deepseek_service.chat(
|
||
user_input=user_input,
|
||
conversation_history=conversation_history,
|
||
system_type=system_type
|
||
)
|
||
|
||
conversation_history.append({"role": "user", "content": user_input})
|
||
conversation_history.append(result)
|
||
session[conversation_key] = conversation_history[-20:]
|
||
|
||
return jsonify({"response": result["content"]})
|
||
|
||
except Exception as e:
|
||
return jsonify({"error": str(e)}), 500
|
||
|
||
|
||
@app.route("/api/resume/optimize", methods=["POST"])
|
||
def optimize_resume():
|
||
data = request.json
|
||
resume_content = data.get("resume_content", "").strip()
|
||
target_position = data.get("target_position", "").strip()
|
||
|
||
if not resume_content:
|
||
return jsonify({"error": "请提供简历内容"}), 400
|
||
|
||
try:
|
||
result = deepseek_service.optimize_resume(resume_content, target_position)
|
||
return jsonify(result)
|
||
|
||
except Exception as e:
|
||
return jsonify({"error": str(e)}), 500
|
||
|
||
|
||
@app.route("/api/interview/start", methods=["POST"])
|
||
def start_interview():
|
||
data = request.json
|
||
job_position = data.get("job_position", "").strip()
|
||
difficulty = data.get("difficulty", "intermediate")
|
||
|
||
if not job_position:
|
||
return jsonify({"error": "请选择目标岗位"}), 400
|
||
|
||
interview_id = str(uuid.uuid4())
|
||
|
||
session[f"interview_{interview_id}"] = {
|
||
"job_position": job_position,
|
||
"difficulty": difficulty,
|
||
"current_phase": "intro",
|
||
"question_count": 0,
|
||
"conversation_history": [],
|
||
"is_active": True
|
||
}
|
||
|
||
first_question = deepseek_service.generate_interview_question(
|
||
job_position=job_position,
|
||
difficulty=difficulty,
|
||
phase="intro"
|
||
)
|
||
|
||
session[f"interview_{interview_id}"]["conversation_history"].append({
|
||
"role": "assistant",
|
||
"content": f"你好！我是面试官，现在开始针对{job_position}岗位的面试。\n\n{first_question}"
|
||
})
|
||
|
||
return jsonify({
|
||
"interview_id": interview_id,
|
||
"job_position": job_position,
|
||
"difficulty": difficulty,
|
||
"question": first_question,
|
||
"phase": "intro"
|
||
})
|
||
|
||
|
||
@app.route("/api/interview/answer", methods=["POST"])
|
||
def answer_question():
|
||
data = request.json
|
||
interview_id = data.get("interview_id", "").strip()
|
||
user_answer = data.get("answer", "").strip()
|
||
request_feedback = data.get("request_feedback", False)
|
||
|
||
if not interview_id:
|
||
return jsonify({"error": "无效的面试ID"}), 400
|
||
|
||
interview_key = f"interview_{interview_id}"
|
||
if interview_key not in session:
|
||
return jsonify({"error": "面试不存在或已结束"}), 400
|
||
|
||
interview_data = session[interview_key]
|
||
|
||
if not interview_data["is_active"]:
|
||
return jsonify({"error": "面试已结束"}), 400
|
||
|
||
if not user_answer:
|
||
return jsonify({"error": "请输入你的回答"}), 400
|
||
|
||
conversation_history = interview_data["conversation_history"]
|
||
|
||
if request_feedback:
|
||
last_question = ""
|
||
for msg in reversed(conversation_history):
|
||
if msg["role"] == "assistant" and "？" in msg["content"]:
|
||
last_question = msg["content"]
|
||
break
|
||
|
||
if last_question:
|
||
try:
|
||
feedback = deepseek_service.chat_with_feedback(
|
||
user_input=last_question,
|
||
user_answer=user_answer,
|
||
conversation_history=conversation_history[:-1]
|
||
)
|
||
conversation_history.append({"role": "user", "content": user_answer})
|
||
conversation_history.append(feedback)
|
||
|
||
return jsonify({
|
||
"feedback": feedback["content"],
|
||
"ended": False
|
||
})
|
||
except Exception as e:
|
||
return jsonify({"error": f"生成反馈失败：{str(e)}"}), 500
|
||
|
||
conversation_history.append({"role": "user", "content": user_answer})
|
||
|
||
interview_data["question_count"] += 1
|
||
current_phase = interview_data["current_phase"]
|
||
phase_order = ["intro", "professional", "scenario", "career", "closing"]
|
||
|
||
current_index = phase_order.index(current_phase)
|
||
questions_in_phase = QUESTION_COUNTS.get(current_phase, 1)
|
||
|
||
next_question = None
|
||
|
||
if interview_data["question_count"] >= questions_in_phase:
|
||
if current_index < len(phase_order) - 1:
|
||
next_phase = phase_order[current_index + 1]
|
||
interview_data["current_phase"] = next_phase
|
||
interview_data["question_count"] = 0
|
||
|
||
if next_phase == "closing":
|
||
conversation_history.append({
|
||
"role": "assistant",
|
||
"content": "面试结束！感谢你的参与。点击下方按钮获取本次面试的详细反馈。"
|
||
})
|
||
interview_data["is_active"] = False
|
||
|
||
return jsonify({
|
||
"question": None,
|
||
"feedback": "面试结束",
|
||
"ended": True,
|
||
"conversation_history": conversation_history[-10:]
|
||
})
|
||
else:
|
||
phase_names = {
|
||
"intro": "自我介绍",
|
||
"professional": "专业能力",
|
||
"scenario": "情景假设",
|
||
"career": "职业规划"
|
||
}
|
||
|
||
try:
|
||
next_question = deepseek_service.generate_interview_question(
|
||
job_position=interview_data["job_position"],
|
||
difficulty=interview_data["difficulty"],
|
||
conversation_history=conversation_history[:-1],
|
||
phase=next_phase
|
||
)
|
||
conversation_history.append({
|
||
"role": "assistant",
|
||
"content": f"（{phase_names.get(next_phase, next_phase)}阶段）\n\n{next_question}"
|
||
})
|
||
except Exception as e:
|
||
return jsonify({"error": f"生成问题失败：{str(e)}"}), 500
|
||
else:
|
||
try:
|
||
next_question = deepseek_service.generate_interview_question(
|
||
job_position=interview_data["job_position"],
|
||
difficulty=interview_data["difficulty"],
|
||
conversation_history=conversation_history[:-1],
|
||
phase=current_phase
|
||
)
|
||
conversation_history.append({"role": "assistant", "content": next_question})
|
||
except Exception as e:
|
||
return jsonify({"error": f"生成问题失败：{str(e)}"}), 500
|
||
|
||
session[interview_key] = interview_data
|
||
|
||
return jsonify({
|
||
"question": next_question,
|
||
"feedback": None,
|
||
"ended": False,
|
||
"phase": interview_data["current_phase"]
|
||
})
|
||
|
||
|
||
@app.route("/api/interview/feedback", methods=["POST"])
|
||
def get_interview_feedback():
|
||
data = request.json
|
||
interview_id = data.get("interview_id", "").strip()
|
||
conversation_history = data.get("conversation_history", [])
|
||
|
||
if not interview_id:
|
||
return jsonify({"error": "无效的面试ID"}), 400
|
||
|
||
system_prompt = """作为一位专业的面试评估专家，请对整场面试进行全面评估。
|
||
|
||
请提供：
|
||
1. 整体表现评分（0-100分）和评级（优秀/良好/一般/需改进）
|
||
2. 各轮回答的详细分析
|
||
3. strengths（优势）
|
||
4. areas_for_improvement（需要改进的方面）
|
||
5. 具体的准备建议
|
||
|
||
请用中文回复，格式清晰、结构化。"""
|
||
|
||
conversation_text = "\n\n".join([
|
||
f"{msg['role']}：{msg['content']}"
|
||
for msg in conversation_history[-30:]
|
||
])
|
||
|
||
messages = [
|
||
{"role": "system", "content": system_prompt},
|
||
{"role": "user", "content": f"请分析以下面试对话并给出综合反馈：\n\n{conversation_text}"}
|
||
]
|
||
|
||
try:
|
||
response = deepseek_service._call_api(messages)
|
||
feedback = response["choices"][0]["message"]["content"]
|
||
|
||
return jsonify({"feedback": feedback})
|
||
|
||
except Exception as e:
|
||
return jsonify({"error": str(e)}), 500
|
||
|
||
|
||
if __name__ == "__main__":
|
||
app.run(
|
||
host=Config.APP_HOST,
|
||
port=Config.APP_PORT,
|
||
debug=Config.DEBUG
|
||
)
|