pi-studio 0.5.40 → 0.5.42

package/CHANGELOG.md

@@ -4,6 +4,27 @@ All notable changes to `pi-studio` are documented here.
 
 ## [Unreleased]
 
+## [0.5.42] — 2026-03-31
+
+### Added
+- Studio now includes a **Refresh from disk** action for file-backed documents, including files originally opened from disk and documents later saved to disk from Studio.
+
+### Changed
+- `/studio-pdf` section, subsection, and deeper heading styling is now more robust for larger-font exports: headings avoid awkward hyphenation, paragraph-level headings (`####`) render cleanly, callout title badges scale better with larger body font sizes, and exported code blocks now use a subtle shaded background.
+
+### Fixed
+- Re-selecting the same file in **Load file content** now reloads it reliably instead of sometimes doing nothing.
+- `/studio-pdf` now supports LaTeX `[H]` float placement in exported documents.
+- Response preview now resets to the top more reliably for genuinely new replies, while keeping editor preview behavior unchanged.
+
+## [0.5.41] — 2026-03-30
+
+### Changed
+- `/studio-pdf` PDF callouts now render with a slightly stronger visual treatment in exported PDFs, including per-kind colours and a more obvious title badge, while staying page-break-friendly.
+
+### Fixed
+- `/studio-pdf` fenced Quarto-style callouts that end with lists now keep their marker paragraphs separate during Pandoc processing, avoiding malformed LaTeX such as callouts closing inside the final list item.
+
 ## [0.5.40] — 2026-03-30
 
 ### Changed

package/client/studio-client.js

@@ -46,7 +46,7 @@
       const rightPaneEl = document.getElementById("rightPane");
       const sourceBadgeEl = document.getElementById("sourceBadge");
       const syncBadgeEl = document.getElementById("syncBadge");
-      const critiqueViewEl = document.getElementById("critiqueView");
+      let critiqueViewEl = document.getElementById("critiqueView");
       const referenceBadgeEl = document.getElementById("referenceBadge");
       const editorViewSelect = document.getElementById("editorViewSelect");
       const rightViewSelect = document.getElementById("rightViewSelect");
@@ -74,6 +74,7 @@
       const loadHistoryPromptBtn = document.getElementById("loadHistoryPromptBtn");
       const saveAsBtn = document.getElementById("saveAsBtn");
       const saveOverBtn = document.getElementById("saveOverBtn");
+      const refreshFromDiskBtn = document.getElementById("refreshFromDiskBtn");
       const sendEditorBtn = document.getElementById("sendEditorBtn");
       const getEditorBtn = document.getElementById("getEditorBtn");
       const loadGitDiffBtn = document.getElementById("loadGitDiffBtn");
@@ -196,6 +197,7 @@
         label: initialSourceState.label,
         path: initialSourceState.path,
       };
+      let fileBackedBaselineText = null;
       let activePane = "left";
       let paneFocusTarget = "off";
       const EDITOR_HIGHLIGHT_MAX_CHARS = 100_000;
@@ -427,6 +429,7 @@
         if (kind === "send_to_editor") return "sending to pi editor";
         if (kind === "get_from_editor") return "loading from pi editor";
         if (kind === "load_git_diff") return "loading git diff";
+        if (kind === "refresh_from_disk") return "refreshing from disk";
         if (kind === "save_as" || kind === "save_over") return "saving editor text";
         return "submitting request";
       }
@@ -779,11 +782,29 @@
         }
       });
 
