feat: add support for plain text response

Author: natocTo

internal/pkg/service/appsproxy/proxy/pagewriter/pagewriter.go

@@ -62,6 +62,15 @@ func (pw *Writer) MountAssets(mux *http.ServeMux) {
 }
 
 func (pw *Writer) writePage(w http.ResponseWriter, req *http.Request, page string, status int, data any) {
+	// Streamlit health checks expect plain text responses, not HTML
+	if IsStreamlitHealthCheck(req.URL.Path) {
+		plainText := renderPlainText(page, status)
+		w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+		w.WriteHeader(status)
+		_, _ = w.Write([]byte(plainText))
+		return
+	}
+
 	var buf bytes.Buffer
 
 	// Render template

internal/pkg/service/appsproxy/proxy/pagewriter/pathutil.go

@@ -0,0 +1,7 @@
+package pagewriter
+
+import "strings"
+
+func IsStreamlitHealthCheck(path string) bool {
+	return strings.HasSuffix(path, "/_stcore/health") || strings.HasSuffix(path, "/_stcore/host-config")
+}

internal/pkg/service/appsproxy/proxy/pagewriter/renderer.go

@@ -0,0 +1,21 @@
+package pagewriter
+
+import (
+	"fmt"
+)
+
+func renderPlainText(page string, status int) string {
+	switch page {
+	case "spinner.gohtml":
+		return `The application is re-starting. Please wait...`
+
+	case "restart_disabled.gohtml":
+		return `The application has been stopped and cannot be restarted automatically.`
+
+	case "error.gohtml":
+		return fmt.Sprintf("An error occurred (status code: %d). Please try again later.", status)
+
+	default:
+		return `No additional information is available.`
+	}
+}