Hammad712 commited on
Commit
0ed2471
·
verified ·
1 Parent(s): b15610e

Update routes.py

Browse files
Files changed (1) hide show
  1. routes.py +11 -1
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)