update

backend/routes.py

@@ -18,7 +18,7 @@ from database import (
 from generator import (
     generate_guide, cancel_guide, chat_with_guide, guide_slot_dateien,
     generate_bausteine, cancel_bausteine, bausteine_status, active_bausteine, reset_bausteine,
-    generate_element, chat_with_element, check_element, style_element,
+    generate_element, chat_with_element, check_element, style_element, refine_suggestion,
 )
 from models import (
     GuideCreateRequest, GuideResponse,
@@ -27,6 +27,7 @@ from models import (
     GuideChatRequest, GuideChatResponse,
     ElementCreateRequest, ElementChatRequest, ElementChatResponse, ElementResponse,
     ElementUpdateRequest, ElementCheckRequest, ElementCheckResponse, ElementStyleResponse,
+    ElementRefineRequest, ElementRefineResponse,
     ProgressUpdate, ProgressResponse, ProjectResponse, ProviderInfo,
 )
 from paths import bausteine_path, bausteine_topics, guide_content_path, project_dir, topic_dir
@@ -288,12 +289,19 @@ async def element_chat(element_id: str, req: ElementChatRequest):
     element = await get_element(element_id)
     if element is None:
         raise HTTPException(404, "Element nicht gefunden")
-    reply, fields = await chat_with_element(element, [m.model_dump() for m in req.messages], provider=req.provider)
-    if fields:
-        now = datetime.now(timezone.utc).isoformat()
-        await update_element(element_id, **fields, updated_at=now)
-        element = await get_element(element_id)
-    return {"reply": reply, "element": element}
+    reply, changes = await chat_with_element(element, [m.model_dump() for m in req.messages], provider=req.provider)
+    return {"reply": reply, "changes": changes}
+
+
+@router.post("/elements/{element_id}/refine", response_model=ElementRefineResponse)
+async def element_refine(element_id: str, req: ElementRefineRequest):
+    element = await get_element(element_id)
+    if element is None:
+        raise HTTPException(404, "Element nicht gefunden")
+    change = await refine_suggestion(element, req.suggestion.model_dump(), req.instruction, provider=req.provider)
+    if change is None:
+        raise HTTPException(502, "Überarbeitung fehlgeschlagen — bitte erneut versuchen")
+    return {"change": change}
 
 
 @router.put("/elements/{element_id}", response_model=ElementResponse)