slop intensifies
This commit is contained in:
522
gui.py
522
gui.py
@@ -1,16 +1,14 @@
|
|||||||
# gui.py
|
# gui.py
|
||||||
|
import dearpygui.dearpygui as dpg
|
||||||
import tomllib
|
import tomllib
|
||||||
import tomli_w
|
import tomli_w
|
||||||
import pygame
|
import threading
|
||||||
import pygame_gui
|
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from tkinter import filedialog, Tk
|
from tkinter import filedialog, Tk
|
||||||
import threading
|
|
||||||
import aggregate
|
import aggregate
|
||||||
import gemini
|
import gemini
|
||||||
|
|
||||||
CONFIG_PATH = Path("config.toml")
|
CONFIG_PATH = Path("config.toml")
|
||||||
WINDOW_W, WINDOW_H = 1280, 720
|
|
||||||
|
|
||||||
def load_config() -> dict:
|
def load_config() -> dict:
|
||||||
with open(CONFIG_PATH, "rb") as f:
|
with open(CONFIG_PATH, "rb") as f:
|
||||||
@@ -28,220 +26,32 @@ def hide_tk_root() -> Tk:
|
|||||||
|
|
||||||
class App:
|
class App:
|
||||||
def __init__(self):
|
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.config = load_config()
|
||||||
self.files = list(self.config["files"].get("paths", []))
|
self.files: list[str] = list(self.config["files"].get("paths", []))
|
||||||
self.screenshots = list(self.config.get("screenshots", {}).get("paths", []))
|
self.screenshots: list[str] = list(self.config.get("screenshots", {}).get("paths", []))
|
||||||
self.history = list(self.config.get("discussion", {}).get("history", []))
|
self.history: list[str] = list(self.config.get("discussion", {}).get("history", []))
|
||||||
|
|
||||||
self.gemini_status = "idle"
|
self.gemini_status = "idle"
|
||||||
self.gemini_response = ""
|
self.gemini_response = ""
|
||||||
self.last_md = ""
|
self.last_md = ""
|
||||||
self.last_md_path = None
|
self.last_md_path: Path | None = None
|
||||||
self.send_thread = None
|
self.send_thread: threading.Thread | None = None
|
||||||
|
|
||||||
self.panels = {}
|
self.file_rows: list[str] = []
|
||||||
self.elements = {}
|
self.shot_rows: list[str] = []
|
||||||
self._build_ui()
|
|
||||||
|
|
||||||
def _col_rect(self, col: int, cols: int = 4, padding: int = 8) -> pygame.Rect:
|
# ------------------------------------------------------------------ helpers
|
||||||
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):
|
def _flush_to_config(self):
|
||||||
self.config["output"]["namespace"] = self.elements["namespace"].get_text()
|
self.config["output"]["namespace"] = dpg.get_value("namespace")
|
||||||
self.config["output"]["output_dir"] = self.elements["output_dir"].get_text()
|
self.config["output"]["output_dir"] = dpg.get_value("output_dir")
|
||||||
self.config["files"]["base_dir"] = self.elements["files_base_dir"].get_text()
|
self.config["files"]["base_dir"] = dpg.get_value("files_base_dir")
|
||||||
if "screenshots" not in self.config:
|
if "screenshots" not in self.config:
|
||||||
self.config["screenshots"] = {}
|
self.config["screenshots"] = {}
|
||||||
self.config["screenshots"]["base_dir"] = self.elements["shots_base_dir"].get_text()
|
self.config["screenshots"]["base_dir"] = dpg.get_value("shots_base_dir")
|
||||||
self.config["files"]["paths"] = self.files
|
self.config["files"]["paths"] = self.files
|
||||||
self.config["screenshots"]["paths"] = self.screenshots
|
self.config["screenshots"]["paths"] = self.screenshots
|
||||||
raw = self.elements["discussion_box"].get_text()
|
raw = dpg.get_value("discussion_box")
|
||||||
self.history = [s.strip() for s in raw.split("---") if s.strip()]
|
self.history = [s.strip() for s in raw.split("---") if s.strip()]
|
||||||
self.config["discussion"] = {"history": self.history}
|
self.config["discussion"] = {"history": self.history}
|
||||||
|
|
||||||
@@ -250,65 +60,96 @@ class App:
|
|||||||
save_config(self.config)
|
save_config(self.config)
|
||||||
return aggregate.run(self.config)
|
return aggregate.run(self.config)
|
||||||
|
|
||||||
def handle_event(self, event: pygame.event.Event):
|
def _update_status(self, status: str):
|
||||||
if event.type == pygame.QUIT:
|
self.gemini_status = status
|
||||||
return False
|
dpg.set_value("gemini_status", f"Status: {status}")
|
||||||
|
|
||||||
if event.type == pygame.VIDEORESIZE:
|
def _update_response(self, text: str):
|
||||||
self._rebuild_ui()
|
self.gemini_response = text
|
||||||
|
dpg.set_value("gemini_response", text)
|
||||||
|
|
||||||
if event.type == pygame_gui.UI_BUTTON_PRESSED:
|
def _rebuild_files_table(self):
|
||||||
el = event.ui_element
|
dpg.delete_item("files_table", children_only=True)
|
||||||
|
for i, f in enumerate(self.files):
|
||||||
|
with dpg.table_row(parent="files_table"):
|
||||||
|
dpg.add_text(f)
|
||||||
|
dpg.add_button(
|
||||||
|
label="x",
|
||||||
|
callback=self._make_remove_file_cb(i),
|
||||||
|
width=24
|
||||||
|
)
|
||||||
|
|
||||||
if el == self.elements["btn_browse_output"]:
|
def _rebuild_shots_table(self):
|
||||||
|
dpg.delete_item("shots_table", children_only=True)
|
||||||
|
for i, s in enumerate(self.screenshots):
|
||||||
|
with dpg.table_row(parent="shots_table"):
|
||||||
|
dpg.add_text(s)
|
||||||
|
dpg.add_button(
|
||||||
|
label="x",
|
||||||
|
callback=self._make_remove_shot_cb(i),
|
||||||
|
width=24
|
||||||
|
)
|
||||||
|
|
||||||
|
def _make_remove_file_cb(self, idx: int):
|
||||||
|
def cb():
|
||||||
|
if idx < len(self.files):
|
||||||
|
self.files.pop(idx)
|
||||||
|
self._rebuild_files_table()
|
||||||
|
return cb
|
||||||
|
|
||||||
|
def _make_remove_shot_cb(self, idx: int):
|
||||||
|
def cb():
|
||||||
|
if idx < len(self.screenshots):
|
||||||
|
self.screenshots.pop(idx)
|
||||||
|
self._rebuild_shots_table()
|
||||||
|
return cb
|
||||||
|
|
||||||
|
# ---------------------------------------------------------------- callbacks
|
||||||
|
|
||||||
|
def cb_browse_output(self):
|
||||||
root = hide_tk_root()
|
root = hide_tk_root()
|
||||||
d = filedialog.askdirectory(title="Select Output Dir")
|
d = filedialog.askdirectory(title="Select Output Dir")
|
||||||
root.destroy()
|
root.destroy()
|
||||||
if d:
|
if d:
|
||||||
self.elements["output_dir"].set_text(d)
|
dpg.set_value("output_dir", d)
|
||||||
|
|
||||||
elif el == self.elements["btn_save_config"]:
|
def cb_save_config(self):
|
||||||
self._flush_to_config()
|
self._flush_to_config()
|
||||||
save_config(self.config)
|
save_config(self.config)
|
||||||
|
self._update_status("config saved")
|
||||||
|
|
||||||
elif el == self.elements["btn_browse_files_base"]:
|
def cb_browse_files_base(self):
|
||||||
root = hide_tk_root()
|
root = hide_tk_root()
|
||||||
d = filedialog.askdirectory(title="Select Files Base Dir")
|
d = filedialog.askdirectory(title="Select Files Base Dir")
|
||||||
root.destroy()
|
root.destroy()
|
||||||
if d:
|
if d:
|
||||||
self.elements["files_base_dir"].set_text(d)
|
dpg.set_value("files_base_dir", d)
|
||||||
|
|
||||||
elif el == self.elements["btn_add_files"]:
|
def cb_add_files(self):
|
||||||
root = hide_tk_root()
|
root = hide_tk_root()
|
||||||
paths = filedialog.askopenfilenames(title="Select Files")
|
paths = filedialog.askopenfilenames(title="Select Files")
|
||||||
root.destroy()
|
root.destroy()
|
||||||
for p in paths:
|
for p in paths:
|
||||||
if p not in self.files:
|
if p not in self.files:
|
||||||
self.files.append(p)
|
self.files.append(p)
|
||||||
self._refresh_files_list()
|
self._rebuild_files_table()
|
||||||
|
|
||||||
elif el == self.elements["btn_add_wildcard"]:
|
def cb_add_wildcard(self):
|
||||||
root = hide_tk_root()
|
root = hide_tk_root()
|
||||||
d = filedialog.askdirectory(title="Select Dir for Wildcard")
|
d = filedialog.askdirectory(title="Select Dir for Wildcard")
|
||||||
root.destroy()
|
root.destroy()
|
||||||
if d:
|
if d:
|
||||||
self.files.append(str(Path(d) / "**" / "*"))
|
self.files.append(str(Path(d) / "**" / "*"))
|
||||||
self._refresh_files_list()
|
self._rebuild_files_table()
|
||||||
|
|
||||||
elif el == self.elements["btn_remove_file"]:
|
def cb_browse_shots_base(self):
|
||||||
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()
|
root = hide_tk_root()
|
||||||
d = filedialog.askdirectory(title="Select Screenshots Base Dir")
|
d = filedialog.askdirectory(title="Select Screenshots Base Dir")
|
||||||
root.destroy()
|
root.destroy()
|
||||||
if d:
|
if d:
|
||||||
self.elements["shots_base_dir"].set_text(d)
|
dpg.set_value("shots_base_dir", d)
|
||||||
|
|
||||||
elif el == self.elements["btn_add_shots"]:
|
def cb_add_shots(self):
|
||||||
root = hide_tk_root()
|
root = hide_tk_root()
|
||||||
paths = filedialog.askopenfilenames(
|
paths = filedialog.askopenfilenames(
|
||||||
title="Select Screenshots",
|
title="Select Screenshots",
|
||||||
@@ -318,45 +159,47 @@ class App:
|
|||||||
for p in paths:
|
for p in paths:
|
||||||
if p not in self.screenshots:
|
if p not in self.screenshots:
|
||||||
self.screenshots.append(p)
|
self.screenshots.append(p)
|
||||||
self._refresh_shots_list()
|
self._rebuild_shots_table()
|
||||||
|
|
||||||
elif el == self.elements["btn_remove_shot"]:
|
def cb_add_excerpt(self):
|
||||||
selected = self.elements["shots_list"].get_single_selection()
|
current = dpg.get_value("discussion_box")
|
||||||
if selected and selected in self.screenshots:
|
dpg.set_value("discussion_box", current + "\n---\n")
|
||||||
self.screenshots.remove(selected)
|
|
||||||
self._refresh_shots_list()
|
|
||||||
|
|
||||||
elif el == self.elements["btn_add_excerpt"]:
|
def cb_clear_discussion(self):
|
||||||
current = self.elements["discussion_box"].get_text()
|
dpg.set_value("discussion_box", "")
|
||||||
self.elements["discussion_box"].set_text(current + "\n---\n")
|
|
||||||
|
|
||||||
elif el == self.elements["btn_clear_discussion"]:
|
def cb_save_discussion(self):
|
||||||
self.elements["discussion_box"].set_text("")
|
|
||||||
|
|
||||||
elif el == self.elements["btn_save_discussion"]:
|
|
||||||
self._flush_to_config()
|
self._flush_to_config()
|
||||||
save_config(self.config)
|
save_config(self.config)
|
||||||
|
self._update_status("discussion saved")
|
||||||
|
|
||||||
elif el == self.elements["btn_md_only"]:
|
def cb_md_only(self):
|
||||||
|
try:
|
||||||
md, path = self._do_generate()
|
md, path = self._do_generate()
|
||||||
self.last_md = md
|
self.last_md = md
|
||||||
self.last_md_path = path
|
self.last_md_path = path
|
||||||
self._update_status(f"md written: {path.name}")
|
self._update_status(f"md written: {path.name}")
|
||||||
|
except Exception as e:
|
||||||
|
self._update_status(f"error: {e}")
|
||||||
|
|
||||||
elif el == self.elements["btn_reset_session"]:
|
def cb_reset_session(self):
|
||||||
gemini.reset_session()
|
gemini.reset_session()
|
||||||
self._update_status("session reset")
|
self._update_status("session reset")
|
||||||
self._update_response("")
|
self._update_response("")
|
||||||
|
|
||||||
elif el == self.elements["btn_generate_send"]:
|
def cb_generate_send(self):
|
||||||
is_sending = self.send_thread is not None and self.send_thread.is_alive()
|
if self.send_thread and self.send_thread.is_alive():
|
||||||
if is_sending:
|
return
|
||||||
return True
|
try:
|
||||||
md, path = self._do_generate()
|
md, path = self._do_generate()
|
||||||
self.last_md = md
|
self.last_md = md
|
||||||
self.last_md_path = path
|
self.last_md_path = path
|
||||||
|
except Exception as e:
|
||||||
|
self._update_status(f"generate error: {e}")
|
||||||
|
return
|
||||||
|
|
||||||
self._update_status("sending...")
|
self._update_status("sending...")
|
||||||
user_msg = self.elements["gemini_input"].get_text()
|
user_msg = dpg.get_value("gemini_input")
|
||||||
|
|
||||||
def do_send():
|
def do_send():
|
||||||
try:
|
try:
|
||||||
@@ -370,25 +213,178 @@ class App:
|
|||||||
self.send_thread = threading.Thread(target=do_send, daemon=True)
|
self.send_thread = threading.Thread(target=do_send, daemon=True)
|
||||||
self.send_thread.start()
|
self.send_thread.start()
|
||||||
|
|
||||||
return True
|
# ---------------------------------------------------------------- build ui
|
||||||
|
|
||||||
|
def _build_ui(self):
|
||||||
|
with dpg.window(
|
||||||
|
label="Config",
|
||||||
|
tag="win_config",
|
||||||
|
pos=(8, 8),
|
||||||
|
width=340,
|
||||||
|
height=220,
|
||||||
|
no_close=True
|
||||||
|
):
|
||||||
|
dpg.add_input_text(
|
||||||
|
label="Namespace",
|
||||||
|
tag="namespace",
|
||||||
|
default_value=self.config["output"]["namespace"],
|
||||||
|
width=-1
|
||||||
|
)
|
||||||
|
dpg.add_input_text(
|
||||||
|
label="Output Dir",
|
||||||
|
tag="output_dir",
|
||||||
|
default_value=self.config["output"]["output_dir"],
|
||||||
|
width=-1
|
||||||
|
)
|
||||||
|
dpg.add_button(label="Browse Output Dir", callback=self.cb_browse_output, width=-1)
|
||||||
|
dpg.add_separator()
|
||||||
|
dpg.add_button(label="Save Config", callback=self.cb_save_config, width=-1)
|
||||||
|
|
||||||
|
with dpg.window(
|
||||||
|
label="Files",
|
||||||
|
tag="win_files",
|
||||||
|
pos=(8, 236),
|
||||||
|
width=340,
|
||||||
|
height=460,
|
||||||
|
no_close=True
|
||||||
|
):
|
||||||
|
dpg.add_input_text(
|
||||||
|
label="Base Dir",
|
||||||
|
tag="files_base_dir",
|
||||||
|
default_value=self.config["files"]["base_dir"],
|
||||||
|
width=-1
|
||||||
|
)
|
||||||
|
dpg.add_button(label="Browse Base Dir##files", callback=self.cb_browse_files_base, width=-1)
|
||||||
|
dpg.add_separator()
|
||||||
|
|
||||||
|
with dpg.table(
|
||||||
|
tag="files_table",
|
||||||
|
header_row=False,
|
||||||
|
resizable=True,
|
||||||
|
borders_innerV=True,
|
||||||
|
scrollY=True,
|
||||||
|
height=280
|
||||||
|
):
|
||||||
|
dpg.add_table_column(label="Path", width_stretch=True)
|
||||||
|
dpg.add_table_column(label="", width_fixed=True, init_width_or_weight=28)
|
||||||
|
|
||||||
|
self._rebuild_files_table()
|
||||||
|
|
||||||
|
dpg.add_separator()
|
||||||
|
with dpg.group(horizontal=True):
|
||||||
|
dpg.add_button(label="Add File(s)", callback=self.cb_add_files, width=-1)
|
||||||
|
with dpg.group(horizontal=True):
|
||||||
|
dpg.add_button(label="Add Wildcard", callback=self.cb_add_wildcard, width=-1)
|
||||||
|
|
||||||
|
with dpg.window(
|
||||||
|
label="Screenshots",
|
||||||
|
tag="win_screenshots",
|
||||||
|
pos=(356, 8),
|
||||||
|
width=340,
|
||||||
|
height=460,
|
||||||
|
no_close=True
|
||||||
|
):
|
||||||
|
dpg.add_input_text(
|
||||||
|
label="Base Dir",
|
||||||
|
tag="shots_base_dir",
|
||||||
|
default_value=self.config.get("screenshots", {}).get("base_dir", "."),
|
||||||
|
width=-1
|
||||||
|
)
|
||||||
|
dpg.add_button(label="Browse Base Dir##shots", callback=self.cb_browse_shots_base, width=-1)
|
||||||
|
dpg.add_separator()
|
||||||
|
|
||||||
|
with dpg.table(
|
||||||
|
tag="shots_table",
|
||||||
|
header_row=False,
|
||||||
|
resizable=True,
|
||||||
|
borders_innerV=True,
|
||||||
|
scrollY=True,
|
||||||
|
height=280
|
||||||
|
):
|
||||||
|
dpg.add_table_column(label="Path", width_stretch=True)
|
||||||
|
dpg.add_table_column(label="", width_fixed=True, init_width_or_weight=28)
|
||||||
|
|
||||||
|
self._rebuild_shots_table()
|
||||||
|
|
||||||
|
dpg.add_separator()
|
||||||
|
dpg.add_button(label="Add Screenshot(s)", callback=self.cb_add_shots, width=-1)
|
||||||
|
|
||||||
|
with dpg.window(
|
||||||
|
label="Discussion History",
|
||||||
|
tag="win_discussion",
|
||||||
|
pos=(704, 8),
|
||||||
|
width=340,
|
||||||
|
height=460,
|
||||||
|
no_close=True
|
||||||
|
):
|
||||||
|
dpg.add_input_text(
|
||||||
|
label="##discussion_box",
|
||||||
|
tag="discussion_box",
|
||||||
|
default_value="\n---\n".join(self.history),
|
||||||
|
multiline=True,
|
||||||
|
width=-1,
|
||||||
|
height=340
|
||||||
|
)
|
||||||
|
dpg.add_separator()
|
||||||
|
with dpg.group(horizontal=True):
|
||||||
|
dpg.add_button(label="Add Separator", callback=self.cb_add_excerpt)
|
||||||
|
dpg.add_button(label="Clear", callback=self.cb_clear_discussion)
|
||||||
|
dpg.add_button(label="Save", callback=self.cb_save_discussion)
|
||||||
|
|
||||||
|
with dpg.window(
|
||||||
|
label="Gemini",
|
||||||
|
tag="win_gemini",
|
||||||
|
pos=(1052, 8),
|
||||||
|
width=340,
|
||||||
|
height=700,
|
||||||
|
no_close=True
|
||||||
|
):
|
||||||
|
dpg.add_text("Status: idle", tag="gemini_status")
|
||||||
|
dpg.add_separator()
|
||||||
|
dpg.add_input_text(
|
||||||
|
label="##gemini_input",
|
||||||
|
tag="gemini_input",
|
||||||
|
multiline=True,
|
||||||
|
width=-1,
|
||||||
|
height=120
|
||||||
|
)
|
||||||
|
dpg.add_separator()
|
||||||
|
with dpg.group(horizontal=True):
|
||||||
|
dpg.add_button(label="Gen + Send", callback=self.cb_generate_send)
|
||||||
|
dpg.add_button(label="MD Only", callback=self.cb_md_only)
|
||||||
|
dpg.add_button(label="Reset", callback=self.cb_reset_session)
|
||||||
|
dpg.add_separator()
|
||||||
|
dpg.add_text("Response:")
|
||||||
|
dpg.add_input_text(
|
||||||
|
label="##gemini_response",
|
||||||
|
tag="gemini_response",
|
||||||
|
multiline=True,
|
||||||
|
readonly=True,
|
||||||
|
width=-1,
|
||||||
|
height=-1
|
||||||
|
)
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
running = True
|
dpg.create_context()
|
||||||
while running:
|
dpg.configure_app(docking=True, docking_space=True)
|
||||||
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)
|
dpg.create_viewport(
|
||||||
self.screen.fill((30, 30, 30))
|
title="manual slop",
|
||||||
self.manager.draw_ui(self.screen)
|
width=1600,
|
||||||
pygame.display.flip()
|
height=900
|
||||||
|
)
|
||||||
|
|
||||||
pygame.quit()
|
dpg.setup_dearpygui()
|
||||||
|
dpg.show_viewport()
|
||||||
|
dpg.maximize_viewport()
|
||||||
|
|
||||||
|
self._build_ui()
|
||||||
|
|
||||||
|
while dpg.is_dearpygui_running():
|
||||||
|
dpg.render_dearpygui_frame()
|
||||||
|
|
||||||
|
dpg.destroy_context()
|
||||||
|
|
||||||
import threading
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
app = App()
|
app = App()
|
||||||
|
|||||||
Reference in New Issue
Block a user