import tkinter as tk from tkinter import filedialog, messagebox, ttk,Canvas import os import numpy as np from PIL import Image, ImageTk import vlc import subprocess import functools import re # Global variables video_files = [] # List of video files current_video_idx = 0 # Index of the current video current_video_path = "" video_player = None thumbnail_images = [] split_info = [] # Create the main window root = tk.Tk() root.title("Video Processing Tool") root.state('zoomed') # Fullscreen screen_width = root.winfo_screenwidth() screen_height = root.winfo_screenheight() # Video playing frame video_frame = tk.Frame(root, width=1080, height=1920) video_frame.pack(side=tk.LEFT, padx=0) # Frame for thumbnail images thumbnail_frame = tk.Frame(root, width=1080, height=1920) thumbnail_frame.pack_forget() # Frame for right-side info and buttons control_frame = tk.Frame(root, width=100, height=200, bg='yellow') control_frame.pack(side=tk.RIGHT, anchor="ne", padx=5, pady=5) # 线程函数来显示视频 def on_playback_ended(event): # player.play() video_frame.pack_forget() # 如果使用 pack() global current_video_path display_thumbnails(current_video_path) def toggle_play_pause(): if player.is_playing(): player.pause() # 暂停播放 play_pause_button.config(text="播放视频") # 更新按钮文本为播放 else: player.play() # 播放 play_pause_button.config(text="暂停视频") # 更新按钮文本为暂停 # 更新进度条并增加调试信息 def update_progress(): # 确保播放器正在播放 if player.is_playing(): current_time = player.get_time() / 1000 # 毫秒转为秒 total_time = player.get_length() / 1000 # 毫秒转为秒 if total_time > 0: progress_percentage = (current_time / total_time) * 100 progress_var.set(progress_percentage) progress_bar.after(50, update_progress) else: progress_bar.after(50, update_progress) def get_thumbnail_frames(video_file): # 获取同名文件夹路径 base_dir = os.path.dirname(video_file) video_name = os.path.splitext(os.path.basename(video_file))[0] thumbnail_dir = os.path.join(base_dir, video_name) # 如果文件夹不存在，调用 bat 文件处理 if not os.path.exists(thumbnail_dir): print(f"文件夹 {thumbnail_dir} 不存在，调用 bat 文件进行处理...") subprocess.run(['mp4_generate_thumbnails.bat', video_file], check=True) # 重新获取文件夹内的所有 jpg 文件 jpg_files = [os.path.join(thumbnail_dir, f) for f in os.listdir( thumbnail_dir) if f.lower().endswith('.jpg')] jpg_files.sort() # 按照文件名顺序排序 return jpg_files selected_labels = [] # 记录当前选中的标签 thumbnails = [] def display_thumbnails(video_file): global thumbnails, selected_labels thumbnails = get_thumbnail_frames(video_file) row, col = 0, 0 max_col = (screen_width - 100) // 90 - 1 # 动态地 # 创建Canvas和垂直滚动条 thumbnail_frame.pack(side=tk.LEFT, padx=0) canvas = tk.Canvas(thumbnail_frame, width=screen_width - 100, height=screen_height) canvas.grid(row=0, column=0, sticky='nsew') scrollbar = ttk.Scrollbar(thumbnail_frame, orient="vertical", command=canvas.yview) scrollbar.grid(row=0, column=1, sticky='ns') canvas.configure(yscrollcommand=scrollbar.set) # 创建一个frame作为Canvas的子项来添加缩略图 thumbnail_container = tk.Frame(canvas) canvas.create_window((0, 0), window=thumbnail_container, anchor="nw") # 启用鼠标滚轮事件来滚动Canvas def on_mouse_wheel(event): """处理鼠标滚轮事件""" canvas.yview_scroll(int(-1 * (event.delta / 120)), "units") # 绑定鼠标滚轮事件 canvas.bind_all("", on_mouse_wheel) # 添加缩略图 for idx, thumb in enumerate(thumbnails): # 使用PIL加载图片 pil_image = Image.open(thumb) # 转换为Tkinter可显示的格式 photo = ImageTk.PhotoImage(pil_image) # 创建一个标签显示图像 thumb_label = tk.Label(thumbnail_container, image=photo) thumb_label.image = photo # 必须保持对图像的引用 thumb_label.grid(row=row, column=col, padx=1, pady=1) # 将toggle_checkmark的调用用 functools.partial包装，传递idx thumb_label.bind("", functools.partial(toggle_checkmark, label=thumb_label, idx=idx)) checkmark_label = None if idx == 0: # 默认第一个图片加对号 checkmark_label = add_checkmark_icon(thumb_label, pil_image) selected_labels.append(idx) # 将checkmark_label保存在标签中，后续点击时可以参考 thumb_label.checkmark_label = checkmark_label # 更新列数和行数 col += 1 if col >= max_col: col = 0 row += 1 # 更新Canvas的滚动区域 thumbnail_container.update_idletasks() canvas.config(scrollregion=canvas.bbox("all")) def toggle_checkmark(event, label, idx): """切换勾选状态""" checkmark_label = label.checkmark_label # 如果已经有勾选图标，移除它；如果没有，添加它 if checkmark_label: # 移除勾选图标 if idx == 0 : return checkmark_label.destroy() label.checkmark_label = None # 从选中的标签列表中移除 if idx in selected_labels: selected_labels.remove(idx) else: # 添加勾选图标 if idx ==len(thumbnails)-1: return pil_image = Image.open(thumbnails[idx]) checkmark_label = add_checkmark_icon(label, pil_image) label.checkmark_label = checkmark_label selected_labels.append(idx) #print(selected_labels) def add_checkmark_icon(thumb_label, pil_image): """在图像上添加勾选图标""" checkmark = Image.open("checkmark.png") # 假设你有一个勾选图标 checkmark = checkmark.resize((30, 30)) # 调整图标大小 checkmark_photo = ImageTk.PhotoImage(checkmark) # 创建一个标签来放置勾选图标 checkmark_label = tk.Label(thumb_label, image=checkmark_photo, bg='white', width=30, height=30) checkmark_label.image = checkmark_photo # 保持图标的引用 # 将勾选图标放在缩略图右上角 checkmark_label.place(x=pil_image.width - 15, y=5, anchor="ne") # 确保图标位置在右上角 return checkmark_label def show_large_image(image): # 创建一个新的窗口来显示大图 top = tk.Toplevel(root) top.title("Large Image") # 使用 PIL 创建 Image 对象 image_pil = Image.fromarray(image) image_tk = ImageTk.PhotoImage(image_pil) # 在新窗口中显示图片 label = tk.Label(top, image=image_tk) label.image = image_tk # 保持对图片的引用 label.pack() # 跳转到指定进度 def on_progress_change(event): total_time = player.get_length() / 1000 # 视频总时长（秒） new_time = (event.x / progress_bar.winfo_width()) * total_time # 计算新的时间 player.set_time(int(new_time * 1000)) # 设置新时间（毫秒） player = vlc.MediaPlayer() player.event_manager().event_attach( vlc.EventType.MediaPlayerEndReached, on_playback_ended) canvas = Canvas(video_frame, width=1080, height=1920) # 设置Canvas大小 canvas.pack(side=tk.BOTTOM, pady=10) # Video player area under progress bar # 初始化 VLC 播放器 # Windows 使用 set_hwnd(embed.winfo_id())，Linux 使用 set_xwindow(embed.winfo_id()) player.set_hwnd(canvas.winfo_id()) # 视频进度条 progress_var = tk.DoubleVar() progress_bar = ttk.Scale(video_frame, variable=progress_var, from_=0, to=100, orient="horizontal", length=800) progress_bar.pack(side=tk.BOTTOM, pady=10) # 绑定进度条点击事件，跳转到指定时间 progress_bar.bind("", on_progress_change) update_progress() # 设置主窗口 # Functions output_dir = "" def open_folder(): global output_dir """Open folder and get all MP4 files""" folder = filedialog.askdirectory() if not folder: return global video_files video_files = [] for root_dir, dirs, files in os.walk(folder): for file in files: if file.lower().endswith(".mp4"): video_files.append(os.path.join(root_dir, file)) if video_files: messagebox.showinfo("Info", f"文件数量：{len(video_files)}") root.title("文件夹："+folder+ ",文件数量："+str(len(video_files))) root.update() # 刷新界面 load_video(video_files[0]) output_dir = os.path.join(folder, "ProcessResults") create_folders(output_dir) else: messagebox.showerror("Error", "没有找到MP4文件") def load_video(video_path): """Load video into the player and start playback""" global current_video_path, video_player current_video_path = video_path thumbnail_frame.pack_forget() video_frame.pack(side=tk.LEFT, padx=0) player.set_media(vlc.Media(current_video_path)) player.play() def delete_video(): """Delete current video and play the next""" global current_video_idx update_title_minus_one() if current_video_idx < len(video_files): os.remove(video_files[current_video_idx]) current_video_idx += 1 if current_video_idx < len(video_files): load_video(video_files[current_video_idx]) else: messagebox.showinfo("Info", "处理完毕！") def split_video(): """Delete current video and play the next""" global current_video_idx update_title_minus_one() if current_video_idx < len(video_files): target_file = video_files[current_video_idx] save_hook_= save_hook.get() == 1 save_music_= save_music.get() == 1 cut_video_by_frames(save_hook_,save_music_,target_file,selected_labels,len(thumbnails),output_dir) current_video_idx += 1 if current_video_idx < len(video_files): load_video(video_files[current_video_idx]) else: messagebox.showinfo("Info", "处理完毕！") def update_title_minus_one(): current_title = root.title() parts = current_title.rsplit("：", 1) if len(parts) == 2 and parts[1].isdigit(): new_count = int(parts[1]) - 1 new_title = f"{parts[0]}：{new_count}" root.title(new_title) root.update() def get_filename_without_extension(video_path): # 获取文件名，去除路径 filename_with_extension = os.path.basename(video_path) # 去除扩展名 filename_without_extension = os.path.splitext(filename_with_extension)[0] return filename_without_extension def cut_video_by_frames(save_hook,save_music,video_path, frame_segments, total_frames, output_dir): """ 将视频按照指定的帧切片进行切割。参数： - video_path: 输入视频的路径 - frame_segments: 切片的帧序号数组，例如：[0, M1, M2, ..., Mlast] - output_dir: 输出视频文件的目录 """ fps = 30 # default fps frame_segments.sort() # 获取视频的帧率 (fps) command = [ "ffmpeg", "-i", video_path, "-vf", "fps=1", "-f", "null", "-" ] result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, encoding='utf-8') if result.returncode != 0: print(f"Error occurred while executing the command: {result.stderr}") else: output = result.stderr if output: fps_line = next(line for line in output.splitlines() if 'fps' in line) # 使用正则表达式提取帧率值（假设fps值总是出现在"fps"前面） match = re.search(r'(\d+(\.\d+)?)\s*fps', fps_line) if match: fps = float(match.group(1)) # 提取并转换为 float print(f"FPS: {fps}") else: print("No output from ffmpeg") print(f"视频的总帧数: {total_frames}, 帧率: {fps}") file_name_witout_extension = get_filename_without_extension(video_path) if save_music: output_file = os.path.join(os.path.join(output_dir,"music"), f"{file_name_witout_extension}.wav") command = ["ffmpeg", "-i", video_path, "-vn", "-acodec", "pcm_s16le","-ar", "44100","-ac", "2", output_file] subprocess.Popen(command) # 遍历切片区间 for i in range(len(frame_segments) - 1): start_frame = frame_segments[i] end_frame = frame_segments[i + 1] start_time = float(start_frame) / fps end_time = float(end_frame) / fps if save_hook and i==0: output_file = os.path.join(os.path.join(output_dir,"hook"), f"{file_name_witout_extension}.mp4") else: output_file = os.path.join(os.path.join(output_dir,"slice"), f"{file_name_witout_extension}_{i + 1}.mp4") # 使用 FFmpeg 命令来切割视频 command = [ "ffmpeg", "-i", video_path, "-ss", f"{start_time:.10f}", "-to", f"{end_time:.10f}", "-c:v", "libx264", "-crf", "18","-preset", "veryslow", output_file ] print(" ".join(command)) subprocess.Popen(command) # 最后一段处理（从最后一个切片到视频结束） last_start_frame = frame_segments[-1] last_start_time = last_start_frame / fps output_file = os.path.join( os.path.join(output_dir,"slice"), f"{file_name_witout_extension}_{len(frame_segments)}.mp4") #print(f"正在切割最后一段：从 {last_start_time:.10f}s 到视频结束，保存为 {output_file}") command = [ "ffmpeg", "-i", video_path, "-ss", f"{last_start_time:.10f}", "-c:v", "libx264","-crf", "18","-preset", "veryslow", output_file ] print(" ".join(command)) subprocess.Popen(command) def create_folders(base_folder): """Create a folder structure for saving results""" hook_folder = os.path.join(base_folder, "hook") music_folder = os.path.join(base_folder, "music") category_folder = os.path.join(base_folder, "slice") if not os.path.exists(hook_folder): os.makedirs(hook_folder) if not os.path.exists(music_folder): os.makedirs(music_folder) if not os.path.exists(category_folder): os.makedirs(category_folder) # Buttons open_folder_button = tk.Button(control_frame, text="打开文件夹", command=open_folder) open_folder_button.pack() # 播放控制按钮（播放/暂停） play_pause_button = tk.Button(control_frame, text="暂停视频", command=toggle_play_pause) play_pause_button.pack() delete_button = tk.Button(control_frame, text="不保留", command=delete_video) delete_button.pack() save_hook = tk.IntVar(value=1) # 默认勾选 save_music = tk.IntVar(value=1) # 默认勾选 # 创建两个 Checkbutton checkbox1 = tk.Checkbutton(control_frame, text="保存Hook视频", variable=save_hook) checkbox1.pack() checkbox2 = tk.Checkbutton(control_frame, text="保存音频", variable=save_music) checkbox2.pack() delete_button = tk.Button(control_frame, text="拆分", command=split_video) delete_button.pack() menu_bar = tk.Menu(root) file_menu = tk.Menu(menu_bar, tearoff=0) file_menu.add_command(label="打开文件夹", command=open_folder) menu_bar.add_cascade(label="文件", menu=file_menu) root.config(menu=menu_bar) #video_frame.pack_forget() #display_thumbnails("E:\抖音-AI风景\新建文件夹\_demoold") # Start the Tkinter loop root.mainloop()