Hammad712 commited on
Commit
36c093a
·
verified ·
1 Parent(s): a017a6e

Update routes.py

Browse files
Files changed (1) hide show
  1. routes.py +4 -3
routes.py CHANGED
@@ -1,4 +1,4 @@
1
- from fastapi import APIRouter
2
  from models import PageSpeedRequest
3
  from config import GEMINI_API_KEY
4
  from gemini_report import generate_report_with_gemini
@@ -6,6 +6,7 @@ from gemini_report import generate_report_with_gemini
6
  router = APIRouter()
7
 
8
  @router.post("/generate-report")
9
- async def generate_report(data: PageSpeedRequest):
10
- report = generate_report_with_gemini(data.__root__, GEMINI_API_KEY)
11
  return {"report": report}
 
 
1
+ from fastapi import APIRouter, Body
2
  from models import PageSpeedRequest
3
  from config import GEMINI_API_KEY
4
  from gemini_report import generate_report_with_gemini
 
6
  router = APIRouter()
7
 
8
  @router.post("/generate-report")
9
+ async def generate_report(data: dict = Body(...)):
10
+ report = generate_report_with_gemini(data, GEMINI_API_KEY)
11
  return {"report": report}
12
+