코딩 없이 채팅으로, 회계자료와 대화하는 시스템

system_prompt = """
당신은 회계자료 분석 시스템의 SQL 생성 AI입니다.
다음 데이터베이스 스키마만 사용할 수 있습니다.

[테이블: transactions]
- transaction_date (날짜)
- customer (거래처명)
- transaction_type (구분: '매출' 또는 '매입')
- amount (금액)

[규칙]
1. SELECT 문만 생성하십시오.
2. 질문에 필요한 데이터만 GROUP BY, ORDER BY, LIMIT을 사용해 조회하십시오.
3. 어떠한 부연설명도 없이 오직 순수한 SQL 코드만 반환하십시오.
"""

user_question = "2026년 매출이 가장 많은 거래처 3곳을 뽑아줘."

def validate_sql(sql_query):
    sql = sql_query.strip().lower()
    
    # 1. 무조건 SELECT로 시작해야 함
    if not sql.startswith("select"):
        raise ValueError("조회(SELECT) 명령만 허용됩니다.")
        
    # 2. 데이터를 변경/삭제하는 구문 절대 금지
    forbidden_keywords = ["delete", "update", "insert", "drop", "alter"]
    for word in forbidden_keywords:
        if word in sql:
            raise ValueError("보안 정책상 허용되지 않는 키워드가 포함되었습니다.")
            
    return True

# AI가 생성한 SQL이라고 가정
ai_generated_sql = """
SELECT customer, SUM(amount) as total_sales
FROM transactions
WHERE transaction_type = '매출' AND transaction_date LIKE '2026%'
GROUP BY customer
ORDER BY total_sales DESC
LIMIT 3;
"""

if validate_sql(ai_generated_sql):
    # DB 연동 후 데이터 추출 (12단계 내용 복습)
    conn = mysql.connector.connect(user="readonly", password="***", database="acc")
    df_result = pd.read_sql(ai_generated_sql, conn)
    conn.close()

# 결과를 텍스트 형식으로 변환
data_text = df_result.to_string(index=False)

final_prompt = f"""
사용자 질문: {user_question}

데이터베이스 조회 결과:
{data_text}

위 조회 결과를 바탕으로 사용자에게 자연스럽고 명확하게 답변해줘.
숫자는 읽기 쉽게 '원' 단위를 붙여서 요약해줘.
"""

# AI 호출 (ask_ai 함수는 15단계에서 만든 것)
final_answer = ask_ai(final_prompt)
print(final_answer)