Update app.py

app.py

@@ -366,29 +366,73 @@ def update_job_status(job_id: str, status: JobStatus, progress: int, message: st
     
     if request_data.get("callback_url"):
         try:
+            callback_url = request_data["callback_url"]
+            
+            # Enhanced callback data for better Telegram notifications
             callback_data = {
                 "job_id": job_id,
                 "status": status.value,
                 "progress": progress,
                 "message": message,
                 "story_title": request_data["story_title"],
-                "timestamp": time.time()
+                "total_scenes": len(request_data["scenes"]),
+                "timestamp": time.time(),
+                "source": "huggingface-storybook-generator",
+                "estimated_time_remaining": calculate_remaining_time(job_id, progress)
             }
             
+            # Add result data for completed jobs
             if status == JobStatus.COMPLETED and result:
-                callback_data["result"] = 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)
+                }
+            
+            # Add current scene info for processing jobs
+            if status == JobStatus.PROCESSING:
+                current_scene = progress // (100 // len(request_data["scenes"])) + 1
+                callback_data["current_scene"] = current_scene
+                callback_data["total_scenes"] = len(request_data["scenes"])
+                callback_data["scene_description"] = request_data["scenes"][current_scene-1]["visual"][:100] + "..."
+            
+            headers = {
+                'Content-Type': 'application/json',
+                'User-Agent': 'Storybook-Generator/1.0'
+            }
             
             response = requests.post(
-                request_data["callback_url"],
+                callback_url,
                 json=callback_data,
-                timeout=10
+                headers=headers,
+                timeout=30
             )
-            print(f"📢 Callback sent to n8n: Status {response.status_code}")
+            
+            print(f"📢 Callback sent: Status {response.status_code}")
+                
         except Exception as e:
             print(f"⚠️ Callback failed: {str(e)}")
     
     return True
 
+def calculate_remaining_time(job_id, progress):
+    """Calculate estimated time remaining"""
+    if progress == 0:
+        return "Calculating..."
+    
+    job_data = job_storage.get(job_id)
+    if not job_data:
+        return "Unknown"
+    
+    time_elapsed = time.time() - job_data["created_at"]
+    if progress > 0:
+        total_estimated = (time_elapsed / progress) * 100
+        remaining = total_estimated - time_elapsed
+        return f"{int(remaining // 60)}m {int(remaining % 60)}s"
+    
+    return "Unknown"
+
 # BACKGROUND TASK FOR COMPLETE STORYBOOK GENERATION (n8n API)
 def generate_storybook_background(job_id: str):
     """Background task to generate complete storybook with images and text"""