+      function markFileBackedBaseline(text) {
+        fileBackedBaselineText = String(text || "");
+      }
+
+      function clearFileBackedBaseline() {
+        fileBackedBaselineText = null;
+      }
+
+      function hasRefreshableFilePath() {
+        return Boolean(sourceState && sourceState.path);
+      }
+
+      function editorDiffersFromFileBackedBaseline() {
+        if (!hasRefreshableFilePath()) return false;
+        if (fileBackedBaselineText === null) return true;
+        return sourceTextEl.value !== fileBackedBaselineText;
+      }
+
       function updateSourceBadge() {
         const label = sourceState && sourceState.label ? sourceState.label : "blank";
         sourceBadgeEl.textContent = "Editor origin: " + label;
         // Show "Set working dir" button when not file-backed
-        var isFileBacked = sourceState.source === "file" && Boolean(sourceState.path);
+        var isFileBacked = hasRefreshableFilePath();
         if (isFileBacked) {
           if (resourceDirInput) resourceDirInput.value = "";
           if (resourceDirLabel) resourceDirLabel.textContent = "";
@@ -1926,12 +1947,52 @@
         });
       }
 
+      function replaceResponsePaneWithClone() {
+        const currentEl = critiqueViewEl;
+        if (!currentEl || !currentEl.parentNode || typeof currentEl.cloneNode !== "function") {
+          return currentEl;
+        }
+
+        const replacement = currentEl.cloneNode(true);
+        if (!replacement || replacement.nodeType !== 1) {
+          return currentEl;
+        }
+
+        currentEl.parentNode.replaceChild(replacement, currentEl);
+        critiqueViewEl = replacement;
+        return critiqueViewEl;
+      }
+
       function applyPendingResponseScrollReset() {
         if (!pendingResponseScrollReset || !critiqueViewEl) return false;
         if (rightView === "editor-preview") return false;
-        critiqueViewEl.scrollTop = 0;
-        critiqueViewEl.scrollLeft = 0;
+
         pendingResponseScrollReset = false;
+        let targetEl = replaceResponsePaneWithClone();
+        const schedule = typeof window.requestAnimationFrame === "function"
+          ? window.requestAnimationFrame.bind(window)
+          : (cb) => window.setTimeout(cb, 16);
+        const resetScroll = () => {
+          if (!targetEl || !targetEl.isConnected) return;
+          if (rightView === "editor-preview") return;
+          targetEl.scrollTop = 0;
+          targetEl.scrollLeft = 0;
+        };
+
+        if (targetEl && targetEl.classList) {
+          targetEl.classList.add("response-scroll-resetting");
+        }
+
+        resetScroll();
+        schedule(() => {
+          resetScroll();
+          schedule(() => {
+            resetScroll();
+            if (targetEl && targetEl.classList) {
+              targetEl.classList.remove("response-scroll-resetting");
+            }
+          });
+        });
         return true;
       }
 
@@ -2518,7 +2579,7 @@
 
       function getEffectiveSavePath() {
         // File-backed: use the original path
-        if (sourceState.source === "file" && sourceState.path) return sourceState.path;
+        if (sourceState.path) return sourceState.path;
         // Upload with working dir + filename: derive path
         if (sourceState.source === "upload" && sourceState.label && resourceDirInput && resourceDirInput.value.trim()) {
           var name = sourceState.label.replace(/^upload:\s*/i, "");
@@ -2557,12 +2618,25 @@
         saveOverBtn.title = "Save editor is available after opening a file, setting a working dir, or using Save editor as…. Shortcut: Cmd/Ctrl+S falls back to Save editor as… when needed.";
       }
 
+      function updateRefreshFromDiskTooltip() {
+        if (!refreshFromDiskBtn) return;
+
+        if (hasRefreshableFilePath()) {
+          refreshFromDiskBtn.title = "Reload the current file-backed document from disk: " + sourceState.path;
+          return;
+        }
+
+        refreshFromDiskBtn.title = "Refresh from disk is only available for documents that currently have a file path.";
+      }
+
       function syncActionButtons() {
         const canSaveOver = Boolean(getEffectiveSavePath());
+        const canRefreshFromDisk = hasRefreshableFilePath();
 
         fileInput.disabled = uiBusy;
         saveAsBtn.disabled = uiBusy;
         saveOverBtn.disabled = uiBusy || !canSaveOver;
+        if (refreshFromDiskBtn) refreshFromDiskBtn.disabled = uiBusy || !canRefreshFromDisk;
         sendEditorBtn.disabled = uiBusy || isEditorOnlyMode;
         if (getEditorBtn) getEditorBtn.disabled = uiBusy;
         if (loadGitDiffBtn) loadGitDiffBtn.disabled = uiBusy;
@@ -2581,6 +2655,7 @@
         insertHeaderBtn.disabled = uiBusy || isEditorOnlyMode;
         lensSelect.disabled = uiBusy || isEditorOnlyMode;
         updateSaveFileTooltip();
+        updateRefreshFromDiskTooltip();
         updateHistoryControls();
         updateResultActionButtons();
       }
@@ -2598,6 +2673,9 @@
           label: next && next.label ? next.label : "blank",
           path: next && next.path ? next.path : null,
         };
