Update app.py

app.py

@@ -22,7 +22,47 @@ import random
 import time
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
-from huggingface_hub import HfApi  # ADDED for HF Dataset
+from huggingface_hub import HfApi
+import threading  # ADDED for background reporting (already imported)
+
+# =============================================
+# ADDED: Tracker Configuration
+# =============================================
+TRACKER_URL = os.environ.get("TRACKER_URL", "https://yukee1992-status-tracker.hf.space")
+
+# =============================================
+# ADDED: Status Reporting Function
+# =============================================
+def report_to_tracker(project_id, service_type, status, file_urls=None, error=None):
+    """Report generation status to central tracker"""
+    if not project_id:  # Don't report if no project_id
+        return
+    
+    def _report():
+        try:
+            payload = {
+                "project_id": project_id,
+                "service_type": service_type,
+                "status": status,
+                "file_urls": file_urls or []
+            }
+            if error:
+                payload["error"] = error
+            
+            response = requests.post(
+                f"{TRACKER_URL}/update",
+                json=payload,
+                timeout=5
+            )
+            print(f"📤 Reported to tracker: {response.status_code}")
+            if response.status_code != 200:
+                print(f"⚠️ Tracker response: {response.text}")
+        except Exception as e:
+            print(f"⚠️ Failed to report to tracker: {e}")
+    
+    # Run in background thread to not block
+    thread = threading.Thread(target=_report, daemon=True)
+    thread.start()
 
 # =============================================
 # HUGGING FACE DATASET CONFIGURATION
@@ -34,6 +74,7 @@ DATASET_ID = f"{HF_USERNAME}/{DATASET_NAME}"
 
 print(f"📦 HF Dataset: {DATASET_ID}")
 print(f"🔑 HF Token: {'✅ Set' if HF_TOKEN else '❌ Missing'}")
+print(f"📡 Tracker URL: {TRACKER_URL}")  # ADDED
 
 # Create local directories for test images
 PERSISTENT_IMAGE_DIR = "generated_test_images"
@@ -77,7 +118,7 @@ class StorybookRequest(BaseModel):
     style: str = "childrens_book"
     callback_url: Optional[str] = None
     consistency_seed: Optional[int] = None
-    project_id: Optional[str] = None  # ADDED for HF Dataset
+    project_id: Optional[str] = None
 
 class JobStatusResponse(BaseModel):
     job_id: str
@@ -495,10 +536,6 @@ def refresh_local_images():
         print(f"Error refreshing local images: {e}")
         return []
 
-# =============================================
-# REMOVED: OCI BUCKET FUNCTIONS
-# =============================================
-
 # JOB MANAGEMENT FUNCTIONS
 def create_job(story_request: StorybookRequest) -> str:
     job_id = str(uuid.uuid4())
@@ -688,6 +725,17 @@ def generate_storybook_background(job_id: str):
             }
         }
         
+        # =============================================
+        # ADDED: Report to tracker when all images are done
+        # =============================================
+        if project_id and image_urls:
+            report_to_tracker(
+                project_id=project_id,
+                service_type="image",
+                status="completed",
+                file_urls=image_urls
+            )
+        
         status_message = f"🎉 Storybook completed! {len(generated_pages)} pages created."
         if hf_success_count > 0:
             status_message += f" {hf_success_count} images uploaded to HF Dataset."
@@ -709,6 +757,16 @@ def generate_storybook_background(job_id: str):
         error_msg = f"Story generation failed: {str(e)}"
         print(f"❌ {error_msg}")
         traceback.print_exc()
+        
+        # Report failure to tracker
+        if 'project_id' in locals() and project_id:
+            report_to_tracker(
+                project_id=project_id,
+                service_type="image",
+                status="failed",
+                error=error_msg
+            )
+        
         update_job_status(job_id, JobStatus.FAILED, 0, error_msg)
 
 # FASTAPI ENDPOINTS
@@ -851,6 +909,7 @@ async def root():
     return {
         "message": "Storybook Generator API with HF Dataset",
         "hf_dataset": DATASET_ID if HF_TOKEN else "Disabled",
+        "tracker_url": TRACKER_URL,  # ADDED for info
         "endpoints": {
             "generate": "POST /api/generate-storybook",
             "status": "GET /api/job-status/{job_id}",
@@ -866,6 +925,7 @@ if __name__ == "__main__":
     if os.environ.get('SPACE_ID'):
         print("🚀 Running on Hugging Face Spaces")
         print(f"📦 HF Dataset: {DATASET_ID if HF_TOKEN else 'Disabled'}")
+        print(f"📡 Tracker URL: {TRACKER_URL}")  # ADDED
         gr.mount_gradio_app(app, demo, path="/ui")
         uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")
     else: