Update main.py

main.py

@@ -4,9 +4,11 @@ import hrequests
 import trafilatura
 from fastapi.middleware.cors import CORSMiddleware
 from typing import Optional
-from functools import lru_cache
 from pytrends.request import TrendReq
 from datetime import datetime, timedelta
+from fastapi_cache import caches, close_caches
+from fastapi_cache.backends.inmemory import InMemoryCacheBackend
+from fastapi_cache.decorator import cache
 
 app = FastAPI()
 
@@ -71,18 +73,25 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Function to fetch real-time trending searches and cache results for 1 hour
-@lru_cache(maxsize=None)
-def fetch_trending_searches(country_code: str):
-    pytrends = TrendReq(hl='en-US', tz=360)
-    pytrends.build_payload(kw_list=[], timeframe='now 1-H')
-    trending_searches_df = pytrends.realtime_trending_searches(pn=country_code)
-    return trending_searches_df
+# Set up in-memory cache
+cache_backend = InMemoryCacheBackend()
+caches.set("default", cache_backend)
 
-# Endpoint to get trending searches for a specific country
-@app.get("/trending-searches/{country_code}")
-async def get_trending_searches(country_code: str = Query(..., description="ISO country code (e.g., US for United States)",enum=["US", "IN"])):
-    return fetch_trending_searches(country_code.upper())
+pytrends = TrendReq()
+
+@app.on_event("startup")
+async def on_startup():
+    pass
+
+@app.on_event("shutdown")
+async def on_shutdown():
+    await close_caches()
+
+@app.get("/trending_searches")
+@cache(expire=3600)  # Cache for 1 hour
+async def get_trending_searches(country: str = Query("US", description="Country code (e.g., 'US' for United States)")):
+    trending_searches = pytrends.realtime_trending_searches(pn=country)
+    return trending_searches.to_dict(orient="records")
 
 @app.get("/", tags=["Home"])
 def api_home():