Add stage information to API status responses

- Add stage field to JobInfo class to track current processing stage
- Update update_job_status function to include stage parameter
- Add stage tracking at each milestone in process_generation_job
- Include stage in API status response for better client feedback
- Update API documentation with stage values and examples
- Stages: queued, initializing, preprocessing, shape_generation, face_reduction, texture_generation, completed, failed

API_README.md

@@ -76,6 +76,7 @@ Check the status of a generation job.
 {
   "status": "completed|processing|queued|failed",
   "progress": 0-100,
+  "stage": "current_processing_stage",
   "model_urls": {
     "glb": "url_to_glb_file"
   }
@@ -88,6 +89,16 @@ Check the status of a generation job.
 - `completed`: Job completed successfully
 - `failed`: Job failed with an error
 
+**Stage Values:**
+- `queued`: Job is waiting to be processed
+- `initializing`: Setting up job and converting images
+- `preprocessing`: Preparing images and options
+- `shape_generation`: Generating 3D mesh from images
+- `face_reduction`: Optimizing mesh geometry
+- `texture_generation`: Creating textures for the 3D model
+- `completed`: Job finished successfully
+- `failed`: Job failed with an error
+
 ## Usage Examples
 
 ### Python Example
@@ -124,7 +135,7 @@ while True:
         print(f"Generation failed: {data.get('error')}")
         break
     
-    print(f"Progress: {data['progress']}%")
+    print(f"Progress: {data['progress']}% - Stage: {data['stage']}")
     time.sleep(5)
 ```
 

gradio_app.py

@@ -82,6 +82,7 @@ class JobInfo:
         self.job_id = job_id
         self.status = JobStatus.QUEUED
         self.progress = 0
+        self.stage = "queued"
         self.start_time = time.time()
         self.end_time = None
         self.error_message = None
@@ -101,12 +102,14 @@ def create_job() -> str:
     return job_id
 
 
-def update_job_status(job_id: str, status: JobStatus, progress: int = None, error_message: str = None):
+def update_job_status(job_id: str, status: JobStatus, progress: int = None, stage: str = None, error_message: str = None):
     """Update job status and progress."""
     if job_id in jobs:
         jobs[job_id].status = status
         if progress is not None:
             jobs[job_id].progress = progress
+        if stage is not None:
+            jobs[job_id].stage = stage
         if error_message is not None:
             jobs[job_id].error_message = error_message
         if status in [JobStatus.COMPLETED, JobStatus.FAILED]:
@@ -132,7 +135,7 @@ def process_generation_job(job_id: str, images: Dict[str, str], options: Dict[st
     global face_reduce_worker, tex_pipeline, HAS_TEXTUREGEN, SAVE_DIR
     
     try:
-        update_job_status(job_id, JobStatus.PROCESSING, progress=10)
+        update_job_status(job_id, JobStatus.PROCESSING, progress=10, stage="initializing")
         
         # Convert base64 images to PIL Images
         pil_images = {}
@@ -144,7 +147,7 @@ def process_generation_job(job_id: str, images: Dict[str, str], options: Dict[st
         should_remesh = options.get("should_remesh", True)
         should_texture = options.get("should_texture", True)
         
-        update_job_status(job_id, JobStatus.PROCESSING, progress=20)
+        update_job_status(job_id, JobStatus.PROCESSING, progress=20, stage="preprocessing")
         
         # Generate 3D mesh
         # For non-MV mode, use the front image as the main image, or the first available image
@@ -169,7 +172,7 @@ def process_generation_job(job_id: str, images: Dict[str, str], options: Dict[st
             randomize_seed=False,
         )
         
-        update_job_status(job_id, JobStatus.PROCESSING, progress=50)
+        update_job_status(job_id, JobStatus.PROCESSING, progress=50, stage="shape_generation")
         
         # Export white mesh
         white_mesh_path = export_mesh(mesh, save_folder, textured=False, type='obj')
@@ -178,7 +181,7 @@ def process_generation_job(job_id: str, images: Dict[str, str], options: Dict[st
         mesh = face_reduce_worker(mesh)
         reduced_mesh_path = export_mesh(mesh, save_folder, textured=False, type='obj')
         
-        update_job_status(job_id, JobStatus.PROCESSING, progress=70)
+        update_job_status(job_id, JobStatus.PROCESSING, progress=70, stage="face_reduction")
         
         # Texture generation if enabled
         textured_mesh_path = None
@@ -215,7 +218,7 @@ def process_generation_job(job_id: str, images: Dict[str, str], options: Dict[st
                 # If texture generation fails, we still have the white mesh as fallback
                 textured_mesh_path = None
         
-        update_job_status(job_id, JobStatus.PROCESSING, progress=90)
+        update_job_status(job_id, JobStatus.PROCESSING, progress=90, stage="texture_generation")
         
         # Prepare model URLs
         model_urls = {}
@@ -228,11 +231,11 @@ def process_generation_job(job_id: str, images: Dict[str, str], options: Dict[st
         
         # Update job with results
         jobs[job_id].model_urls = model_urls
-        update_job_status(job_id, JobStatus.COMPLETED, progress=100)
+        update_job_status(job_id, JobStatus.COMPLETED, progress=100, stage="completed")
         
     except Exception as e:
         logger.error(f"Job {job_id} failed: {e}")
-        update_job_status(job_id, JobStatus.FAILED, error_message=str(e))
+        update_job_status(job_id, JobStatus.FAILED, stage="failed", error_message=str(e))
 
 
 MAX_SEED = 1e7
@@ -1245,7 +1248,8 @@ if __name__ == '__main__':
         
         response = {
             "status": job.status.value,
-            "progress": job.progress
+            "progress": job.progress,
+            "stage": job.stage
         }
         
         if job.status == JobStatus.COMPLETED: