LPC/app.py

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
    )