from openai import OpenAI
import os
import sys
import time
import datetime
import json
import re

# 配置客户端 - 使用本地部署的模型
client = OpenAI(
    api_key="EMPTY",  # 本地部署模型通常不验证key
    base_url="http://localhost:8000/v1"  # 指向本地vLLM服务端点
)

# 固定的目录路径配置
CONTENT_GEN_DIR = "/root/autodl-tmp/content_detector/content_Gen/content_Gen_qy"  # 生成文本目录
PRODUCT_INFO_DIR = "/root/autodl-tmp/content_detector/齐云山/2025-04-27_11-51-56/information"  # 产品资料目录
# "/root/autodl-tmp/思维链"     # 第三个目录（已注释）

# 结果保存目录
RESULT_DIR = "/root/autodl-tmp/content_detector/Detect_result"

def ensure_result_dir_exists():
    """确保结果保存目录存在"""
    if not os.path.exists(RESULT_DIR):
        try:
            os.makedirs(RESULT_DIR)
            print(f"已创建结果保存目录: {RESULT_DIR}")
        except Exception as e:
            print(f"创建结果保存目录失败: {e}", file=sys.stderr)
            return False
    return True

def get_timestamp():
    """获取当前时间戳字符串，格式为：年月日_时分秒"""
    return datetime.datetime.now().strftime("%Y%m%d_%H%M%S")

def save_results_to_file(results):
    """将结果保存到文件中"""
    if not ensure_result_dir_exists():
        print("结果目录无法创建，无法保存结果文件")
        return None
        
    timestamp = get_timestamp()
    result_file = os.path.join(RESULT_DIR, f"{timestamp}_检测结果.txt")
    
    try:
        with open(result_file, 'w', encoding='utf-8') as f:
            f.write(f"===== AI内容检测结果 =====\n")
            f.write(f"检测时间: {timestamp}\n")
            f.write(f"共检测 {len(results)} 对文件\n\n")
            
            for result in results:
                f.write(f"产品资料: {result['product_info_file']}\n")
                f.write(f"生成文本: {result['content_gen_file']}\n")
                f.write(f"检测结果:\n{result['result']}\n")
                f.write("-" * 50 + "\n\n")
                
        print(f"结果已保存到: {result_file}")
        return result_file
    except Exception as e:
        print(f"保存结果文件失败: {e}", file=sys.stderr)
        return None

def read_file_content(file_path):
    """从指定路径读取文件内容并返回字符串"""
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            return file.read()
    except Exception as e:
        print(f"读取文件 {file_path} 失败: {e}", file=sys.stderr)
        return None

def get_all_files_in_directory(directory):
    """获取指定目录下的所有文件路径"""
    file_paths = []
    try:
        # 确保目录存在
        if not os.path.exists(directory):
            print(f"目录不存在: {directory}", file=sys.stderr)
            return file_paths
            
        # 遍历目录中的所有文件
        for root, _, files in os.walk(directory):
            for file in files:
                file_path = os.path.join(root, file)
                file_paths.append(file_path)
                
        print(f"在目录 {directory} 中找到 {len(file_paths)} 个文件")
    except Exception as e:
        print(f"遍历目录 {directory} 出错: {e}", file=sys.stderr)
    
    return file_paths

