redis setup and startup functionality

.env

@@ -1 +1,2 @@
-GNEWS_API_KEY=6c61f5da1b24fa83fbf964f8b280c438
+GNEWS_API_KEY=6c61f5da1b24fa83fbf964f8b280c438
+UPSTASH_REDIS_URL=rediss://:AU6-AAIjcDEyZDY4Njk3OGUwNzg0NTczODUxYmRmMDUyZDlmZWNiZXAxMA@clever-turtle-20158.upstash.io:6379

app.py

@@ -1,12 +1,17 @@
 from fastapi import FastAPI
 from routes import router  # routes.py must be in same folder
 from dotenv import load_dotenv
+from cache_init import fetch_and_cache_articles
 
 load_dotenv()
 
 app = FastAPI()
 app.include_router(router)
 
+@app.on_event("startup")
+async def startup_event():
+    fetch_and_cache_articles()
+
 @app.get("/health")
 def health_check():
     return {"status": "ok"}

cache_init.py

@@ -0,0 +1,59 @@
+# cache_init.py
+import os
+import redis
+import requests
+import uuid
+from dotenv import load_dotenv
+
+load_dotenv()
+
+CATEGORIES = [
+    "technology", "business", "science", "health", "world", "entertainment"
+]
+
+GNEWS_API_KEY = os.getenv("GNEWS_API_KEY")
+REDIS_URL = os.getenv("UPSTASH_REDIS_URL")
+
+r = redis.Redis.from_url(REDIS_URL, decode_responses=True)
+
+
+def fetch_and_cache_articles():
+    print("[INIT] Fetching and caching articles...")
+    for category in CATEGORIES:
+        base_url = "https://gnews.io/api/v4/top-headlines"
+        params = {
+        "topic": category,
+        "lang": "en",
+        "max": 5,
+        "expand": "content",
+        "token": GNEWS_API_KEY
+        }
+        try:
+            response = requests.get(base_url, params=params, timeout=10)
+            response.raise_for_status()
+            articles = response.json().get("articles", [])
+
+            for article in articles:
+                article_id = str(uuid.uuid4())
+                article_data = {
+                    "id": article_id,
+                    "title": article["title"],
+                    "url": article["url"],
+                    "description": article.get("description"),
+                    "content": article.get("content"),
+                    "image": article.get("image"),
+                    "publishedAt": article["publishedAt"],
+                    "category": category,
+                    "source": article["source"]["name"]
+                }
+                r.hset(f"article:{article_id}", mapping=article_data)
+                r.sadd(f"category:{category}", article_id)
+
+        except Exception as e:
+            print(f"[ERROR] Failed for category {category}: {e}")
+
+    print("[INIT] Article caching complete.")
+
+
+if __name__ == "__main__":
+    fetch_and_cache_articles()

requirements.txt

@@ -2,3 +2,4 @@ fastapi
 uvicorn[standard]
 requests
 python-dotenv
+redis