pipi cucu
This commit is contained in:
@@ -74,8 +74,7 @@ class OpenAICompatProvider(AgentProvider):
|
||||
f"Total frames: {len(context.frames)}\n"
|
||||
)
|
||||
|
||||
# Include mentioned frames as images, fall back to last 3 frames
|
||||
frames_to_send = context.mentioned_frames or context.frames[-3:]
|
||||
frames_to_send = context.mentioned_frames
|
||||
|
||||
content: list[dict] = [{"type": "text", "text": ctx_text + message}]
|
||||
for frame in frames_to_send:
|
||||
|
||||
@@ -111,9 +111,14 @@ def extract_scene_frames(input_path, output_dir, scene_threshold=0.10,
|
||||
except ffmpeg.Error as e:
|
||||
# ffmpeg may exit non-zero on growing files (corrupt tail) but still
|
||||
# produce valid frames. Return the stderr for parsing anyway.
|
||||
log.debug("ffmpeg exited with error (may still have valid frames)")
|
||||
stdout = e.stdout or b""
|
||||
stderr = e.stderr or b""
|
||||
err_text = stderr.decode("utf-8", errors="replace")
|
||||
# Log the last meaningful line so we can see the real cause
|
||||
for line in reversed(err_text.splitlines()):
|
||||
if line.strip() and not line.startswith(" "):
|
||||
log.debug("ffmpeg scene error: %s", line.strip())
|
||||
break
|
||||
stdout = e.stdout or b""
|
||||
return stdout.decode("utf-8", errors="replace"), stderr.decode("utf-8", errors="replace")
|
||||
|
||||
|
||||
|
||||
177
cht/window.py
177
cht/window.py
@@ -331,6 +331,7 @@ class ChtWindow(Adw.ApplicationWindow):
|
||||
self._agent_output_view.set_cursor_visible(False)
|
||||
self._agent_output_view.set_left_margin(8)
|
||||
self._agent_output_view.set_right_margin(8)
|
||||
self._setup_md_tags(self._agent_output_view.get_buffer())
|
||||
|
||||
scroll = Gtk.ScrolledWindow()
|
||||
scroll.set_vexpand(True)
|
||||
@@ -392,25 +393,183 @@ class ChtWindow(Adw.ApplicationWindow):
|
||||
stream_mgr=self._stream_mgr,
|
||||
tracker=self._tracker,
|
||||
on_chunk=lambda chunk: GLib.idle_add(self._replace_thinking, chunk),
|
||||
on_done=lambda err: GLib.idle_add(
|
||||
self._append_agent_output,
|
||||
f"[Error: {err}]\n" if err else "\n"
|
||||
),
|
||||
on_done=lambda err: GLib.idle_add(self._on_agent_done, err),
|
||||
)
|
||||
self._thinking_replaced = False
|
||||
self._response_start_mark = None
|
||||
self._response_accum = []
|
||||
|
||||
def _replace_thinking(self, chunk: str):
|
||||
"""Replace the '…' placeholder with the first chunk, then append normally."""
|
||||
"""Replace the '…' placeholder with the first real chunk."""
|
||||
buf = self._agent_output_view.get_buffer()
|
||||
if not self._thinking_replaced:
|
||||
self._thinking_replaced = True
|
||||
buf = self._agent_output_view.get_buffer()
|
||||
# Remove the trailing '…\n' (3 chars)
|
||||
end = buf.get_end_iter()
|
||||
start = end.copy()
|
||||
start.backward_chars(2)
|
||||
start.backward_chars(2) # remove '…\n'
|
||||
buf.delete(start, end)
|
||||
# Mark where the response starts for later MD formatting
|
||||
self._response_start_mark = buf.create_mark(
|
||||
None, buf.get_end_iter(), left_gravity=True
|
||||
)
|
||||
self._response_accum.append(chunk)
|
||||
self._append_agent_output(chunk)
|
||||
|
||||
def _on_agent_done(self, err: str | None):
|
||||
if err:
|
||||
self._append_agent_output(f"[Error: {err}]\n")
|
||||
return
|
||||
# Re-render accumulated response with markdown formatting
|
||||
if self._response_start_mark and self._response_accum:
|
||||
buf = self._agent_output_view.get_buffer()
|
||||
start = buf.get_iter_at_mark(self._response_start_mark)
|
||||
end = buf.get_end_iter()
|
||||
buf.delete(start, end)
|
||||
self._render_md(buf, start, "".join(self._response_accum))
|
||||
buf.delete_mark(self._response_start_mark)
|
||||
self._append_agent_output("\n")
|
||||
|
||||
def _setup_md_tags(self, buf):
|
||||
buf.create_tag("h1", weight=700, scale=1.4)
|
||||
buf.create_tag("h2", weight=700, scale=1.2)
|
||||
buf.create_tag("h3", weight=700, scale=1.05)
|
||||
buf.create_tag("bold", weight=700)
|
||||
buf.create_tag("italic", style=2) # Pango.Style.ITALIC = 2
|
||||
buf.create_tag("code", family="monospace", background="#2a2a2a", foreground="#e8e8e8")
|
||||
buf.create_tag("codeblock", family="monospace", background="#1e1e1e",
|
||||
foreground="#e8e8e8", left_margin=16, right_margin=16,
|
||||
pixels_above_lines=4, pixels_below_lines=4)
|
||||
buf.create_tag("bullet", left_margin=16)
|
||||
|
||||
def _render_md(self, buf, insert_iter, text: str):
|
||||
"""Insert markdown-formatted text into buf at insert_iter using text tags."""
|
||||
import re
|
||||
lines = text.split("\n")
|
||||
i = 0
|
||||
while i < len(lines):
|
||||
line = lines[i]
|
||||
|
||||
# Fenced code block
|
||||
if line.startswith("```"):
|
||||
lang = line[3:].strip()
|
||||
block_lines = []
|
||||
i += 1
|
||||
while i < len(lines) and not lines[i].startswith("```"):
|
||||
block_lines.append(lines[i])
|
||||
i += 1
|
||||
code = "\n".join(block_lines) + "\n"
|
||||
block_mark = buf.create_mark(None, insert_iter, True)
|
||||
self._insert_highlighted_code(buf, insert_iter, code, lang)
|
||||
buf.apply_tag_by_name("codeblock", buf.get_iter_at_mark(block_mark), insert_iter)
|
||||
buf.delete_mark(block_mark)
|
||||
buf.insert(insert_iter, "\n")
|
||||
i += 1
|
||||
continue
|
||||
|
||||
# Headers
|
||||
header_match = re.match(r"^(#{1,3})\s+(.*)", line)
|
||||
if header_match:
|
||||
level = len(header_match.group(1))
|
||||
content = header_match.group(2)
|
||||
tag = f"h{level}"
|
||||
mark = buf.create_mark(None, insert_iter, True)
|
||||
buf.insert(insert_iter, content + "\n")
|
||||
buf.apply_tag_by_name(tag, buf.get_iter_at_mark(mark), insert_iter)
|
||||
i += 1
|
||||
continue
|
||||
|
||||
# Bullet points
|
||||
bullet_match = re.match(r"^[\-\*]\s+(.*)", line)
|
||||
if bullet_match:
|
||||
self._insert_inline(buf, insert_iter, "• " + bullet_match.group(1), "bullet")
|
||||
buf.insert(insert_iter, "\n")
|
||||
i += 1
|
||||
continue
|
||||
|
||||
# Inline formatting pass on normal lines
|
||||
self._insert_inline_line(buf, insert_iter, line)
|
||||
buf.insert(insert_iter, "\n")
|
||||
i += 1
|
||||
|
||||
def _insert_inline(self, buf, it, text: str, outer_tag: str | None = None):
|
||||
mark = buf.create_mark(None, it, True)
|
||||
self._insert_inline_line(buf, it, text)
|
||||
if outer_tag:
|
||||
buf.apply_tag_by_name(outer_tag, buf.get_iter_at_mark(mark), it)
|
||||
|
||||
def _insert_inline_line(self, buf, it, text: str):
|
||||
"""Insert text with inline bold/italic/code formatting."""
|
||||
import re
|
||||
pattern = re.compile(r"(\*\*(.+?)\*\*|__(.+?)__|"
|
||||
r"\*(.+?)\*|_(.+?)_|"
|
||||
r"`([^`]+?)`)")
|
||||
pos = 0
|
||||
for m in pattern.finditer(text):
|
||||
# Plain text before match
|
||||
if m.start() > pos:
|
||||
buf.insert(it, text[pos:m.start()])
|
||||
full = m.group(0)
|
||||
if full.startswith("**") or full.startswith("__"):
|
||||
inner = m.group(2) or m.group(3)
|
||||
mark = buf.create_mark(None, it, True)
|
||||
buf.insert(it, inner)
|
||||
buf.apply_tag_by_name("bold", buf.get_iter_at_mark(mark), it)
|
||||
elif full.startswith("*") or full.startswith("_"):
|
||||
inner = m.group(4) or m.group(5)
|
||||
mark = buf.create_mark(None, it, True)
|
||||
buf.insert(it, inner)
|
||||
buf.apply_tag_by_name("italic", buf.get_iter_at_mark(mark), it)
|
||||
elif full.startswith("`"):
|
||||
inner = m.group(6)
|
||||
mark = buf.create_mark(None, it, True)
|
||||
buf.insert(it, inner)
|
||||
buf.apply_tag_by_name("code", buf.get_iter_at_mark(mark), it)
|
||||
pos = m.end()
|
||||
if pos < len(text):
|
||||
buf.insert(it, text[pos:])
|
||||
|
||||
def _insert_highlighted_code(self, buf, it, code: str, lang: str):
|
||||
"""Insert syntax-highlighted code using Pygments token tags."""
|
||||
try:
|
||||
from pygments.lexers import get_lexer_by_name, guess_lexer
|
||||
from pygments.styles import get_style_by_name
|
||||
except ImportError:
|
||||
buf.insert(it, code)
|
||||
return
|
||||
|
||||
try:
|
||||
lexer = get_lexer_by_name(lang, stripall=False) if lang else guess_lexer(code)
|
||||
except Exception:
|
||||
try:
|
||||
lexer = guess_lexer(code)
|
||||
except Exception:
|
||||
buf.insert(it, code)
|
||||
return
|
||||
|
||||
try:
|
||||
style = get_style_by_name("monokai")
|
||||
except Exception:
|
||||
buf.insert(it, code)
|
||||
return
|
||||
|
||||
tag_table = buf.get_tag_table()
|
||||
for ttype, value in lexer.get_tokens(code):
|
||||
tag_name = f"pyg_{ttype}"
|
||||
tag = tag_table.lookup(tag_name)
|
||||
if tag is None:
|
||||
sf = style.style_for_token(ttype)
|
||||
tag = buf.create_tag(tag_name, family="monospace")
|
||||
if sf.get("color"):
|
||||
tag.set_property("foreground", f"#{sf['color']}")
|
||||
if sf.get("bold"):
|
||||
tag.set_property("weight", 700)
|
||||
if sf.get("italic"):
|
||||
tag.set_property("style", 2)
|
||||
mark = buf.create_mark(None, it, True)
|
||||
buf.insert(it, value)
|
||||
buf.apply_tag(tag, buf.get_iter_at_mark(mark), it)
|
||||
buf.delete_mark(mark)
|
||||
|
||||
def _check_agent_auth(self):
|
||||
import os
|
||||
if os.environ.get("GROQ_API_KEY") or os.environ.get("OPENAI_API_KEY"):
|
||||
@@ -480,7 +639,7 @@ class ChtWindow(Adw.ApplicationWindow):
|
||||
# Click to highlight — does NOT switch mode or seek
|
||||
# (future: jump to timestamp in scrub bar without leaving live)
|
||||
gesture = Gtk.GestureClick()
|
||||
gesture.connect("released", lambda g, n, x, y: log.debug("Frame clicked: %s at %.1fs", frame_id, timestamp))
|
||||
gesture.connect("released", lambda g, n, x, y, fid=frame_id: self._send_message(f"solve this @{fid}"))
|
||||
box.add_controller(gesture)
|
||||
|
||||
self._frames_strip.append(box)
|
||||
|
||||
Reference in New Issue
Block a user