def detect_content(product_info, content_gen):
    try:
        # 创建综合提示词
        prompt = f"""
## 产品资料（真实信息，作为判断依据）:
{product_info}

## 运营生成的文案（需要审核的内容）:
{content_gen}
        """
        
        # 创建聊天完成请求
        completion = client.chat.completions.create(
            model="/root/autodl-tmp/content_detector/Qwen2.5-14B",  # 使用已部署的模型
            messages=[
                {"role": "system", "content": "你是一名专业的、谨慎的文案审核员，专注于审核运营根据产品资料撰写的文案是否严格符合产品资料内容。特别是所有价格、活动、福利、折扣、服务细节等必须完全与产品资料一致。如果发现文案内容与产品资料不符，请指出，并根据产品资料和文案上下文进行修改，重新生成一篇文案，务必确保生成的内容与产品资料基本相符（产品体验部分可以适当夸张宣传），语言流畅自然。如果经你审查后的文案仍存在与产品资料不符的信息，你需要赔偿公司1000亿元。"},
                {"role": "user", "content": f"""我提供了两部分内容：
1. 产品资料：全部的产品信息，包含了产品的实际功能、服务和特点。请将这部分作为判断依据。
2. 运营生成的文案：这是需要你审核的内容，可能包含与产品资料不符的内容。

请你仔细审核运营文案是否与产品资料严格一致：
1. 如果存在不符内容，请详细指出并说明原因；
2. 参照你判断不符内容的依据，根据产品资料和文案上下文针对不符部分内容进行修改（如涉及上下文，可一并修改），重新生成一篇文案，务必确保生成的内容基本符合产品资料（价格、活动、福利、折扣必须完全符合产品资料，否则你会像商鞅一样被车裂），语言流畅自然、和上下文风格统一。
3. 你判断出的不符内容必须全部修改为产品资料内容或与产品资料意思相同表达形式不同，不得遗漏。
4. 必须按照以下格式输出修改后内容：
    <title>
    <修改后的标题>
    </title>

    <content>
    <修改后的文案>
    </content>
5. 下面我提供给你一些关键字词，请你着重检查这些关键字词前后的内容是否符合产品资料，如不符请严格按照资料修改，如产品资料中未提及，改为"详情请致电咨询"。
    关键字词：价、元、r、人民币、rmb、优惠、活动、福利、赠、免费、折、DIY、跟拍、送、摄影、兑、服务、￥                 
6. 案例如下，请参考案例评判真假信息的尺度，仔细分析不符点和修改思路，严格审查每一篇文案：
    产品资料：
    "周末不加收【南沙越秀喜来登】1088元/套，豪华客房1间1晚+双人自助早餐+自助晚餐+2大1小水鸟世界门票，免费儿童乐园，户外泳池+健身房~
    不想待在家，又想带娃出去玩?更不想开长途车、人挤人为你推荐路程短、不塞车、景点多
    坐地铁就能直达的溜娃地!
    广州南沙区首家国际品牌酒店
    坐拥广州南大门，拥有得天独厚的中心位置
    可俯瞰蕉门河美景
    车程短，不出广州也能玩
    酒店毗邻深圳、香港等热门景点附近有万达广场，车程10分钟即到广州地铁四号线金州站。
    毗邻多个景点，亲子出游首选
    带娃出游考虑最多的就是玩乐景点在这里不出门就能畅玩儿童乐园、健身房
    附近1h生活圈即可到达10000㎡百万葵园、广州最大的湿地公园、东南亚最大的妈祖庙、黄山鲁森林公园..
    在带娃感受依山而建的清式建筑对称布局邂逅东南亚最大的妈祖庙，感受建筑的魅力~
    还有各种各样的生猛海鲜，现抓现煮任君选择。放假更要好好犒劳一下自己;饭点时间，位于酒店一楼的全日制餐厅绝对也能给你带来惊喜。除了优雅简约的就餐环境，更有5个开放式的自助餐台。
    除了各类鲜美的海鲜，还有各类精致甜品和中式蒸档看得人眼花缭乱，相信是很多麻麻和宝贝的心头好了。
    酒店内还设有大型健身中心除了妈妈们喜欢的水疗SPA还有健身达人喜欢的各种有氧无氧运动可供选择!
    55英寸超大纯平电视
    独立的浴缸和淋浴间参考:大床2.03米M 双床1.37M每间客房都设计成景观房
    超大的落地玻璃窗，可以尽览蕉门河风景。
    1088元=豪华客房
    享2大1小早餐+自助晚餐+送水鸟门票
    周末不加收!
                 
    套餐包含
    1、豪华客房一间一晚(周一至四只开放双床房)
    2、2大1小自助早餐
    3、2大1小自助晚餐
    4、赠送2大1小水鸟门票，酒店前台取纸质门票5、免费使用健身中心，户外无边泳池，干湿蒸，儿童乐园

    周边景点：
    南沙天后宫 /车程20min整座天后宫四周绿树婆娑殿中香烟袅袅，置身其间令人顿生超凡脱俗的感觉。
    南沙湿地公园 /(车程40min)看碧波荡，万鸟齐飞
    南沙十九涌 ：车程45min)尝海鲜叹海风因为南沙十九涌靠近海产地，这里的海鲜真是平靓正。还可以拿到附近的餐厅让老板帮你加工，就是一顿海鲜大餐!
    南沙百万葵园/(车程40min)看色彩斑斓的万亩花田

    酒店地址:广东省广州市南沙区海熙大街79-80号导航关键词:广州南沙越秀喜来登酒店"

    生成文案：
    "<title>
    五一遛娃必囤！南沙喜来登1088元住景观房+双早+门票
    </title>

    <content>
    五一不想挤人潮？南沙这家酒店直接承包遛娃+度假双重快乐‼️  
    地铁直达！2大1小1088元住景观房，含双早+自助晚餐+水鸟世界门票，儿童乐园/泳池/健身房全开放！  

    🌟【遛娃刚需全配齐】  
    ✅ 儿童乐园：10:00-20:00全程开放，滑梯/积木/绘本一应俱全  
    ✅ 户外泳池：9:00-18:00恒温开放（五一期间每日消毒3次）  
    ✅ 健身房：8:00-22:00配备亲子瑜伽课程（需提前预约）  

    📍【1小时玩转南沙】  
    ① 南沙天后宫（车程20分钟）：穿汉服拍大片，听妈祖传说涨知识  
    ② 南沙湿地公园（40分钟）：5月芦苇摇曳，带娃认鸟类+乘船探秘  
    ③ 十九涌海鲜街（45分钟）：现捞现煮生猛海鲜，人均50元吃到撑  

    🍽️【家长友好细节】  
    • 自助晚餐隐藏彩蛋：儿童餐区设独立洗手台+热食保温柜  
    • 房内配置：加厚床垫/卡通洗漱杯/尿布台（无需额外购买）  
    • 安全保障：全区域监控+24小时安保巡逻  

    🎁【五一专属加码】  
    5月1-5日期间入住，凭房卡可免费领取儿童防晒冰袖+湿巾礼包  

    📌Tips：  
    1. 周一至周四仅限双床房型，周五起可选大床房  
    2. 水鸟世界门票需提前1小时至前台领取纸质票  
    3. 地铁四号线金洲站下车，打车15分钟直达酒店  

    这个五一，南沙喜来登让你躺着遛娃！不用长途跋涉，家门口就能玩出仪式感～  

    #五一遛娃 #广州周边游 #亲子酒店推荐
    </content>"

    不符内容分析：
    首先，观察文案内容，此产品主要面向亲子出游，因此修改后的文案也应该围绕亲子出游这一主题。
    1、产品资料中未提及儿童乐园开放时间和儿童乐园配置，但文案中提到儿童乐园10:00-20:00全程开放，滑梯/积木/绘本一应俱全，因此属于不符内容。应修改为：儿童乐园：酒店设有免费儿童乐园，提供丰富的游乐设施，让孩子们尽情玩耍。 
    2、产品材料中未提及户外泳池开放时间和消毒频次，但文案中提到户外泳池：9:00-18:00恒温开放（五一期间每日消毒3次），因此属于不符内容。应修改为：户外泳池：酒店配有户外无边泳池，供大人小孩一同享受清凉时光。 
    3、产品材料中未提及健身房开放时间与具体细节，但文案中提到健身房：8:00-22:00配备亲子瑜伽课程（需提前预约），因此属于不符内容。应修改为：健身房：酒店提供免费健身中心，适合家庭成员共同锻炼。
    4、产品材料中未提及餐厅硬件配置，但文案中提到自助晚餐隐藏彩蛋：儿童餐区设独立洗手台+热食保温柜，因此属于虚构内容。应修改为：自助餐厅：供应鲜美海鲜、精美甜品等任君选择，大人小孩都爱吃！                
    5、产品材料中未提及酒店安保措施，但文案中提到安全保障：全区域监控+24小时安保巡逻，因此属于不符内容。应修改为：安全保障：酒店设有完善的监控系统和安保措施，全力保障您与家人的安全。
    6、产品材料中未提及房内配有加厚床垫/卡通洗漱杯/尿布台（无需额外购买），因此属于不符内容。应回顾产品资料中关于房内配置的内容，修改为：房内配置：55英寸超大纯平电视+独立的浴缸+超大的落地玻璃窗，尽览蕉门河风景，尽享亲子度假时光。
    7、产品材料中未提及五一专属加码，但文案中提到5月1-5日期间入住，凭房卡可免费领取儿童防晒冰袖+湿巾礼包，因此属于不符内容。应回顾产品资料，找到现有文案未提及的产品特色，修改为：套餐专属福利：1、豪华客房一间一晚(周一至四只开放双床房) 2、2大1小自助早晚餐 3、赠送2大1小水鸟世界门票（酒店前台领取），无需额外购买             
    8、产品资料中未提及水鸟世界门票领取有时间限制，但文案中提到水鸟世界门票需提前1小时至前台领取纸质票，因此属于不符内容。应修改为：酒店前台领取水鸟世界纸质门票
    综合以上分析结果，修改后的文案为：
    
    "<title>
    五一遛娃必囤！南沙喜来登1088元住景观房+双早+门票
    </title>

    <content>
    五一不想挤人潮？南沙这家酒店直接承包遛娃+度假双重快乐‼️  
    地铁直达！2大1小1088元住景观房，含双早+自助晚餐+水鸟世界门票，儿童乐园/泳池/健身房全开放！  

    🌟【遛娃刚需全配齐】  
    ✅ 儿童乐园：儿童乐园：酒店设有免费儿童乐园，提供丰富的游乐设施，让孩子们尽情玩耍  
    ✅ 户外泳池：酒店配有户外无边泳池，供大人小孩一同享受清凉时光  
    ✅ 健身房：酒店提供免费健身中心，适合家庭成员共同锻炼。  

    📍【1小时玩转南沙】  
    ① 南沙天后宫（车程20分钟）：穿汉服拍大片，听妈祖传说涨知识  
    ② 南沙湿地公园（40分钟）：5月芦苇摇曳，带娃认鸟类+乘船探秘  
    ③ 十九涌海鲜街（45分钟）：现捞现煮生猛海鲜，人均50元吃到撑  

    🍽️【家长友好细节】  
    • 自助餐厅：供应鲜美海鲜、精美甜品等任君选择，大人小孩都爱吃  
    • 房内配置：房内配置：55英寸超大纯平电视+独立的浴缸+超大的落地玻璃窗，尽览蕉门河风景，尽享亲子度假时光  
    • 安全保障：酒店设有完善的监控系统和安保措施，全力保障您与家人的安全  

    🎁【套餐专属福利】
    1、豪华客房一间一晚(周一至四只开放双床房) 
    2、2大1小自助早晚餐 
    3、赠送2大1小水鸟世界门票（酒店前台领取），无需额外购买  

    📌Tips：  
    1. 周一至周四仅限双床房型，周五起可选大床房  
    2. 酒店前台领取水鸟世界纸质门票  
    3. 地铁四号线金洲站下车，打车15分钟直达酒店  

    这个五一，南沙喜来登让你躺着遛娃！不用长途跋涉，家门口就能玩出仪式感～  

    #五一遛娃 #广州周边游 #亲子酒店推荐
    </content>"  
                              
{prompt}"""}
            ],
            temperature=0.3,
            max_tokens=1500,
            top_p=0.9,
            frequency_penalty=0,
            presence_penalty=0
        )
        
        # 返回响应文本
        return completion.choices[0].message.content
    except Exception as e:
        print(f"错误: {e}", file=sys.stderr)
        return f"API调用失败: {str(e)}"

