From a09221bb2602d2b5e80b0aa0d0376fd3867b8c37 Mon Sep 17 00:00:00 2001
From: davidtparks <Davidthomasparks@gmail.com>
Date: Sat, 11 Jan 2025 15:37:40 -0600
Subject: [PATCH] Fix types

---
 app/api/generate/[inferenceId]/route.ts |  9 ---------
 components/create/generation-set.tsx    |  5 +++--
 types/scenario.ts                       | 20 +++++++++++++++-----
 3 files changed, 18 insertions(+), 16 deletions(-)

diff --git a/app/api/generate/[inferenceId]/route.ts b/app/api/generate/[inferenceId]/route.ts
index a6f886e..b5dd757 100644
--- a/app/api/generate/[inferenceId]/route.ts
+++ b/app/api/generate/[inferenceId]/route.ts
@@ -110,7 +110,6 @@ export const pixelateImageScenario = async ({
     const pixelateData: ScenarioPixelateResponse = await pixelateResponse.json()
 
 
-    console.log("pixelate data response", pixelateData)
     // @ts-ignore
     return pixelateData.asset
 }
@@ -153,11 +152,9 @@ export async function GET(
             }
         ).then((res) => res.json())
 
-        console.log(jobProgress)
 
         // If the job was successful, process the images and update credits
         if (jobProgress.job.status === "success") {
-            console.log("Job success")
             // Wrap this in try-catch since findUniqueOrThrow can fail
             try {
                 const generation = await db.generation.findUniqueOrThrow({
@@ -172,8 +169,6 @@ export async function GET(
                     },
                 })
 
-                console.log("generateion jere", generation)
-
                 // If already processed, return existing output images
                 if (generation.status === "COMPLETE") {
                     return new Response(
@@ -227,8 +222,6 @@ export async function GET(
                     })
                 )
 
-                console.log("pixelateImages", pixelatedImages)
-                console.log("jobProgress.job.metadata.assetIds", jobProgress.job.metadata.assetIds)
 
                 const imagesWithPixelated = jobProgress.job.metadata.assetIds.map(
                     (assetId, index) => {
@@ -242,8 +235,6 @@ export async function GET(
                     }
                 )
 
-                console.log("imagesWithPixelated", imagesWithPixelated)
-                console.log("generation", generation)
 
                 await db.$transaction([
                     db.user.update({
diff --git a/components/create/generation-set.tsx b/components/create/generation-set.tsx
index 86dcdac..018fe2f 100644
--- a/components/create/generation-set.tsx
+++ b/components/create/generation-set.tsx
@@ -85,8 +85,9 @@ export const GenerationSet = ({
                         jsonResponse?.outputImages
                     ) {
                         generatedImages = jsonResponse.outputImages
-
-                        setImages(generatedImages)
+                        if (generatedImages) {
+                            setImages(generatedImages)
+                        }
                     } else if (jsonResponse.job?.status === "failed") {
                         toast({
                             title: "Generation failed",
diff --git a/types/scenario.ts b/types/scenario.ts
index 8c1e359..38d1519 100644
--- a/types/scenario.ts
+++ b/types/scenario.ts
@@ -1,6 +1,7 @@
 import { Generation, OutputImage } from "@prisma/client"
 
 export interface ScenarioInferenceResponse {
+    inference: Inference
     job: Job
 }
 
@@ -40,23 +41,32 @@ export interface Inference {
     authorId: string
     modelId: string
     createdAt: string
-    parameters: Parameters
+    parameters: InferenceParameters
     status: string
-    images: any[]
+    images: InferenceImage[]
     imagesNumber: number
     displayPrompt: string
 }
 
-export interface Parameters {
+export interface InferenceParameters {
     numSamples: number
     guidance: number
     numInferenceSteps: number
-    enableSafetyCheck: boolean
     width: number
     height: number
     type: string
     prompt: string
-    strength: number
+    negativePrompt?: string
+    strength?: number
+    modality?: string
+    image?: string
+}
+
+export interface InferenceImage {
+    id: string
+    url: string
+    seed: string
+    pixelated?: string
 }
 
 export interface ScenarioInferenceProgressResponse {