import json
import requests

class OCRClient:
    def __init__(self, app_id: str, secret_code: str):
        self.app_id = app_id
        self.secret_code = secret_code

    def recognize(self, file_content: bytes, options: dict) -> str:
        # 構建請求參數
        params = {}
        for key, value in options.items():
            params[key] = str(value)

        # 設置請求頭
        headers = {
            "x-ti-app-id": self.app_id,
            "x-ti-secret-code": self.secret_code,
            # 方式一：讀取本地文件
            "Content-Type": "application/octet-stream"
            # 方式二：使用URL方式
            # "Content-Type": "text/plain"
        }

        # 發送請求
        response = requests.post(
            f"https://api.textin.com/ai/service/v1/pdf_to_markdown",
            params=params,
            headers=headers,
            data=file_content
        )

        # 檢查響應狀態
        response.raise_for_status()
        return response.text

def main():
    # 創建客戶端實例
    client = OCRClient("你的x-ti-app-id", "你的x-ti-secret-code")

        # 插入下面的示例代碼

if __name__ == "__main__":
    main()

# 在main函數中插入
    # 讀取本地文件
    with open("你的文件.pdf", "rb") as f:
        file_content = f.read()

    # 設置URL參數，可按需設置，這里已為你默認設置了一些參數
    options = dict(
        dpi=144,
        get_image="objects",
        markdown_details=1,
        page_count=10,
        parse_mode="auto",
        table_flavor="html",
    )

    try:
        response = client.recognize(file_content, options)

        # 保存完整的JSON響應到result.json文件
        with open("result.json", "w", encoding="utf-8") as f:
            f.write(response)

        # 解析JSON響應以提取markdown內容
        json_response = json.loads(response)
        if"result"in json_response and "markdown"in json_response["result"]:
            markdown_content = json_response["result"]["markdown"]
            with open("result.md", "w", encoding="utf-8") as f:
                f.write(markdown_content)

        print(response)
    except Exception as e:
        print(f"Error: {e}")

# 在main函數中插入
    # 讀取本地文件夾
    input_dir = "./tmp"# 你可以修改為自己的文件夾
    output_dir = "./output"# 輸出結果的文件夾
        import os
    os.makedirs(output_dir, exist_ok=True)

    # 支持的文件類型
    exts = (".pdf",".png",".jpg",".jpeg",".bmp",".tiff",".webp",".doc",".docx",".html",".mhtml",".xls",".xlsx",".csv",".ppt",".pptx",".txt")
    files = [f for f in os.listdir(input_dir) if f.lower().endswith(exts)]

    # 設置URL參數，可按需設置，這里已為你默認設置了一些參數
    options = dict(
        dpi=144,
        get_image="objects",
        markdown_details=1,
        page_count=10,
        parse_mode="auto",
        table_flavor="html",
    )

    #循環處理
    for filename in files:
        file_path = os.path.join(input_dir, filename)
        with open(file_path, "rb") as f:
            file_content = f.read()
        try:
            response = client.recognize(file_content, options)
            base_name = os.path.splitext(filename)[0]
            # 保存JSON
            with open(os.path.join(output_dir, f"{base_name}.json"), "w", encoding="utf-8") as fw:
                fw.write(response)
            # 保存Markdown
            json_response = json.loads(response)
            if"result"in json_response and "markdown"in json_response["result"]:
                markdown_content = json_response["result"]["markdown"]
                with open(os.path.join(output_dir, f"{base_name}.md"), "w", encoding="utf-8") as fw:
                    fw.write(markdown_content)
            print(f"{filename} 處理完成")
        except Exception as e:
            print(f"{filename} 處理出錯: {e}")