jketreno/backstory
1
0
Fork 0
Code Issues 12 Actions Releases Activity

Multi-modality now working

Browse Source
This commit is contained in:
James Ketr 2025-05-27 12:33:33 -07:00
parent 5c16c780a3
commit 7311027811
3 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frontend/src/NewApp/Components/GenerateImage.tsx
View File

@ -81,7 +81,12 @@ const GenerateImage = (props: GenerateImageProps) => {
controllerRef.current = null;
break;
default:
const data = JSON.parse(msg.response || '');
let data: any = {};
try {
data = typeof msg.response === 'string' ? JSON.parse(msg.response) : msg.response;
} catch (e) {
data = { message: msg.response };
}
if (msg.status !== "heartbeat") {
console.log(data);
}
@ -112,7 +117,7 @@ const GenerateImage = (props: GenerateImageProps) => {
maxWidth: { xs: '100%', md: '700px', lg: '1024px' },
minHeight: "max-content",
}}>
{image !== '' && <img src={image}/>}
{image !== '' && <img alt={prompt} src={`${image}/${sessionId}`} />}
{ prompt &&
<Quote size={processing ? "normal" : "small"} quote={prompt} sx={{ "& *": { color: "#2E2E2E !important" }}}/>
}

7
frontend/src/NewApp/Pages/GenerateCandidate.tsx
View File

@ -215,7 +215,12 @@ const GenerateCandidate = (props: BackstoryElementProps) => {
setShouldGenerateProfile(false);
break;
default:
const data = JSON.parse(msg.response || '');
let data: any = {};
try {
data = typeof msg.response === 'string' ? JSON.parse(msg.response) : msg.response;
} catch (e) {
data = { message: msg.response };
}
if (msg.status !== "heartbeat") {
console.log(data);
}

2
src/utils/agents/image_generator.py
View File

@ -166,7 +166,7 @@ class ImageGenerator(Agent):
# Image generated
message.status = "done"
if self.filename != "profile.png":
message.response = f"/apu/u/{self.username}/images/{self.filename}"
message.response = f"/api/u/{self.username}/images/{self.filename}"
else:
message.response = f"/api/u/{self.username}/profile"