# gui.py import tomllib import tomli_w import pygame import pygame_gui from pathlib import Path from tkinter import filedialog, Tk import threading import aggregate import gemini CONFIG_PATH = Path("config.toml") WINDOW_W, WINDOW_H = 1280, 720 def load_config() -> dict: with open(CONFIG_PATH, "rb") as f: return tomllib.load(f) def save_config(config: dict): with open(CONFIG_PATH, "wb") as f: tomli_w.dump(config, f) def hide_tk_root() -> Tk: root = Tk() root.withdraw() root.wm_attributes("-topmost", True) return root class App: def __init__(self): pygame.init() self.screen = pygame.display.set_mode((WINDOW_W, WINDOW_H), pygame.RESIZABLE) pygame.display.set_caption("manual slop") self.manager = pygame_gui.UIManager((WINDOW_W, WINDOW_H)) self.clock = pygame.time.Clock() self.config = load_config() self.files = list(self.config["files"].get("paths", [])) self.screenshots = list(self.config.get("screenshots", {}).get("paths", [])) self.history = list(self.config.get("discussion", {}).get("history", [])) self.gemini_status = "idle" self.gemini_response = "" self.last_md = "" self.last_md_path = None self.send_thread = None self.panels = {} self.elements = {} self._build_ui() def _col_rect(self, col: int, cols: int = 4, padding: int = 8) -> pygame.Rect: w, h = self.screen.get_size() col_w = (w - padding * (cols + 1)) // cols x = padding + col * (col_w + padding) return pygame.Rect(x, padding, col_w, h - padding * 2) def _build_ui(self): w, h = self.screen.get_size() p = 8 cols = 4 col_w = (w - p * (cols + 1)) // cols col_h = h - p * 2 def col_x(i): return p + i * (col_w + p) # --- CONFIG PANEL (col 0 top) --- config_h = 220 self.elements["lbl_config"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(0), p, col_w, 24), text="Config", manager=self.manager) self.elements["namespace"] = pygame_gui.elements.UITextEntryLine( relative_rect=pygame.Rect(col_x(0), p + 30, col_w, 32), manager=self.manager) self.elements["namespace"].set_text(self.config["output"]["namespace"]) self.elements["lbl_ns"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(0), p + 24, col_w, 24), text="Namespace", manager=self.manager) self.elements["output_dir"] = pygame_gui.elements.UITextEntryLine( relative_rect=pygame.Rect(col_x(0), p + 90, col_w, 32), manager=self.manager) self.elements["output_dir"].set_text(self.config["output"]["output_dir"]) self.elements["lbl_outdir"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(0), p + 66, col_w, 24), text="Output Dir", manager=self.manager) self.elements["btn_browse_output"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(0), p + 128, col_w, 30), text="Browse Output Dir", manager=self.manager) self.elements["btn_save_config"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(0), p + 164, col_w, 30), text="Save Config", manager=self.manager) # --- FILES PANEL (col 0 bottom) --- fy = p + config_h fh = col_h - config_h self.elements["lbl_files"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(0), fy, col_w, 24), text="Files", manager=self.manager) self.elements["files_base_dir"] = pygame_gui.elements.UITextEntryLine( relative_rect=pygame.Rect(col_x(0), fy + 30, col_w - 90, 32), manager=self.manager) self.elements["files_base_dir"].set_text(self.config["files"]["base_dir"]) self.elements["btn_browse_files_base"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(0) + col_w - 86, fy + 30, 86, 32), text="Browse", manager=self.manager) self.elements["files_list"] = pygame_gui.elements.UISelectionList( relative_rect=pygame.Rect(col_x(0), fy + 68, col_w, fh - 140), item_list=self.files, manager=self.manager) self.elements["btn_add_files"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(0), fy + fh - 68, col_w // 2 - 2, 30), text="Add File(s)", manager=self.manager) self.elements["btn_add_wildcard"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(0) + col_w // 2 + 2, fy + fh - 68, col_w // 2 - 2, 30), text="Add Wildcard", manager=self.manager) self.elements["btn_remove_file"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(0), fy + fh - 34, col_w, 30), text="Remove Selected File", manager=self.manager) # --- SCREENSHOTS PANEL (col 1) --- self.elements["lbl_shots"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(1), p, col_w, 24), text="Screenshots", manager=self.manager) self.elements["shots_base_dir"] = pygame_gui.elements.UITextEntryLine( relative_rect=pygame.Rect(col_x(1), p + 30, col_w - 90, 32), manager=self.manager) self.elements["shots_base_dir"].set_text( self.config.get("screenshots", {}).get("base_dir", ".")) self.elements["btn_browse_shots_base"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(1) + col_w - 86, p + 30, 86, 32), text="Browse", manager=self.manager) self.elements["shots_list"] = pygame_gui.elements.UISelectionList( relative_rect=pygame.Rect(col_x(1), p + 68, col_w, col_h - 140), item_list=self.screenshots, manager=self.manager) self.elements["btn_add_shots"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(1), p + col_h - 68, col_w, 30), text="Add Screenshot(s)", manager=self.manager) self.elements["btn_remove_shot"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(1), p + col_h - 34, col_w, 30), text="Remove Selected Screenshot", manager=self.manager) # --- DISCUSSION PANEL (col 2) --- self.elements["lbl_discussion"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(2), p, col_w, 24), text="Discussion History", manager=self.manager) self.elements["discussion_box"] = pygame_gui.elements.UITextEntryBox( relative_rect=pygame.Rect(col_x(2), p + 30, col_w, col_h - 100), manager=self.manager) self.elements["discussion_box"].set_text("\n---\n".join(self.history)) self.elements["btn_add_excerpt"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(2), p + col_h - 66, col_w // 2 - 2, 30), text="Add Separator", manager=self.manager) self.elements["btn_clear_discussion"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(2) + col_w // 2 + 2, p + col_h - 66, col_w // 2 - 2, 30), text="Clear", manager=self.manager) self.elements["btn_save_discussion"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(2), p + col_h - 32, col_w, 30), text="Save Discussion", manager=self.manager) # --- GEMINI PANEL (col 3) --- self.elements["lbl_gemini"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(3), p, col_w, 24), text="Gemini", manager=self.manager) self.elements["lbl_status"] = pygame_gui.elements.UILabel( relative_rect=pygame.Rect(col_x(3), p + 28, col_w, 24), text=f"Status: {self.gemini_status}", manager=self.manager) self.elements["gemini_input"] = pygame_gui.elements.UITextEntryBox( relative_rect=pygame.Rect(col_x(3), p + 56, col_w, 120), manager=self.manager) self.elements["btn_generate_send"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(3), p + 182, col_w // 3 - 2, 30), text="Gen + Send", manager=self.manager) self.elements["btn_md_only"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(3) + col_w // 3 + 2, p + 182, col_w // 3 - 2, 30), text="MD Only", manager=self.manager) self.elements["btn_reset_session"] = pygame_gui.elements.UIButton( relative_rect=pygame.Rect(col_x(3) + (col_w // 3) * 2 + 4, p + 182, col_w // 3 - 2, 30), text="Reset", manager=self.manager) self.elements["gemini_response"] = pygame_gui.elements.UITextBox( relative_rect=pygame.Rect(col_x(3), p + 218, col_w, col_h - 222), html_text="", manager=self.manager) def _rebuild_ui(self): for el in self.elements.values(): el.kill() self.elements.clear() self.manager.set_window_resolution(self.screen.get_size()) self._build_ui() def _refresh_files_list(self): self.elements["files_list"].set_item_list(self.files) def _refresh_shots_list(self): self.elements["shots_list"].set_item_list(self.screenshots) def _update_status(self, status: str): self.gemini_status = status self.elements["lbl_status"].set_text(f"Status: {status}") def _update_response(self, text: str): self.gemini_response = text self.elements["gemini_response"].set_text(text) def _flush_to_config(self): self.config["output"]["namespace"] = self.elements["namespace"].get_text() self.config["output"]["output_dir"] = self.elements["output_dir"].get_text() self.config["files"]["base_dir"] = self.elements["files_base_dir"].get_text() if "screenshots" not in self.config: self.config["screenshots"] = {} self.config["screenshots"]["base_dir"] = self.elements["shots_base_dir"].get_text() self.config["files"]["paths"] = self.files self.config["screenshots"]["paths"] = self.screenshots raw = self.elements["discussion_box"].get_text() self.history = [s.strip() for s in raw.split("---") if s.strip()] self.config["discussion"] = {"history": self.history} def _do_generate(self) -> tuple[str, Path]: self._flush_to_config() save_config(self.config) return aggregate.run(self.config) def handle_event(self, event: pygame.event.Event): if event.type == pygame.QUIT: return False if event.type == pygame.VIDEORESIZE: self._rebuild_ui() if event.type == pygame_gui.UI_BUTTON_PRESSED: el = event.ui_element if el == self.elements["btn_browse_output"]: root = hide_tk_root() d = filedialog.askdirectory(title="Select Output Dir") root.destroy() if d: self.elements["output_dir"].set_text(d) elif el == self.elements["btn_save_config"]: self._flush_to_config() save_config(self.config) elif el == self.elements["btn_browse_files_base"]: root = hide_tk_root() d = filedialog.askdirectory(title="Select Files Base Dir") root.destroy() if d: self.elements["files_base_dir"].set_text(d) elif el == self.elements["btn_add_files"]: root = hide_tk_root() paths = filedialog.askopenfilenames(title="Select Files") root.destroy() for p in paths: if p not in self.files: self.files.append(p) self._refresh_files_list() elif el == self.elements["btn_add_wildcard"]: root = hide_tk_root() d = filedialog.askdirectory(title="Select Dir for Wildcard") root.destroy() if d: self.files.append(str(Path(d) / "**" / "*")) self._refresh_files_list() elif el == self.elements["btn_remove_file"]: selected = self.elements["files_list"].get_single_selection() if selected and selected in self.files: self.files.remove(selected) self._refresh_files_list() elif el == self.elements["btn_browse_shots_base"]: root = hide_tk_root() d = filedialog.askdirectory(title="Select Screenshots Base Dir") root.destroy() if d: self.elements["shots_base_dir"].set_text(d) elif el == self.elements["btn_add_shots"]: root = hide_tk_root() paths = filedialog.askopenfilenames( title="Select Screenshots", filetypes=[("Images", "*.png *.jpg *.jpeg *.gif *.bmp *.webp"), ("All", "*.*")] ) root.destroy() for p in paths: if p not in self.screenshots: self.screenshots.append(p) self._refresh_shots_list() elif el == self.elements["btn_remove_shot"]: selected = self.elements["shots_list"].get_single_selection() if selected and selected in self.screenshots: self.screenshots.remove(selected) self._refresh_shots_list() elif el == self.elements["btn_add_excerpt"]: current = self.elements["discussion_box"].get_text() self.elements["discussion_box"].set_text(current + "\n---\n") elif el == self.elements["btn_clear_discussion"]: self.elements["discussion_box"].set_text("") elif el == self.elements["btn_save_discussion"]: self._flush_to_config() save_config(self.config) elif el == self.elements["btn_md_only"]: md, path = self._do_generate() self.last_md = md self.last_md_path = path self._update_status(f"md written: {path.name}") elif el == self.elements["btn_reset_session"]: gemini.reset_session() self._update_status("session reset") self._update_response("") elif el == self.elements["btn_generate_send"]: is_sending = self.send_thread is not None and self.send_thread.is_alive() if is_sending: return True md, path = self._do_generate() self.last_md = md self.last_md_path = path self._update_status("sending...") user_msg = self.elements["gemini_input"].get_text() def do_send(): try: response = gemini.send(self.last_md, user_msg) self._update_response(response) self._update_status("done") except Exception as e: self._update_response(f"ERROR: {e}") self._update_status("error") self.send_thread = threading.Thread(target=do_send, daemon=True) self.send_thread.start() return True def run(self): running = True while running: td = self.clock.tick(30) / 1000.0 for event in pygame.event.get(): self.manager.process_events(event) running = self.handle_event(event) self.manager.update(td) self.screen.fill((30, 30, 30)) self.manager.draw_ui(self.screen) pygame.display.flip() pygame.quit() import threading def main(): app = App() app.run() if __name__ == "__main__": main()