+        if (!sourceState.path) {
+          clearFileBackedBaseline();
+        }
         updateSourceBadge();
         syncActionButtons();
       }
@@ -3778,6 +3856,9 @@
               label: message.initialDocument.label || "blank",
               path: message.initialDocument.path || null,
             });
+            if (message.initialDocument.path) {
+              markFileBackedBaseline(message.initialDocument.text);
+            }
             refreshResponseUi();
             if (typeof message.initialDocument.label === "string" && message.initialDocument.label.length > 0) {
               setStatus("Loaded " + message.initialDocument.label + ".", "success");
@@ -3978,6 +4059,8 @@
           if (typeof message.requestId === "string" && pendingRequestId === message.requestId) {
             pendingRequestId = null;
             pendingKind = null;
+            clearArmedTitleAttention(message.requestId);
+            stickyStudioKind = null;
           }
           if (message.path) {
             setSourceState({
@@ -3985,6 +4068,7 @@
               label: message.label || message.path,
               path: message.path,
             });
+            markFileBackedBaseline(sourceTextEl.value);
           }
           setBusy(false);
           setWsState("Ready");
@@ -4032,6 +4116,15 @@
             return;
           }
 
+          if (typeof message.requestId === "string" && pendingRequestId === message.requestId) {
+            pendingRequestId = null;
+            pendingKind = null;
+            clearArmedTitleAttention(message.requestId);
+            stickyStudioKind = null;
+            setBusy(false);
+            setWsState("Ready");
+          }
+
           const nextSource =
             nextDoc.source === "file" || nextDoc.source === "last-response"
               ? nextDoc.source
@@ -4045,6 +4138,9 @@
 
           setEditorText(nextDoc.text, { preserveScroll: false, preserveSelection: false });
           setSourceState({ source: nextSource, label: nextLabel, path: nextPath });
+          if (nextPath) {
+            markFileBackedBaseline(nextDoc.text);
+          }
           refreshResponseUi();
           setStatus(
             typeof message.message === "string" && message.message.trim()
@@ -4828,6 +4924,34 @@
         }
       });
 
