diff --git a/prophet/app.py b/prophet/app.py index 2d874d5..b9b03c2 100644 --- a/prophet/app.py +++ b/prophet/app.py @@ -162,6 +162,14 @@ def improve_headline(content: str): return rewrite_title_with_groq(content) +@app.get("/improve-summary") +def improve_summary(original_title: str, new_title: str, original_summary: str): + o = Original( + title=original_title, summary=original_summary, link="", date=datetime.now() + ) + return rewrite_summary_with_groq(o, new_title) + + def start() -> None: from uvicorn import run