Update routes.py
Browse files
routes.py
CHANGED
@@ -1,12 +1,22 @@
|
|
1 |
from fastapi import APIRouter, Body ,HTTPException
|
2 |
-
from models import PageSpeedURLRequest
|
3 |
from config import GEMINI_API_KEY, PAGESPEED_API_KEY
|
4 |
from pagespeed_fetch import get_pagespeed_data
|
|
|
5 |
from gemini_report import generate_report_with_gemini
|
6 |
|
7 |
|
8 |
router = APIRouter()
|
9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
@router.post("/generate-report")
|
11 |
async def generate_report(data: dict = Body(...)):
|
12 |
report = generate_report_with_gemini(data, GEMINI_API_KEY)
|
|
|
1 |
from fastapi import APIRouter, Body ,HTTPException
|
2 |
+
from models import PageSpeedURLRequest, ReportIngestRequest
|
3 |
from config import GEMINI_API_KEY, PAGESPEED_API_KEY
|
4 |
from pagespeed_fetch import get_pagespeed_data
|
5 |
+
from vectorstore_manager import ingest_report
|
6 |
from gemini_report import generate_report_with_gemini
|
7 |
|
8 |
|
9 |
router = APIRouter()
|
10 |
|
11 |
+
@router.post("/ingest-report")
|
12 |
+
async def ingest_user_report(request: ReportIngestRequest):
|
13 |
+
try:
|
14 |
+
vectorstore, retriever = ingest_report(request.user_id, request.report_text)
|
15 |
+
return {"status": "success", "message": f"Report ingested for user {request.user_id}"}
|
16 |
+
except Exception as e:
|
17 |
+
raise HTTPException(status_code=500, detail=str(e))
|
18 |
+
|
19 |
+
|
20 |
@router.post("/generate-report")
|
21 |
async def generate_report(data: dict = Body(...)):
|
22 |
report = generate_report_with_gemini(data, GEMINI_API_KEY)
|