From fd7d73e9c38c9f925ff42691062e45b4b3d6128b Mon Sep 17 00:00:00 2001
From: Marty Oehme <contact@martyoeh.me>
Date: Thu, 5 Jun 2025 15:12:45 +0200
Subject: [PATCH] Add basic joke improvements with Groq

---
 prophet/app.py | 34 ++++++++++++++++++++++++++++++++++
 pyproject.toml |  5 ++++-
 uv.lock        | 32 +++++++++++++++++++++++++++++++-
 3 files changed, 69 insertions(+), 2 deletions(-)

diff --git a/prophet/app.py b/prophet/app.py
index 09a6807..ba55005 100644
--- a/prophet/app.py
+++ b/prophet/app.py
@@ -1,5 +1,35 @@
+import os
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
+from groq import Groq
+def improve_with_groq(original: str) -> str:
+    client = Groq(api_key=os.getenv("GROQ_API_KEY", "NO_API_KEY_FOUND"))
+
+    suggestions = client.chat.completions.create(
+        messages=[
+            {
+                "role": "user",
+                "content": f"Improve on the following satirical headline. The headline should be funny, can involve current political events and should have an edge to it. Print only the suggestions, with one suggestion on each line.\nOriginal: '{original}'",
+            }
+        ],
+        model="llama-3.3-70b-versatile",
+    )
+    print("Suggestions: ", suggestions.choices[0].message.content)
+    winner = client.chat.completions.create(
+        messages=[
+            {
+                "role": "user",
+                "content": f"From the following satirical headline suggestions, pick the strongest one and print it. Do not print anything else. The chosen suggestion should be the most edgy, able to trend on social media and funniest. The suggestions:\n\n{suggestions}",
+            }
+        ],
+        model="llama-3.3-70b-versatile",
+    )
+    print("Winner: ", winner.choices[0].message.content)
+    winner_str = winner.choices[0].message.content
+    if not winner_str:
+        raise ValueError
+    return winner_str
+
 
 app = FastAPI()
 
@@ -18,6 +48,10 @@ app.add_middleware(
 
 
 @app.get("/test")
+@app.get("/improve")
+def improve_headline(content: str):
+    return improve_with_groq(content)
+
 def read_root():
     response = {
         "data": [
diff --git a/pyproject.toml b/pyproject.toml
index 3f99e58..912aeb5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,10 @@ version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"
-dependencies = ["fastapi[standard]>=0.115.12"]
+dependencies = [
+    "fastapi[standard]>=0.115.12",
+    "groq>=0.26.0",
+]
 
 [dependency-groups]
 dev = ["mypy>=1.16.0", "ruff>=0.11.12"]
diff --git a/uv.lock b/uv.lock
index 0ad9bda..5035527 100644
--- a/uv.lock
+++ b/uv.lock
@@ -54,6 +54,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335, upload-time = "2022-10-25T02:36:20.889Z" },
 ]
 
+[[package]]
+name = "distro"
+version = "1.9.0"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/fc/f8/98eea607f65de6527f8a2e8885fc8015d3e6f5775df186e443e0964a11c3/distro-1.9.0.tar.gz", hash = "sha256:2fa77c6fd8940f116ee1d6b94a2f90b13b5ea8d019b98bc8bafdcabcdd9bdbed", size = 60722, upload-time = "2023-12-24T09:54:32.31Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/12/b3/231ffd4ab1fc9d679809f356cebee130ac7daa00d6d6f3206dd4fd137e9e/distro-1.9.0-py3-none-any.whl", hash = "sha256:7bffd925d65168f85027d8da9af6bddab658135b840670a223589bc0c8ef02b2", size = 20277, upload-time = "2023-12-24T09:54:30.421Z" },
+]
+
 [[package]]
 name = "dnspython"
 version = "2.7.0"
@@ -119,6 +128,23 @@ standard = [
     { name = "uvicorn", extra = ["standard"] },
 ]
 
+[[package]]
+name = "groq"
+version = "0.26.0"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "anyio" },
+    { name = "distro" },
+    { name = "httpx" },
+    { name = "pydantic" },
+    { name = "sniffio" },
+    { name = "typing-extensions" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/b6/68/a820a22937e4a2f48b7a60e1aaf7948fb57d1c124072829b5cc06a01cfa0/groq-0.26.0.tar.gz", hash = "sha256:1f1e50d26c6134f6fb580ea7002e8f9ff5c7c1685c9e0f50d71adecd039ae5d4", size = 128500, upload-time = "2025-05-29T18:25:23.332Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/da/2e/14bef74aa760199c7a179c753953512d1aa1ed264f7477f6bd6fe9b9cff3/groq-0.26.0-py3-none-any.whl", hash = "sha256:4dc0900d506876ea39a9aa1985f12a51859bacf486fe939664248eff1f451af3", size = 129572, upload-time = "2025-05-29T18:25:22.077Z" },
+]
+
 [[package]]
 name = "h11"
 version = "0.16.0"
@@ -285,6 +311,7 @@ version = "0.1.0"
 source = { editable = "." }
 dependencies = [
     { name = "fastapi", extra = ["standard"] },
+    { name = "groq" },
 ]
 
 [package.dev-dependencies]
@@ -294,7 +321,10 @@ dev = [
 ]
 
 [package.metadata]
-requires-dist = [{ name = "fastapi", extras = ["standard"], specifier = ">=0.115.12" }]
+requires-dist = [
+    { name = "fastapi", extras = ["standard"], specifier = ">=0.115.12" },
+    { name = "groq", specifier = ">=0.26.0" },
+]
 
 [package.metadata.requires-dev]
 dev = [