from openai import OpenAI
import yaml
import os
import json
from dotenv import load_dotenv

load_dotenv()

# 클라이언트 및 프롬프트 초기화
client = OpenAI()

try:
    current_dir = os.path.dirname(os.path.abspath(__file__))
    prompt_path = os.path.join(current_dir, 'prompt.yaml')
    with open(prompt_path, "r", encoding='utf-8') as file:
        prompts = yaml.safe_load(file)
except Exception as e:
    print(f"Warning: prompt.yaml 로드 실패. 기본 프롬프트를 사용합니다. 오류: {e}")
    prompts = {
        "Interviewer": "You are a job interviewer.",
        "Student": "You are a job applicant.",
        "CoverLetter": "Write a cover letter based on the conversation.",
        "Memory": "Create a memory based on the conversation history."
    }

def get_interviewer_response(example_info):
    """
    진행률(progress)을 포함한 면접관의 응답을 스트리밍으로 생성합니다.
    """
    # 프롬프트 포매팅에 필요한 모든 변수를 kwargs로 묶기
    format_kwargs = {
        **example_info
    }
    system_prompt = prompts.get("Interviewer", "").format(**format_kwargs)
    
    with open("system_prompt.txt", "w", encoding='utf-8') as f:
        f.write(system_prompt)
    conversation = [{"role": "system", "content": "You must generate the response in json format."}, {"role": "user", "content": system_prompt}]
    # for role, content in messages:
    #     conversation.append({"role": role, "content": content})
        
    response_stream = client.chat.completions.create(
        model="gpt-4o",
        messages=conversation,
        stream=True
    )
    for chunk in response_stream:
        yield chunk.choices[0].delta.content or ""

def get_student_response(example_info):
    """학생의 AI 답변을 스트리밍으로 생성합니다."""
    system_prompt = prompts.get("Student", "").format(**example_info)
    
    conversation = [{"role": "system", "content": "You must generate the response in json format."}]
    
    with open("student_input.txt", "w", encoding='utf-8') as f:
        f.write(system_prompt)
    # for speaker, content in history:
    #     conversation.append({"role": "user", "content": f"{speaker}: {content}"})
    
    conversation.append({"role": "user", "content": f"{system_prompt}"})

    response_stream = client.chat.completions.create(
        model="gpt-4o",
        messages=conversation,
        stream=True
    )
    for chunk in response_stream:
        yield chunk.choices[0].delta.content or ""

def generate_cover_letter_response(question, conversation_history, example_info, flow, word_limit):
    """
    진행률을 포함하여 자기소개서 답변을 스트리밍으로 생성합니다.
    """
    # conversation_history가 비어있으면 example_info의 conversation을 사용
    if conversation_history:
        conversation_text = "\n".join([f"{speaker}: {content}" for speaker, content in conversation_history])
    else:
        conversation_text = example_info.get('conversation', '')
    
    # 필요한 변수들 추출
    company_name = example_info.get('company_name', '회사')
    job_position = example_info.get('job_position', '해당 직무')
    experience_level = example_info.get('experience_level', '신입')
    
    # 진행률 표시를 요청하는 프롬프트
    prompt = prompts.get("CoverLetter", "").format(
        question=question,
        guideline=flow,
        company_name=company_name,
        job_position=job_position,
        experience_level=experience_level,
        word_limit=word_limit,
        conversation=conversation_text
    )
    
    response_stream = client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": prompt}],
        stream=True
    )
    for chunk in response_stream:
        yield chunk.choices[0].delta.content or ""

def generate_memory(conversation_history, current_memory=""):
    """
    대화 기록을 바탕으로 메모리를 생성합니다.
    """
    # 대화 기록을 문자열로 변환
    if isinstance(conversation_history, list):
        conversation_text = "\n".join([f"{speaker}: {content}" for speaker, content in conversation_history])
    else:
        conversation_text = conversation_history
    
    # Memory 프롬프트 사용
    prompt = prompts.get("Memory", "").format(
        conversation=conversation_text,
        memory=current_memory
    )
    
    response_stream = client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": prompt}],
        stream=True
    )
    
    full_response = ""
    for chunk in response_stream:
        chunk_content = chunk.choices[0].delta.content or ""
        full_response += chunk_content
        yield chunk_content
        
    # 최종 응답에서 JSON 파싱 시도
    try:
        import json
        import re
        json_match = re.search(r'\{.*\}', full_response, re.DOTALL)
        if json_match:
            parsed_data = json.loads(json_match.group())
            return parsed_data.get('memory', full_response)
    except:
        pass
    
    return full_response