Spaces:
Runtime error
Runtime error
from fastapi import FastAPI, Request | |
from Flask import jsonify | |
from scripts import mongo_utils | |
from scripts import rag_utils | |
from dotenv import load_dotenv | |
import os | |
load_dotenv() | |
app = FastAPI | |
client = mongo_utils.connect_to_mongo() | |
print("Connected to MongoDB") | |
def captitalize_name(name): | |
name_split = name.split("_") | |
return " ".join([x.capitalize() for x in name_split]) | |
def summarize(request: Request): | |
pdf_path = request.body()['pdf_path'] | |
text = request.body()['text'] | |
vs = mongo_utils.get_vs(pdf_path, client) | |
summary = rag_utils.summ(vs, text) | |
return {'summary': summary} | |
def classify(request: Request): | |
pdf_path = request.body()['pdf_path'] | |
text = request.body()['text'] | |
vs = mongo_utils.get_vs(pdf_path, client) | |
decision = rag_utils.clf_seq(vs, text).lower() | |
return jsonify({'decision': decision}) | |
def options(request: Request): | |
pdf_path = request.body()['pdf_path'] | |
text = request.body()['text'] | |
vs = mongo_utils.get_vs(pdf_path, client) | |
options = eval(rag_utils.gen_options(vs, text)) | |
return jsonify({'options': options}) | |
def path(request: Request): | |
pdf_path = request.body()['pdf_path'] | |
text = request.body()['text'] | |
decision = request.body()['decision'] | |
vs = mongo_utils.get_vs(pdf_path, client) | |
path = rag_utils.gen_path(vs, text, decision) | |
return jsonify({'path': path}) | |
if __name__ == '__main__': | |
app.run() |