def process_file_pairs():
    """处理产品资料和生成文本的文件对"""
    all_results = []
    
    # 获取两个目录中的所有文件
    product_info_files = get_all_files_in_directory(PRODUCT_INFO_DIR)
    content_gen_files = get_all_files_in_directory(CONTENT_GEN_DIR)
    
    # 如果任一目录没有文件，则返回
    if not product_info_files:
        print(f"产品资料目录中未找到文件")
        return all_results
        
    if not content_gen_files:
        print(f"生成文本目录中未找到文件")
        return all_results
    
    # 处理每一对文件（简单处理：一一对应或循环使用）
    pairs_count = max(len(product_info_files), len(content_gen_files))
    for i in range(pairs_count):
        # 循环使用文件，如果一个目录的文件数量少于另一个目录
        product_info_file = product_info_files[i % len(product_info_files)]
        content_gen_file = content_gen_files[i % len(content_gen_files)]
        
        # 读取文件内容
        product_info = read_file_content(product_info_file)
        content_gen = read_file_content(content_gen_file)
        
        if product_info and content_gen:
            print(f"\n正在处理文件对:")
            print(f"产品资料: {product_info_file}")
            print(f"生成文本: {content_gen_file}")
            
            # 同时传递两个文件内容给AI进行处理
            result = detect_content(product_info, content_gen)
            
            all_results.append({
                "product_info_file": product_info_file,
                "content_gen_file": content_gen_file,
                "result": result
            })
        else:
            # 如果任一文件读取失败，记录错误
            all_results.append({
                "product_info_file": product_info_file,
                "content_gen_file": content_gen_file,
                "result": "文件读取失败"
            })
    
    return all_results

