Add BadJoke GoodJoke data classes

prophet/app.py

@@ -1,7 +1,30 @@
 import os
+from dataclasses import dataclass
+from datetime import datetime
+from uuid import uuid4
+
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from groq import Groq
+
+
+@dataclass
+class Original:  # BadJoke: Sting
+    id: str  # should probably be a sha256sum of the title/link?
+    title: str
+    summary: str
+    link: str
+    date: datetime
+
+
+@dataclass
+class Improvement:  # GoodJoke: Queen
+    original: Original
+    title: str
+    summary: str
+    id: str = str(uuid4())
+
+
 def improve_with_groq(original: str) -> str:
     client = Groq(api_key=os.getenv("GROQ_API_KEY", "NO_API_KEY_FOUND"))
 
@@ -52,6 +75,7 @@ app.add_middleware(
 def improve_headline(content: str):
     return improve_with_groq(content)
 
+
 def read_root():
     response = {
         "data": [