import tkinter as tk from tkinter import filedialog, ttk from PIL import Image, ImageTk import requests import base64 import os import io import random import threading import time def encode_pil_to_base64(image): with io.BytesIO() as output_bytes: image.save(output_bytes, format="PNG") bytes_data = output_bytes.getvalue() return base64.b64encode(bytes_data).decode("utf-8") def save_decoded_image(b64_image, folder_path, image_name): seq = 0 output_path = os.path.join(folder_path, f"{image_name}.png") while os.path.exists(output_path): seq += 1 output_path = os.path.join(folder_path, f"{image_name}({seq}).png") with open(output_path, 'wb') as image_file: image_file.write(base64.b64decode(b64_image)) print(f"Image saved to: {output_path}") return output_path # 返回儲存的圖片路徑 class PhotoSketchGUI: def __init__(self, root): self.root = root self.root.title("建築效果圖轉彩色手繪") self.root.configure(bg='black') self.root.state('zoomed') self.folder_path = '' # 初始化 folder_path self.timer_running = False # 初始化計時器執行狀態為False self.start_time = None # 初始化計時器開始時間 self.buttons_y = None # 初始化按鈕的y座標 self.setup_ui() def setup_ui(self): screen_width = self.root.winfo_screenwidth() screen_height = self.root.winfo_screenheight() self.canvas_width = screen_width * 0.4 self.canvas_height = screen_height * 0.8 self.left_canvas_x = screen_width * 0.25 - self.canvas_width / 2 self.right_canvas_x = screen_width * 0.75 - self.canvas_width / 2 self.setup_buttons_and_canvas() self.add_intro_labels() def setup_buttons_and_canvas(self): self.left_frame = tk.Canvas(self.root, width=self.canvas_width, height=self.canvas_height, bg='black', highlightthickness=0) self.left_frame.place(x=self.left_canvas_x, rely=0.5, anchor='w') self.left_frame.create_rectangle(2, 2, self.canvas_width - 2, self.canvas_height - 2, outline='white', dash=(5, 5)) self.buttons_y = self.left_frame.winfo_y() + self.canvas_height + 95 # 在畫布下方95畫素處 self.upload_btn = tk.Button(self.root, text="上傳參考圖", command=self.upload_image, bg='darkgrey', fg='white', font=('微軟雅黑', 12)) self.upload_btn.place(x=self.left_canvas_x, y=self.buttons_y, width=(self.canvas_width / 2 - 5), height=30, anchor='nw') self.save_dir_btn = tk.Button(self.root, text="選擇儲存路徑", command=self.select_save_folder, bg='darkgrey', fg='white', font=('微軟雅黑', 12)) self.save_dir_btn.place(x=self.left_canvas_x + self.canvas_width / 2 + 5, y=self.buttons_y, width=(self.canvas_width / 2 - 5), height=30, anchor='nw') self.right_frame = tk.Canvas(self.root, width=self.canvas_width, height=self.canvas_height, bg='black', highlightthickness=0) self.right_frame.place(x=self.right_canvas_x, rely=0.5, anchor='w') self.right_frame.create_rectangle(2, 2, self.canvas_width - 2, self.canvas_height - 2, outline='white', dash=(5, 5)) def add_intro_labels(self): intro_label1 = tk.Label(self.root, text="建築效果圖轉彩色手繪", bg='black', fg='white', font=('微軟雅黑', 18, 'bold')) intro_label1.place(x=self.left_canvas_x, y=10) intro_label2 = tk.Label(self.root, text="使用介紹: 請在左側框內上傳一張建築渲染圖", bg='black', fg='white', font=('微軟雅黑', 10)) intro_label2.place(x=self.left_canvas_x, y=45) def select_save_folder(self): folder_selected = filedialog.askdirectory() if folder_selected: self.folder_path = folder_selected print(f"Selected folder: {self.folder_path}") def start_timer(self): self.timer_running = True self.start_time = time.time() self.timer_label = tk.Label(self.root, text="生成時間：0秒", bg='black', fg='white', font=('微軟雅黑', 10)) self.timer_label.place(x=self.right_frame.winfo_x(), y=self.buttons_y, anchor='w') self.update_timer() def update_timer(self): if self.timer_running: elapsed_time = round(time.time() - self.start_time) self.timer_label.config(text=f"生成時間：{elapsed_time}秒") self.root.after(1000, self.update_timer) def stop_timer(self): self.timer_running = False elapsed_time = round(time.time() - self.start_time) self.timer_label.config(text=f"生成時間：{elapsed_time}秒") def upload_image(self): self.upload_btn.config(state='disabled') # 禁用上傳按鈕以防止多次上傳 filepath = filedialog.askopenfilename() if filepath: self.start_timer() # 開始計時 self.display_image(filepath, self.left_frame, 'image_label', self.left_canvas_x) threading.Thread(target=self.call_api_for_sketch, args=(filepath,)).start() # 在新執行緒中呼叫API def display_image(self, image_path, canvas, label_attr, canvas_x): img = Image.open(image_path) img.thumbnail((canvas.winfo_width(), canvas.winfo_height())) img_tk = ImageTk.PhotoImage(img) if hasattr(self, label_attr): getattr(self, label_attr).configure(image=img_tk) getattr(self, label_attr).image = img_tk else: label = tk.Label(self.root, image=img_tk, bg='black') setattr(self, label_attr, label) getattr(self, label_attr).image = img_tk label.place(x=canvas_x, rely=0.5, anchor='w') if label_attr == 'generated_image_label': self.stop_timer() def call_api_for_sketch(self, image_path): # 模擬一些處理時間 time.sleep(1) with Image.open(image_path) as img: encoded_image = encode_pil_to_base64(img) data = { "prompt": "<lora:CWG_archisketch_v1:1>,Building,pre sketch,masterpiece,best quality,featuring markers,(3D:0.7)", "negative_prompt": "blurry, lower quality, glossy finish,insufficient contrast", "init_images": [encoded_image], "steps": 30, "width": img.width, "height": img.height, "seed": random.randint(1, 10000000), "alwayson_scripts": { "ControlNet": { "args": [ { "enabled": "true", "pixel_perfect": "true", "module": "canny", "model": "control_v11p_sd15_canny_fp16 [b18e0966]", "weight": 1, "image": encoded_image }, { "enabled": "true", "pixel_perfect": "true", "module": "depth", "model": "control_v11f1p_sd15_depth_fp16 [4b72d323]", "weight": 1, "image": encoded_image } ] } } } url = "http://127.0.0.1:7860/sdapi/v1/txt2img" response = requests.post(url, json=data) if response.status_code == 200: response_json = response.json() saved_image_path = save_decoded_image(response_json['images'][0], self.folder_path, "sketched_image") self.root.after(0, self.display_image, saved_image_path, self.right_frame, 'generated_image_label', self.right_canvas_x) self.root.after(0, lambda: self.upload_btn.config( state='normal')) # Re-enable the upload button after processing else: print("Failed to get response from API, status code:", response.status_code) self.root.after(0, lambda: self.upload_btn.config( state='normal')) # Re-enable the upload button if the API call fails if __name__ == "__main__": root = tk.Tk() app = PhotoSketchGUI(root) root.mainloop()