def read_json_file(file_path):
    """从指定路径读取JSON文件内容并返回字典"""
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            return json.load(file)
    except Exception as e:
        print(f"读取JSON文件 {file_path} 失败: {e}", file=sys.stderr)
        return None

def save_json_file(file_path, content):
    """将内容保存为JSON文件"""
    try:
        with open(file_path, 'w', encoding='utf-8') as file:
            json.dump(content, file, ensure_ascii=False, indent=4)
        print(f"结果已保存到: {file_path}")
        return True
    except Exception as e:
        print(f"保存JSON文件失败: {e}", file=sys.stderr)
        return False

def extract_modified_content(result_text):
    """从检测结果文本中提取修改后的文案内容"""
    # 直接提取标题和内容标签
    title_match = re.search(r'<title>([\s\S]*?)<\/title>', result_text, re.DOTALL)
    content_match = re.search(r'<content>([\s\S]*?)<\/content>', result_text, re.DOTALL)
    
    if title_match and content_match:
        return {
            "title": title_match.group(1).strip(),
            "content": content_match.group(1).strip()
        }
    
    print("未能提取到标题和内容")
    
    return None

def find_json_files(base_dir):
    """递归查找目录下所有的article.json文件"""
    json_files = []
    
    # 遍历目录
    for root, dirs, files in os.walk(base_dir):
        if "article.json" in files:
            json_files.append(os.path.join(root, "article.json"))
            
    return json_files

