rewire prompting

This commit is contained in:
Brandon Thomas committed 2025-04-08 02:21:17 -04:00
1 parent b41b4f8e17
commit 2ea9464be2
3 files changed
+27 -13

No files matched your search

-2
View File
@@ -338,8 +338,6 @@ export class Engine {
});
uiEvents.toolbarMain.SAVE.onClick(async (/*event*/) => {
//document.location.href = "https://auth.openai.com/log-in";
//console.log('save clicked');
//await this.realTimeDrawEngine.saveOutput(); // This works.
await invoke("open_login_command");
});
@@ -27,6 +27,7 @@ import {
import { ensureBase64Prefix } from "../EngineUtitlities/Base64Helpers";
import { DecodeBase64ToImage } from "~/utilities/DecodeBase64ToImage.ts";
import { EncodeImageBitmapToBase64 } from "~/utilities/EncodeImageBitmapToBase64.ts";
import { setCanvasRenderBitmap } from "~/signal/canvasRenderBitmap";
interface ServerSettings {
model_path: string;
@@ -789,6 +790,11 @@ export class RealTimeDrawEngine {
const base64Bitmap = await this.imageBitmapToBase64(
this.lastRenderedBitmap,
);
const generateResponse = await invoke("image_generation_command", {
image: base64Bitmap,
prompt: this.currentPrompt,
});
}
public async render() {
@@ -822,18 +828,9 @@ export class RealTimeDrawEngine {
this.lastRenderedBitmap = bitmap;
setCanvasRenderBitmap(bitmap);
try {
//const base64Bitmap = await this.imageBitmapToBase64(bitmap);
//const base64BitmapResponse = await invoke("image_generation_command", {
// image: base64Bitmap,
// prompt: this.currentPrompt,
//});
////console.log(base64BitmapResponse);
//const decoded = await DecodeBase64ToImage(
// base64BitmapResponse as string,
//);
//this.outputBitmap = decoded;
//this.previewCanvas.image(decoded);
} catch (error) {
console.error("Error during image processing:", error);
} finally {
@@ -1,4 +1,5 @@
import { useState, useRef, useEffect } from "react";
import { invoke } from "@tauri-apps/api/core";
import { useSignals } from "@preact/signals-react/runtime";
import {
UploaderStates,
@@ -33,6 +34,8 @@ import {
import { FontAwesomeIcon } from "@fortawesome/react-fontawesome";
import { PopoverItem } from "~/components/reusable/Popover";
import { getCanvasRenderBitmap } from "~/signal/canvasRenderBitmap";
import { EncodeImageBitmapToBase64 } from "~/utilities/EncodeImageBitmapToBase64";
interface ReferenceImage {
id: string;
@@ -48,6 +51,8 @@ interface ExtendedPopoverItem extends PopoverItem {
export const PromptBox = () => {
useSignals();
//const { lastRenderedBitmap } = useCanvasSignal();
const [prompt, setPrompt] = useState("");
const [isEnqueueing, setisEnqueueing] = useState(false);
const [useSystemPrompt, setUseSystemPrompt] = useState(true);
@@ -253,6 +258,20 @@ export const PromptBox = () => {
"and reference images:",
referenceImages,
);
let image = getCanvasRenderBitmap();
if (image === undefined) {
return;
}
const base64Bitmap = await EncodeImageBitmapToBase64(image);
const generateResponse = await invoke("image_generation_command", {
image: base64Bitmap,
prompt: prompt,
});
await new Promise((resolve) => setTimeout(resolve, 2000));
} finally {
setisEnqueueing(false);