Update app/routers/content.py

app/routers/content.py

@@ -5,6 +5,7 @@ from app.database import get_db
 from app import models, schemas
 from app.ingest import extract_from_pdf, extract_from_url
 from app.tasks import task_generate_draft
+from app.config import settings
 
 router = APIRouter(prefix="/content", tags=["content"])
 
@@ -29,7 +30,6 @@ async def create_draft(
         raw = text or ""
         source_ref = "text"
 
-
     draft = models.Draft(
         source_type=source_type, source_ref=source_ref,
         raw_text=raw[:20000], content_type=content_type, tone=tone,
@@ -37,12 +37,17 @@ async def create_draft(
     )
     db.add(draft); db.commit(); db.refresh(draft)
 
+    # Queue if broker exists, otherwise generate synchronously so Spaces demo works
+    try:
+        if settings.CELERY_BROKER_URL:
+            task_generate_draft.delay(draft.id)
+        else:
+            task_generate_draft(draft.id)  # run inline
+    except Exception:
+        task_generate_draft(draft.id)
 
-    task_generate_draft.delay(draft.id)
-     # 生成完了後はstatus=pendingになる。UIはポーリングで取得。
     return draft
 
-
 @router.post("/{draft_id}/approve", response_model=schemas.DraftOut)
 async def approve_draft(draft_id: int, payload: schemas.ApproveInput, db: Session = Depends(get_db)):
     draft = db.get(models.Draft, draft_id)
@@ -60,10 +65,9 @@ async def approve_draft(draft_id: int, payload: schemas.ApproveInput, db: Sessio
     db.add(draft); db.commit(); db.refresh(draft)
     return draft
 
-
 @router.get("/{draft_id}", response_model=schemas.DraftOut)
 async def get_draft(draft_id: int, db: Session = Depends(get_db)):
     draft = db.get(models.Draft, draft_id)
     if not draft:
         raise RuntimeError("not found")
-    return draft
+    return draft