headlines

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 from routes.category import router  # routes.py must be in same folder
 from routes.question import askMe
+from routes.headlines import headlines
 from dotenv import load_dotenv
 from cache_init import fetch_and_cache_articles
 from fastapi.middleware.cors import CORSMiddleware
@@ -11,6 +12,7 @@ load_dotenv()
 app = FastAPI()
 app.include_router(router)
 app.include_router(askMe)
+app.include_router(headlines)
 
 app.add_middleware(
     CORSMiddleware,

nuse_modules/fetchHeadlines.py

@@ -0,0 +1,26 @@
+from nuse_modules.google_search import search_google_news
+from models_initialization.mistral_registry import mistral_generate
+
+def summarize_headlines_with_mistral(headlines: list[dict]) -> str:
+    prompt = (
+        "Summarize the following news headlines into a short 3-sentence digest. "
+        "Be factual and neutral. Mention the sources.\n\n"
+    )
+    for item in headlines:
+        prompt += f"- {item['title']} – {item.get('link', '')}\n"
+
+    return mistral_generate(prompt, max_new_tokens=200, temperature=0.5)
+
+
+def generate_headline_digest(keywords: list[str], num_results: int = 5):
+    articles = search_google_news(keywords, num_results=num_results)
+
+    if isinstance(articles, dict) and "error" in articles:
+        return {"error": articles["error"]}
+
+    summary = summarize_headlines_with_mistral(articles)
+    return {
+        "summary": summary,
+        "sources": list({article["link"] for article in articles}),
+        "headlines": [article["title"] for article in articles]
+    }

routes/headlines.py

@@ -0,0 +1,10 @@
+from fastapi import APIRouter, Query
+from nuse_modules.fetchHeadlines import generate_headline_digest
+
+headlines = APIRouter()
+
+@headlines.get("/headlines")
+def get_headlines(q: str = Query("India", description="Comma-separated keywords to search")):
+    keywords = [kw.strip() for kw in q.split(",")]
+    result = generate_headline_digest(keywords)
+    return result