Update plugins/jobs-imggen.js

plugins/jobs-imggen.js

@@ -1,4 +1,3 @@
-const axios = require('axios');
 const fs = require('fs');
 const path = require('path');
 
@@ -16,83 +15,6 @@ const saveJobs = (jobs) => {
     fs.writeFileSync(jobsPath, JSON.stringify(jobs, null, 2));
 };
 
-const checkres = async (token, iid) => {
-    const res = await axios.get(`https://supawork.ai/supawork/headshot/api/media/aigc/result/list/v1?page_no=1&page_size=10&identity_id=${iid}`, {
-        headers: { 
-            'Authorization': token,
-            'X-Identity-Id': iid,
-            'X-Auth-Challenge': '',
-            'User-Agent': 'Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36'
-        }
-    });
-    
-    return res.data.data;
-};
-
-const processJob = async (jobId) => {
-    const jobs = loadJobs();
-    const job = jobs[jobId];
-    
-    if (!job || job.status === 'completed' || job.status === 'failed') {
-        return;
-    }
-
-    try {
-        console.log(`[Job ${jobId}] Checking result...`);
-        
-        const res = await checkres(job.token, job.identityId);
-        
-        if (res.list && res.list.length > 0) {
-            const task = res.list[0];
-            console.log(`[Job ${jobId}] Task status: ${task.status}, Image status: ${task.list[0].status}`);
-            
-            if (task.status === 1 && task.list[0].status === 1) {
-                const resurl = task.list[0].url[0];
-                console.log(`[Job ${jobId}] Image ready`);
-                
-                job.status = 'completed';
-                job.result_url = resurl;
-                job.completed_at = new Date().toISOString();
-                saveJobs(jobs);
-                
-                console.log(`[Job ${jobId}] Completed successfully`);
-                
-                setTimeout(() => {
-                    const jobs = loadJobs();
-                    delete jobs[jobId];
-                    saveJobs(jobs);
-                    console.log(`[Job ${jobId}] Job data cleaned up`);
-                }, 300000);
-                
-                return;
-            }
-            
-            if (task.status === 2 || task.list[0].status === 2) {
-                throw new Error('Generation failed');
-            }
-            
-            console.log(`[Job ${jobId}] Still generating... will check again in 5s`);
-            setTimeout(() => processJob(jobId), 5000);
-        } else {
-            console.log(`[Job ${jobId}] No results yet... will check again in 5s`);
-            setTimeout(() => processJob(jobId), 5000);
-        }
-        
-    } catch (error) {
-        console.error(`[Job ${jobId}] Error:`, error.message);
-        job.status = 'failed';
-        job.error = error.message;
-        job.failed_at = new Date().toISOString();
-        saveJobs(jobs);
-        
-        setTimeout(() => {
-            const jobs = loadJobs();
-            delete jobs[jobId];
-            saveJobs(jobs);
-        }, 300000);
-    }
-};
-
 const handler = async (req, res) => {
     try {
         const { id, key } = req.query;
@@ -111,10 +33,6 @@ const handler = async (req, res) => {
             });
         }
 
-        if (job.status === 'processing' && job.stage === 'waiting_generation') {
-            processJob(id);
-        }
-
         const response = {
             author: "Herza",
             success: true,
@@ -123,17 +41,29 @@ const handler = async (req, res) => {
                 status: job.status,
                 stage: job.stage,
                 prompt: job.prompt,
-                ratio: job.ratio,
                 created_at: job.created_at
             }
         };
 
         if (job.status === 'completed') {
-            response.data.result_url = job.result_url;
+            response.data.result = job.result;
             response.data.completed_at = job.completed_at;
+            
+            setTimeout(() => {
+                const jobs = loadJobs();
+                delete jobs[id];
+                saveJobs(jobs);
+                console.log(`[Job ${id}] Job data cleaned up`);
+            }, 300000);
         } else if (job.status === 'failed') {
             response.data.error = job.error;
             response.data.failed_at = job.failed_at;
+            
+            setTimeout(() => {
+                const jobs = loadJobs();
+                delete jobs[id];
+                saveJobs(jobs);
+            }, 300000);
         } else {
             response.data.message = 'Job is still processing. Check again in a few seconds.';
         }
@@ -151,7 +81,7 @@ const handler = async (req, res) => {
 };
 
 module.exports = {
-    name: 'Job Status Checker',
+    name: 'Check Image Generator result',
     description: 'Check image generation job status and get result',
     type: 'GET',
     routes: ['api/job/imagegen'],