Update main.py

main.py

@@ -1,9 +1,12 @@
-from fastapi import FastAPI, HTTPException, Query
+from fastapi import FastAPI, HTTPException, Query, Depends
 from pydantic import BaseModel
 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
 
 app = FastAPI()
 
@@ -68,6 +71,20 @@ 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
+
+# Endpoint to get trending searches for a specific country
+@app.get("/trending-searches/{country_code}")
+async def get_trending_searches(country_code: str):
+    return fetch_trending_searches(country_code.upper())
+
+
 @app.get("/", tags=["Home"])
 def api_home():
     return {'detail': 'Welcome to Web-Scraping API! Visit https://pvanand-web-scraping.hf.space/docs to test'}