udpate

backend/routes.py

@@ -8,8 +8,8 @@ from fastapi.responses import FileResponse
 
 from config import FORMAT_META, STORAGE_DIR
 from database import create_guide, delete_guide, get_guide, list_guides
-from generator import generate_guide, cancel_guide
-from models import GuideCreateRequest, GuideResponse
+from generator import generate_guide, rework_guide, cancel_guide
+from models import GuideCreateRequest, GuideReworkRequest, GuideResponse
 
 router = APIRouter(prefix="/api")
 
@@ -26,6 +26,7 @@ async def create(req: GuideCreateRequest):
         "id": str(uuid.uuid4()),
         "topic": req.topic.strip(),
         "format": req.format,
+        "instructions": req.instructions.strip(),
         "status": "queued",
         "progress": None,
         "html_path": None,
@@ -34,7 +35,7 @@ async def create(req: GuideCreateRequest):
         "updated_at": now,
     }
     await create_guide(guide)
-    asyncio.create_task(generate_guide(guide["id"], guide["topic"], guide["format"]))
+    asyncio.create_task(generate_guide(guide["id"], guide["topic"], guide["format"], guide["instructions"]))
     return guide
 
 
@@ -64,6 +65,17 @@ async def download_html(guide_id: str):
     return FileResponse(path, filename=f"{guide['topic']}-{guide['format']}.html", media_type="text/html")
 
 
+@router.post("/guides/{guide_id}/rework")
+async def rework(guide_id: str, req: GuideReworkRequest):
+    guide = await get_guide(guide_id)
+    if guide is None:
+        raise HTTPException(404, "Guide nicht gefunden")
+    if guide["status"] != "done":
+        raise HTTPException(400, "Guide muss fertig sein")
+    asyncio.create_task(rework_guide(guide_id, guide["topic"], guide["format"], req.instructions.strip()))
+    return {"ok": True}
+
+
 @router.post("/guides/{guide_id}/cancel")
 async def cancel(guide_id: str):
     cancelled = await cancel_guide(guide_id)