Add FastAPI application with pytrends endpoints

app.py

@@ -0,0 +1,143 @@
+from fastapi import FastAPI, Query
+from pytrends.request import TrendReq
+from fastapi.responses import JSONResponse
+from typing import List, Optional
+import pandas as pd
+import json
+
+app = FastAPI(
+    title="PyTrends API",
+    description="HTTP API for Google Trends data",
+    version="1.0.0"
+)
+
+# Initialize pytrends
+pytrends = TrendReq(hl="en-US", tz=360)
+
+@app.get("/")
+def root():
+    return {
+        "message": "PyTrends API",
+        "endpoints": {
+            "/interest_over_time": "Get interest over time for keywords",
+            "/interest_by_region": "Get interest by region for keywords",
+            "/trending_searches": "Get trending searches for a country",
+            "/related_queries": "Get related queries for keywords"
+        }
+    }
+
+@app.get("/interest_over_time")
+def get_interest_over_time(
+    kw: List[str] = Query(..., description="Keywords to search"),
+    timeframe: str = Query("today 5-y", description="Timeframe for search"),
+    geo: str = Query("", description="Geographic location"),
+    gprop: str = Query("", description="Google property (images, news, youtube, etc.)"),
+    cat: int = Query(0, description="Category")
+):
+    try:
+        pytrends.build_payload(
+            kw_list=kw,
+            timeframe=timeframe,
+            geo=geo,
+            gprop=gprop,
+            cat=cat
+        )
+        df = pytrends.interest_over_time()
+        df = df.drop('isPartial', axis=1)
+        data = df.reset_index().to_dict('records')
+        # Convert datetime to string for JSON serialization
+        for record in data:
+            if 'date' in record:
+                record['date'] = str(record['date'])
+        return {
+            "data": data,
+            "keywords": kw,
+            "timeframe": timeframe,
+            "geo": geo
+        }
+    except Exception as e:
+        return JSONResponse(
+            status_code=400,
+            content={"error": str(e)}
+        )
+
+@app.get("/interest_by_region")
+def get_interest_by_region(
+    kw: List[str] = Query(..., description="Keywords to search"),
+    timeframe: str = Query("today 5-y", description="Timeframe for search"),
+    geo: str = Query("", description="Geographic location"),
+    gprop: str = Query("", description="Google property"),
+    resolution: str = Query("country", description="Resolution (country, region, metro, city)")
+):
+    try:
+        pytrends.build_payload(
+            kw_list=kw,
+            timeframe=timeframe,
+            geo=geo,
+            gprop=gprop
+        )
+        df = pytrends.interest_by_region(resolution=resolution)
+        data = df.reset_index().to_dict('records')
+        return {
+            "data": data,
+            "keywords": kw,
+            "resolution": resolution
+        }
+    except Exception as e:
+        return JSONResponse(
+            status_code=400,
+            content={"error": str(e)}
+        )
+
+@app.get("/trending_searches")
+def get_trending_searches(
+    country: str = Query("united_states", description="Country code (e.g., united_states, india, etc.)")
+):
+    try:
+        df = pytrends.trending_searches(pn=country)
+        data = df.values.tolist()
+        return {
+            "trending": data,
+            "country": country
+        }
+    except Exception as e:
+        return JSONResponse(
+            status_code=400,
+            content={"error": str(e)}
+        )
+
+@app.get("/related_queries")
+def get_related_queries(
+    kw: List[str] = Query(..., description="Keywords to search"),
+    timeframe: str = Query("today 5-y", description="Timeframe for search"),
+    geo: str = Query("", description="Geographic location")
+):
+    try:
+        pytrends.build_payload(
+            kw_list=kw,
+            timeframe=timeframe,
+            geo=geo
+        )
+        related = pytrends.related_queries()
+        result = {}
+        for kw_item in kw:
+            if kw_item in related:
+                top_queries = related[kw_item]['top']
+                rising_queries = related[kw_item]['rising']
+                result[kw_item] = {
+                    "top": top_queries.reset_index().to_dict('records') if top_queries is not None else [],
+                    "rising": rising_queries.reset_index().to_dict('records') if rising_queries is not None else []
+                }
+        return {
+            "related_queries": result,
+            "keywords": kw
+        }
+    except Exception as e:
+        return JSONResponse(
+            status_code=400,
+            content={"error": str(e)}
+        )
+
+@app.get("/health")
+def health_check():
+    return {"status": "healthy"}