Update app.py

app.py

@@ -396,7 +396,7 @@ def update_job_status(job_id: str, status: JobStatus, progress: int, message: st
         try:
             callback_url = request_data["callback_url"]
             
-            # Simple callback data
+            # Enhanced callback data with scene information
             callback_data = {
                 "job_id": job_id,
                 "status": status.value,
@@ -405,16 +405,33 @@ def update_job_status(job_id: str, status: JobStatus, progress: int, message: st
                 "story_title": request_data["story_title"],
                 "total_scenes": len(request_data["scenes"]),
                 "timestamp": time.time(),
-                "source": "huggingface-storybook-generator"
+                "source": "huggingface-storybook-generator",
+                "estimated_time_remaining": calculate_remaining_time(job_id, progress)
             }
             
+            # Add current scene info for processing jobs
+            if status == JobStatus.PROCESSING:
+                # Calculate current scene based on progress
+                total_scenes = len(request_data["scenes"])
+                if total_scenes > 0:
+                    current_scene = min((progress - 5) // (90 // total_scenes) + 1, total_scenes)
+                    callback_data["current_scene"] = current_scene
+                    callback_data["total_scenes"] = total_scenes
+                    
+                    # Add scene description if available
+                    if current_scene <= len(request_data["scenes"]):
+                        scene_data = request_data["scenes"][current_scene-1]
+                        callback_data["scene_description"] = scene_data.get("visual", "")[:100] + "..."
+                        callback_data["current_prompt"] = scene_data.get("visual", "")
+            
             # Add result data for completed jobs
             if status == JobStatus.COMPLETED and result:
                 callback_data["result"] = {
                     "total_pages": result.get("total_pages", 0),
                     "generation_time": result.get("generation_time", 0),
                     "oci_bucket_url": result.get("oci_bucket_url", ""),
-                    "pages_generated": result.get("generated_pages", 0)
+                    "pages_generated": result.get("generated_pages", 0),
+                    "consistency_seed": result.get("consistency_seed", None)
                 }
             
             headers = {
@@ -422,6 +439,9 @@ def update_job_status(job_id: str, status: JobStatus, progress: int, message: st
                 'User-Agent': 'Storybook-Generator/1.0'
             }
             
+            print(f"📢 Sending callback to: {callback_url}")
+            print(f"📊 Callback data: {json.dumps(callback_data, indent=2)}")
+            
             response = requests.post(
                 callback_url,
                 json=callback_data,