rework to allow for image gen

ollamaClient.js

@@ -0,0 +1,78 @@
+const OLLAMA_API_URL = process.env.OLLAMA_API_URL;
+const OLLAMA_API_KEY = process.env.OLLAMA_API_KEY;
+
+async function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+// Utility: strip markdown artifacts
+function cleanText(str) {
+  return str
+    .replace(/^#+\s*/gm, "")          // remove headers
+    .replace(/\*\*(.*?)\*\*/g, "$1")  // remove bold
+    .replace(/[*_`]/g, "")            // remove stray formatting
+    .replace(/\s+/g, " ")             // normalize whitespace
+    .trim();
+}
+
+export async function callOllama(prompt, model = "gemma3n:e4b", retries = 5, stepName = "unknown") {
+  const isUsingOpenWebUI = !!OLLAMA_API_KEY;
+
+  for (let attempt = 1; attempt <= retries; attempt++) {
+    try {
+      const promptCharCount = prompt.length;
+      const promptWordCount = prompt.split(/\s+/).length;
+
+      console.log(`\n[${stepName}] Sending prompt (attempt ${attempt}/${retries})`);
+      console.log(`Prompt: ${promptCharCount} chars, ~${promptWordCount} words`);
+
+      const headers = { "Content-Type": "application/json" };
+
+      if (isUsingOpenWebUI) {
+        headers["Authorization"] = `Bearer ${OLLAMA_API_KEY}`;
+      }
+
+      const body = isUsingOpenWebUI
+        ? {
+          model,
+          messages: [{ role: "user", content: prompt }],
+        }
+        : {
+          model,
+          messages: [{ role: "user", content: prompt }],
+          stream: false,
+        };
+
+      const response = await fetch(OLLAMA_API_URL, {
+        method: "POST",
+        headers,
+        body: JSON.stringify(body),
+      });
+
+      if (!response.ok) throw new Error(`Ollama request failed: ${response.status} ${response.statusText}`);
+
+      const data = await response.json();
+
+      const rawText = isUsingOpenWebUI
+        ? data.choices?.[0]?.message?.content
+        : data.message?.content;
+
+      if (!rawText) throw new Error("No response from Ollama");
+
+      const cleaned = cleanText(rawText);
+
+      console.log(`[${stepName}] Received: ${rawText.length} chars, ~${rawText.split(/\s+/).length} words`);
+      console.log(`Raw output:\n${rawText}\n`);
+      console.log(`Cleaned output:\n${cleaned}\n`);
+
+      return cleaned;
+
+    } catch (err) {
+      console.warn(`[${stepName}] Attempt ${attempt} failed: ${err.message}`);
+      if (attempt === retries) throw err;
+      const delay = 1000 * Math.pow(2, attempt) + Math.random() * 500;
+      console.log(`Retrying in ${Math.round(delay / 1000)}s...`);
+      await sleep(delay);
+    }
+  }
+}