Spaces:

dippoo
/

content-engine

Running

App Files Files

dippoo commited on Feb 17

Commit

136976f

1 Parent(s): a5fa801

Fix: Add job tracking to img2img cloud generation

Browse files

Files changed (3) hide show

src/content_engine/__pycache__/main.cpython-311.pyc +0 -0
src/content_engine/api/__pycache__/routes_generation.cpython-311.pyc +0 -0
src/content_engine/api/ui.html +12 -2

src/content_engine/__pycache__/main.cpython-311.pyc CHANGED Viewed

Binary files a/src/content_engine/__pycache__/main.cpython-311.pyc and b/src/content_engine/__pycache__/main.cpython-311.pyc differ

src/content_engine/api/__pycache__/routes_generation.cpython-311.pyc CHANGED Viewed

Binary files a/src/content_engine/api/__pycache__/routes_generation.cpython-311.pyc and b/src/content_engine/api/__pycache__/routes_generation.cpython-311.pyc differ

src/content_engine/api/ui.html CHANGED Viewed

@@ -2191,8 +2191,18 @@ async function doGenerate() {
       const data = await res.json();
       if (!res.ok) throw new Error(data.detail || 'Generation failed');
-      const backendLabel = selectedBackend === 'cloud' ? 'Cloud edit' : 'Local img2img';
-      toast(`${backendLabel} generation started!`, 'info');
       await pollForNewImage(startTime);
       return;
     }

       const data = await res.json();
       if (!res.ok) throw new Error(data.detail || 'Generation failed');
+      // Track job ID for cloud img2img
+      if (selectedBackend === 'cloud' && data.job_id) {
+        currentJobId = data.job_id;
+        toast(`Cloud edit started (${data.job_id.substring(0,8)})`, 'info');
+        pollJobStatus(data.job_id).catch(e => {
+          if (e.message) toast('Job failed: ' + e.message, 'error');
+        });
+      } else {
+        const backendLabel = selectedBackend === 'cloud' ? 'Cloud edit' : 'Local img2img';
+        toast(`${backendLabel} generation started!`, 'info');
+      }
       await pollForNewImage(startTime);
       return;
     }