+      if (refreshFromDiskBtn) {
+        refreshFromDiskBtn.addEventListener("click", () => {
+          if (!hasRefreshableFilePath()) {
+            setStatus("Refresh from disk is only available for file-backed documents.", "warning");
+            return;
+          }
+
+          if (editorDiffersFromFileBackedBaseline()) {
+            const confirmed = window.confirm("Replace current editor contents with the latest version from disk?");
+            if (!confirmed) return;
+          }
+
+          const requestId = beginUiAction("refresh_from_disk");
+          if (!requestId) return;
+
+          const sent = sendMessage({
+            type: "refresh_from_disk_request",
+            requestId,
+          });
+
+          if (!sent) {
+            pendingRequestId = null;
+            pendingKind = null;
+            setBusy(false);
+          }
+        });
+      }
+
       sendEditorBtn.addEventListener("click", () => {
         const content = sourceTextEl.value;
         if (!content.trim()) {
@@ -5136,6 +5260,10 @@
         const file = fileInput.files && fileInput.files[0];
         if (!file) return;
 
+        // Clear the input immediately so selecting the same file again will
+        // still fire a future change event.
+        fileInput.value = "";
+
         const reader = new FileReader();
         reader.onload = () => {
           const text = typeof reader.result === "string" ? reader.result : "";

package/client/studio.css

@@ -1220,6 +1220,10 @@
       transform: translateZ(0);
     }
 
+    .panel-scroll.response-scroll-resetting {
+      overflow-anchor: none;
+    }
+
     .preview-error {
       color: var(--warn);
       margin-bottom: 0.75em;

package/index.ts

@@ -159,6 +159,11 @@ interface SaveOverRequestMessage {
 	content: string;
 }
 
+interface RefreshFromDiskRequestMessage {
+	type: "refresh_from_disk_request";
+	requestId: string;
+}
+
 interface SendToEditorRequestMessage {
 	type: "send_to_editor_request";
 	requestId: string;
@@ -192,6 +197,7 @@ type IncomingStudioMessage =
 	| CompactRequestMessage
 	| SaveAsRequestMessage
 	| SaveOverRequestMessage
+	| RefreshFromDiskRequestMessage
 	| SendToEditorRequestMessage
 	| GetFromEditorRequestMessage
 	| LoadGitDiffRequestMessage
@@ -234,36 +240,65 @@ function buildStudioPdfTitleSpacingLength(value: string | undefined, fallback: s
 	return trimmed || fallback;
 }
 
+function buildStudioPdfCalloutTitleSizeCommand(options?: StudioPdfRenderOptions): string {
+	const sizePt = getStudioRequestedPdfFontsizePt(options);
+	if (sizePt && sizePt >= 14) return "\\normalsize";
+	if (sizePt && sizePt >= 13) return "\\small";
+	return "\\footnotesize";
+}
+
 function buildStudioPdfPreamble(options?: StudioPdfRenderOptions): string {
 	const sectionHeadingSize = buildStudioPdfHeadingSizeCommand(options?.sectionSize, "\\Large");
 	const subsectionHeadingSize = buildStudioPdfHeadingSizeCommand(options?.subsectionSize, "\\large");
 	const subsubsectionHeadingSize = buildStudioPdfHeadingSizeCommand(options?.subsubsectionSize, "\\normalsize");
+	const calloutTitleSize = buildStudioPdfCalloutTitleSizeCommand(options);
 	const sectionSpaceBefore = buildStudioPdfTitleSpacingLength(options?.sectionSpaceBefore, "1.5ex plus 0.5ex minus 0.2ex");
 	const sectionSpaceAfter = buildStudioPdfTitleSpacingLength(options?.sectionSpaceAfter, "1ex plus 0.2ex");
 	const subsectionSpaceBefore = buildStudioPdfTitleSpacingLength(options?.subsectionSpaceBefore, "1.2ex plus 0.4ex minus 0.2ex");
 	const subsectionSpaceAfter = buildStudioPdfTitleSpacingLength(options?.subsectionSpaceAfter, "0.6ex plus 0.1ex");
 	return `\\usepackage{titlesec}
-\\titleformat{\\section}{${sectionHeadingSize}\\bfseries\\sffamily}{}{0pt}{}[\\vspace{3pt}\\titlerule\\vspace{12pt}]
-\\titleformat{\\subsection}{${subsectionHeadingSize}\\bfseries\\sffamily}{}{0pt}{}
-\\titleformat{\\subsubsection}{${subsubsectionHeadingSize}\\bfseries\\sffamily}{}{0pt}{}
+\\titleformat{\\section}{${sectionHeadingSize}\\bfseries\\sffamily\\raggedright\\hyphenpenalty=10000\\exhyphenpenalty=10000\\relax}{}{0pt}{}[\\vspace{3pt}\\titlerule\\vspace{12pt}]
+\\titleformat{\\subsection}{${subsectionHeadingSize}\\bfseries\\sffamily\\raggedright\\hyphenpenalty=10000\\exhyphenpenalty=10000\\relax}{}{0pt}{}
+\\titleformat{\\subsubsection}{${subsubsectionHeadingSize}\\bfseries\\sffamily\\raggedright\\hyphenpenalty=10000\\exhyphenpenalty=10000\\relax}{}{0pt}{}
+\\titleformat{\\paragraph}[runin]{\\normalsize\\bfseries\\sffamily\\raggedright\\hyphenpenalty=10000\\exhyphenpenalty=10000\\relax}{}{0pt}{}
+\\titleformat{\\subparagraph}[runin]{\\small\\bfseries\\sffamily\\raggedright\\hyphenpenalty=10000\\exhyphenpenalty=10000\\relax}{}{0pt}{}
 \\titlespacing*{\\section}{0pt}{${sectionSpaceBefore}}{${sectionSpaceAfter}}
 \\titlespacing*{\\subsection}{0pt}{${subsectionSpaceBefore}}{${subsectionSpaceAfter}}
+\\titlespacing*{\\paragraph}{0pt}{0.9ex plus 0.3ex minus 0.1ex}{0.8em}
+\\titlespacing*{\\subparagraph}{0pt}{0.7ex plus 0.2ex minus 0.1ex}{0.7em}
 \\usepackage{xcolor}
 \\usepackage{varwidth}
 \\definecolor{StudioAnnotationBg}{HTML}{EAF3FF}
 \\definecolor{StudioAnnotationBorder}{HTML}{8CB8FF}
 \\definecolor{StudioAnnotationText}{HTML}{1F5FBF}
+\\definecolor{StudioCodeBlockBg}{HTML}{F6F8FA}
 \\definecolor{StudioDiffAddText}{HTML}{1A7F37}
 \\definecolor{StudioDiffDelText}{HTML}{CF222E}
 \\definecolor{StudioDiffMetaText}{HTML}{57606A}
 \\definecolor{StudioDiffHunkText}{HTML}{0969DA}
+\\definecolor{StudioCalloutNoteBorder}{HTML}{2F6FEB}
+\\definecolor{StudioCalloutNoteText}{HTML}{1F4B99}
+\\definecolor{StudioCalloutNoteLabelBg}{HTML}{EAF2FF}
+\\definecolor{StudioCalloutTipBorder}{HTML}{1A7F37}
+\\definecolor{StudioCalloutTipText}{HTML}{175C2C}
+\\definecolor{StudioCalloutTipLabelBg}{HTML}{EAF7EE}
+\\definecolor{StudioCalloutWarningBorder}{HTML}{B76E00}
+\\definecolor{StudioCalloutWarningText}{HTML}{8A5300}
+\\definecolor{StudioCalloutWarningLabelBg}{HTML}{FFF3D6}
+\\definecolor{StudioCalloutImportantBorder}{HTML}{CF222E}
+\\definecolor{StudioCalloutImportantText}{HTML}{A40E26}
+\\definecolor{StudioCalloutImportantLabelBg}{HTML}{FDEBEC}
+\\definecolor{StudioCalloutCautionBorder}{HTML}{CF222E}
+\\definecolor{StudioCalloutCautionText}{HTML}{A40E26}
+\\definecolor{StudioCalloutCautionLabelBg}{HTML}{FDEBEC}
 \\newcommand{\\studioannotation}[1]{\\begingroup\\setlength{\\fboxsep}{1.5pt}\\fcolorbox{StudioAnnotationBorder}{StudioAnnotationBg}{\\begin{varwidth}{\\dimexpr\\linewidth-2\\fboxsep-2\\fboxrule\\relax}\\raggedright\\textcolor{StudioAnnotationText}{\\sffamily\\footnotesize\\strut #1}\\end{varwidth}}\\endgroup}
 \\newcommand{\\StudioDiffAddTok}[1]{\\textcolor{StudioDiffAddText}{#1}}
 \\newcommand{\\StudioDiffDelTok}[1]{\\textcolor{StudioDiffDelText}{#1}}
 \\newcommand{\\StudioDiffMetaTok}[1]{\\textcolor{StudioDiffMetaText}{#1}}
 \\newcommand{\\StudioDiffHunkTok}[1]{\\textcolor{StudioDiffHunkText}{#1}}
 \\newcommand{\\StudioDiffHeaderTok}[1]{\\textcolor{StudioDiffHunkText}{\\textbf{#1}}}
-\\newenvironment{studiocallout}[1]{\\par\\vspace{0.22em}\\noindent\\begingroup\\color{StudioAnnotationBorder}\\hrule height 0.45pt\\color{black}\\vspace{0.08em}\\noindent{\\sffamily\\bfseries\\textcolor{StudioAnnotationText}{#1}}\\par\\vspace{0.02em}\\leftskip=0.7em\\rightskip=0pt\\parindent=0pt\\parskip=0.15em}{\\par\\vspace{0.02em}\\noindent\\color{StudioAnnotationBorder}\\hrule height 0.45pt\\par\\endgroup\\vspace{0.22em}}
+\\newenvironment{studiocallout}[4]{\\par\\vspace{0.6em}\\noindent\\begingroup\\def\\StudioCalloutBorder{#2}\\def\\StudioCalloutText{#3}\\def\\StudioCalloutLabelBg{#4}\\color{\\StudioCalloutBorder}\\hrule height 0.8pt\\relax\\vspace{0.32em}\\noindent\\colorbox{\\StudioCalloutLabelBg}{\\strut\\hspace{0.55em}{${calloutTitleSize}\\sffamily\\bfseries\\textcolor{\\StudioCalloutText}{#1}}\\hspace{0.55em}}\\par\\vspace{0.24em}\\normalcolor\\leftskip=0.9em\\rightskip=0pt\\parindent=0pt\\parskip=0.18em}{\\par\\vspace{0.12em}\\noindent\\color{\\StudioCalloutBorder}\\hrule height 0.55pt\\par\\endgroup\\vspace{0.5em}}
+\\usepackage{float}
 \\usepackage{caption}
 \\captionsetup[figure]{justification=raggedright,singlelinecheck=false}
 \\usepackage{enumitem}
@@ -273,9 +308,9 @@ function buildStudioPdfPreamble(options?: StudioPdfRenderOptions): string {
 \\usepackage{fvextra}
 \\makeatletter
 \\@ifundefined{Highlighting}{%
-  \\DefineVerbatimEnvironment{Highlighting}{Verbatim}{commandchars=\\\\\\{\\},breaklines,breakanywhere}%
+  \\DefineVerbatimEnvironment{Highlighting}{Verbatim}{commandchars=\\\\\\{\\},breaklines,breakanywhere,bgcolor=StudioCodeBlockBg,framesep=2mm}%
 }{%
-  \\RecustomVerbatimEnvironment{Highlighting}{Verbatim}{commandchars=\\\\\\{\\},breaklines,breakanywhere}%
+  \\RecustomVerbatimEnvironment{Highlighting}{Verbatim}{commandchars=\\\\\\{\\},breaklines,breakanywhere,bgcolor=StudioCodeBlockBg,framesep=2mm}%
 }
 \\makeatother
 `;
@@ -3405,9 +3440,19 @@ function preprocessStudioMarkdownCalloutsForPdf(markdown: string): { markdown: s
 			content: contentLines.join("\n").trim(),
 		};
 		blocks.push(block);
+		// Keep markers on their own paragraphs so pandoc does not absorb them
+		// into neighbouring list items or paragraphs. Without these blank
+		// lines, the end marker for a callout that finishes with a list can be
+		// emitted inside the final \item, which then produces malformed LaTeX
+		// when we later replace the marker range with a custom callout
+		// environment.
+		out.push("");
 		out.push(`PISTUDIOPDFCALLOUTSTART${block.kind.toUpperCase()}${block.markerId}`);
+		out.push("");
 		if (block.content) out.push(block.content);
+		out.push("");
 		out.push(`PISTUDIOPDFCALLOUTEND${block.kind.toUpperCase()}${block.markerId}`);
+		out.push("");
 		i = j;
 	}
 
@@ -3480,6 +3525,52 @@ function preprocessStudioMarkdownImageAlignmentForPdf(markdown: string): { markd
 	return { markdown: out.join("\n"), blocks };
 }
 
+function getStudioPdfCalloutStyle(kind: StudioPdfMarkdownCalloutBlock["kind"]): {
+	label: string;
+	borderColor: string;
+	textColor: string;
+	labelBgColor: string;
+} {
+	switch (kind) {
+		case "note":
+			return {
+				label: "Note",
+				borderColor: "StudioCalloutNoteBorder",
+				textColor: "StudioCalloutNoteText",
+				labelBgColor: "StudioCalloutNoteLabelBg",
+			};
+		case "tip":
+			return {
+				label: "Tip",
+				borderColor: "StudioCalloutTipBorder",
+				textColor: "StudioCalloutTipText",
+				labelBgColor: "StudioCalloutTipLabelBg",
+			};
+		case "warning":
+			return {
+				label: "Warning",
+				borderColor: "StudioCalloutWarningBorder",
+				textColor: "StudioCalloutWarningText",
+				labelBgColor: "StudioCalloutWarningLabelBg",
+			};
+		case "important":
+			return {
+				label: "Important",
+				borderColor: "StudioCalloutImportantBorder",
+				textColor: "StudioCalloutImportantText",
+				labelBgColor: "StudioCalloutImportantLabelBg",
+			};
+		case "caution":
+		default:
+			return {
+				label: "Caution",
+				borderColor: "StudioCalloutCautionBorder",
+				textColor: "StudioCalloutCautionText",
+				labelBgColor: "StudioCalloutCautionLabelBg",
+			};
+	}
+}
+
 function replaceStudioPdfCalloutBlocksInGeneratedLatex(
 	latex: string,
 	blocks: StudioPdfMarkdownCalloutBlock[],
@@ -3494,16 +3585,8 @@ function replaceStudioPdfCalloutBlocksInGeneratedLatex(
 		const endIndex = transformed.indexOf(endMarker, startIndex + startMarker.length);
 		if (endIndex < 0) continue;
 		const inner = transformed.slice(startIndex + startMarker.length, endIndex).trim();
-		const label = block.kind === "note"
-			? "Note"
-			: block.kind === "tip"
-				? "Tip"
-				: block.kind === "warning"
-					? "Warning"
-					: block.kind === "important"
-						? "Important"
-						: "Caution";
-		const replacement = `\\begin{studiocallout}{${label}}\n${inner}\n\\end{studiocallout}`;
+		const style = getStudioPdfCalloutStyle(block.kind);
+		const replacement = `\\begin{studiocallout}{${style.label}}{${style.borderColor}}{${style.textColor}}{${style.labelBgColor}}\n${inner}\n\\end{studiocallout}`;
 		transformed = transformed.slice(0, startIndex) + replacement + transformed.slice(endIndex + endMarker.length);
 	}
 	return transformed;
@@ -4033,11 +4116,12 @@ async function renderStudioLiteralTextPdf(text: string, title = "Studio export",
 \\usepackage[${literalPdfConfig.geometryOptions}]{geometry}
 ${literalPdfConfig.fontCommands}\\usepackage{fvextra}
 \\usepackage{xcolor}
+\\definecolor{StudioCodeBlockBg}{HTML}{F6F8FA}
 \\usepackage{upquote}
 \\begin{document}
 \\renewcommand{\\baselinestretch}{${literalPdfConfig.lineStretch}}\\selectfont
 ${literalPdfConfig.fontSizeCommand}\\section*{${title.replace(/[{}\\]/g, "").trim() || "Studio export"}}
-\\VerbatimInput[breaklines,breakanywhere,fontsize=\\small,frame=single,rulecolor=\\color{black!15},framesep=2mm]{input.txt}
+\\VerbatimInput[breaklines,breakanywhere,fontsize=\\small,bgcolor=StudioCodeBlockBg,frame=single,rulecolor=\\color{black!15},framesep=2mm]{input.txt}
 \\end{document}
 `;
 
@@ -5337,6 +5421,13 @@ function parseIncomingMessage(data: RawData): IncomingStudioMessage | null {
 		};
 	}
 
+	if (msg.type === "refresh_from_disk_request" && typeof msg.requestId === "string") {
+		return {
+			type: "refresh_from_disk_request",
+			requestId: msg.requestId,
+		};
+	}
+
 	if (msg.type === "send_to_editor_request" && typeof msg.requestId === "string" && typeof msg.content === "string") {
 		return {
 			type: "send_to_editor_request",
@@ -5734,6 +5825,7 @@ ${cssVarsBlock}
     <div class="controls">
       <button id="saveAsBtn" type="button" title="Save editor content to a new file path. Cmd/Ctrl+S falls back here when no direct save path is available.">Save editor as…</button>
       <button id="saveOverBtn" type="button" title="Overwrite current file with editor content. Shortcut: Cmd/Ctrl+S.">Save editor</button>
+      <button id="refreshFromDiskBtn" type="button" title="Reload the current file-backed document from disk.">Refresh from disk</button>
       <label class="file-label" title="Load a local file into editor text.">Load file content<input id="fileInput" type="file" accept=".md,.markdown,.mdx,.qmd,.js,.mjs,.cjs,.jsx,.ts,.mts,.cts,.tsx,.py,.pyw,.sh,.bash,.zsh,.json,.jsonc,.json5,.rs,.c,.h,.cpp,.cxx,.cc,.hpp,.hxx,.jl,.f90,.f95,.f03,.f,.for,.r,.R,.m,.tex,.latex,.diff,.patch,.java,.go,.rb,.swift,.html,.htm,.css,.xml,.yaml,.yml,.toml,.lua,.txt,.rst,.adoc" /></label>
       <button id="loadGitDiffBtn" type="button" title="Load the current git diff from the Studio context into the editor.">Load git diff</button>
       <button id="getEditorBtn" type="button" title="Load the current terminal editor draft into Studio.">Load from pi editor</button>
@@ -7119,6 +7211,50 @@ export default function (pi: ExtensionAPI) {
 			return;
 		}
 
+		if (msg.type === "refresh_from_disk_request") {
+			if (!isValidRequestId(msg.requestId)) {
+				sendToClient(client, { type: "error", requestId: msg.requestId, message: "Invalid request ID." });
+				return;
+			}
+			if (isStudioBusy()) {
+				sendToClient(client, { type: "busy", requestId: msg.requestId, message: "Studio is busy." });
+				return;
+			}
+			if (!initialStudioDocument || !initialStudioDocument.path) {
+				sendToClient(client, {
+					type: "error",
+					requestId: msg.requestId,
+					message: "Refresh from disk is only available for file-backed documents.",
+				});
+				return;
+			}
+
+			const refreshed = readStudioFile(initialStudioDocument.path, studioCwd);
+			if (refreshed.ok === false) {
+				sendToClient(client, {
+					type: "error",
+					requestId: msg.requestId,
+					message: refreshed.message,
+				});
+				return;
+			}
+
+			initialStudioDocument = {
+				text: refreshed.text,
+				label: refreshed.label,
+				source: "file",
+				path: refreshed.resolvedPath,
+			};
+
+			broadcast({
+				type: "studio_document",
+				requestId: msg.requestId,
+				document: initialStudioDocument,
+				message: `Reloaded ${refreshed.label} from disk.`,
+			});
+			return;
+		}
+
 		if (msg.type === "send_to_editor_request") {
 			if (!isValidRequestId(msg.requestId)) {
 				sendToClient(client, { type: "error", requestId: msg.requestId, message: "Invalid request ID." });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-studio",
-  "version": "0.5.40",
+  "version": "0.5.42",
   "description": "Two-pane browser workspace for pi with prompt/response editing, annotations, critiques, prompt/response history, and live Markdown/LaTeX/code preview",
   "type": "module",
   "license": "MIT",