Spaces:
Sleeping
Sleeping
File size: 2,077 Bytes
9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
from fastapi import Form, APIRouter, File, UploadFile, HTTPException, Request
from db.repository import get_db_conn
from db.get_data import GetDatabase
from db.save_data import InsertDatabase
from config import MYSQL_CONFIG
from api.function import data_ingestion, get_data, delete_data, update_data
from script.vector_db import IndexManager
from service.dto import MetadataRequest
router = APIRouter(tags=["Topics"])
db_conn = get_db_conn(MYSQL_CONFIG)
get_database = GetDatabase(db_conn)
index_manager = IndexManager()
@router.post("/topic")
async def upload_file(
title: str = Form(...),
author: str = Form(...),
category: str = Form(...),
year: int = Form(...),
publisher: str = Form(...),
file: UploadFile = File(...),
# content_table: UploadFile = File(...)
):
reference = {
"title": title,
"author": author,
"category": category,
"year": year,
"publisher": publisher,
}
# response = await data_ingestion(db_conn, reference, file, content_table)
response = await data_ingestion(db_conn, reference, file)
return {"filename": file.filename, "response": response}
@router.get("/topic")
async def get_metadata():
results = await get_data(db_conn)
return results
@router.put("/topic/{id}")
async def update_metadata(id: int, reference: MetadataRequest):
try :
old_reference = await get_database.get_data_by_id(id)
index_manager.update_vector_database(old_reference, reference)
return await update_data(id, reference, db_conn)
except Exception as e:
raise HTTPException(status_code=500, detail="An error occurred while updating metadata")
@router.delete("/topic/{id}")
async def delete_metadata(id: int):
try:
old_reference = await get_database.get_data_by_id(id)
index_manager.delete_vector_database(old_reference)
return await delete_data(id, db_conn)
except Exception as e:
print(e)
raise HTTPException(status_code=500, detail="An error occurred while delete metadata")
|