pip install requests pillow

import os
import tkinter as tk
from tkinter import filedialog, messagebox
import json
import requests
import base64
from PIL import Image
from io import BytesIO

def get_file_content(filePath):
    with open(filePath, 'rb') as fp:
        return fp.read()
    
# 之前的水印去除API調(diào)用部分
class CommonOcr(object):
    def __init__(self, img_path=None, is_url=False):
        # 圖像水印去除
        self._url = 'https://api.textin.com/ai/service/v1/image/watermark_remove'
        # 請(qǐng)登錄后前往 “工作臺(tái)-賬號(hào)設(shè)置-開發(fā)者信息” 查看 x-ti-app-id
        # 示例代碼中 x-ti-app-id 非真實(shí)數(shù)據(jù)
        self._app_id = '73b************************d269'
        # 請(qǐng)登錄后前往 “工作臺(tái)-賬號(hào)設(shè)置-開發(fā)者信息” 查看 x-ti-secret-code
        # 示例代碼中 x-ti-secret-code 非真實(shí)數(shù)據(jù)
        self._secret_code = 'de1ac7*******************48993131'
        self._img_path = img_path
        self._is_url = is_url

    def recognize(self):
        head = {}
        try:
            head['x-ti-app-id'] = self._app_id
            head['x-ti-secret-code'] = self._secret_code
            if self._is_url:
                head['Content-Type'] = 'text/plain'
                body = self._img_path
            else:
                image = get_file_content(self._img_path)
                head['Content-Type'] = 'application/octet-stream'
                body = image
            result = requests.post(self._url, data=body, headers=head)
            return result.text
        except Exception as e:
            return e

def down_img(base64str,output_folder,img_name):
    try:
        img_data = base64.b64decode(base64str)
        img = Image.open(BytesIO(img_data))
        file_name = os.path.join(output_folder,img_name)
        img.save(file_name)
        # with open(output_folder, 'wb') as f:
        #     f.write(image_data)
        print(f"去水印圖片已經(jīng)保存到 {file_name}")
    except Exception as e:
        print(f"圖片去水印失敗: {e}")

    
# GUI界面部分
class WatermarkRemoverApp:
    def __init__(self, root):
        self.root = root
        self.root.title("去水印工具")
        self.root.geometry("400x200")

        self.select_button = tk.Button(root, text="選擇文件夾", command=self.select_folder)
        self.select_button.pack(pady=20)

        self.process_button = tk.Button(root, text="開始去水印", command=self.process_images, state=tk.DISABLED)
        self.process_button.pack(pady=20)

    def select_folder(self):
        self.folder_path = filedialog.askdirectory()
        if self.folder_path:
            self.process_button.config(state=tk.NORMAL)

    def process_images(self):
        output_folder = os.path.join(self.folder_path, "output")
        os.makedirs(output_folder, exist_ok=True)
        
        for file_name in os.listdir(self.folder_path):
            if file_name.endswith((".jpg",".jpeg", ".png",".bmp")):
                file_path = os.path.join(self.folder_path, file_name)
                response = CommonOcr(img_path=file_path).recognize()
                # 解析JSON字符串為Python字典
                data = json.loads(response)
                
                if data["code"] ==200:  # 假設(shè)接口返回包含"success"表示成功
                    down_img(data["result"]["image"],output_folder,file_name)
                else:
                    messagebox.showerror("錯(cuò)誤", f"圖片 {file_name} 去水印失??！{data['msg']}")
        
        messagebox.showinfo("完成", "所有圖片已處理完畢！")

if __name__ == "__main__":
    root = tk.Tk()
    app = WatermarkRemoverApp(root)
    root.mainloop()