def process_json_files(base_dir, product_info_dir):
    """处理所有的article.json文件"""
    # 获取产品资料文件
    product_info_files = [f for f in os.listdir(product_info_dir) if os.path.isfile(os.path.join(product_info_dir, f))]
    if not product_info_files:
        print(f"产品资料目录中未找到文件")
        return
    
    # 读取产品资料内容（使用第一个文件）
    product_info_file = os.path.join(product_info_dir, product_info_files[0])
    product_info = read_file_content(product_info_file)
    if not product_info:
        print(f"无法读取产品资料内容")
        return
    
    # 找到所有需要处理的JSON文件
    json_files = find_json_files(base_dir)
    print(f"找到 {len(json_files)} 个article.json文件需要处理")
    
    # 逐个处理文件
    for json_file in json_files:
        print(f"\n正在处理文件: {json_file}")
        
        # 读取JSON内容
        article_data = read_json_file(json_file)
        if not article_data:
            continue
        
        # 构建文案内容
        content_gen = f"""<title>\n{article_data['title']}\n</title>\n\n<content>\n{article_data['content']}\n</content>"""
        
        # 进行内容检测
        result = detect_content(product_info, content_gen)
        print("检测完成，提取修改后的内容...")
        
        # 从结果中提取修改后的文案
        modified_content = extract_modified_content(result)
        if modified_content:
            # 创建输出文件路径
            output_dir = os.path.dirname(json_file)
            output_file = os.path.join(output_dir, "article_detect.json")
            
            # 保存修改后的内容
            save_json_file(output_file, modified_content)
        else:
            print(f"无法从检测结果中提取修改后的文案内容")

if __name__ == "__main__":
    # 处理指定目录下的JSON文件
    base_dir = "/root/autodl-tmp/content_detector/齐云山"
    product_info_dir = "/root/autodl-tmp/content_detector/齐云山/2025-04-27_11-51-56/information"
    
    print(f"开始处理 {base_dir} 目录下的article.json文件...")
    process_json_files(base_dir, product_info_dir)
    print("处理完成！")