import os
import pandas as pd

video_data_list = []
user_data_list = []

folder_path = "out_put_US"

for filename in os.listdir(folder_path):
    if filename.endswith(".xlsx"):
        file_path = os.path.join(folder_path, filename)

        try:
            video_df = pd.read_excel(file_path, sheet_name="视频信息")
            user_df = pd.read_excel(file_path, sheet_name="用户信息")

            # 正确添加 来源文件列，不改动原来的 Index
            video_df["来源文件"] = filename
            user_df["来源文件"] = filename

            video_data_list.append(video_df)
            user_data_list.append(user_df)

        except Exception as e:
            print(f"❌ 读取失败: {filename}, 错误信息: {e}")

# 合并
all_video_df = pd.concat(video_data_list, ignore_index=True)
all_user_df = pd.concat(user_data_list, ignore_index=True)

# 写入一个Excel文件中两个Sheet
output_path = "合并视频用户信息.xlsx"
with pd.ExcelWriter(output_path, engine="openpyxl") as writer:
    all_video_df.to_excel(writer, sheet_name="视频信息", index=False)
    all_user_df.to_excel(writer, sheet_name="用户信息", index=False)

print(f"✅ 合并完成，文件保存为：{output_path}")