Viewing File: /home/ubuntu/codegamaai-test/broker_bot/app.py

import os
from fastapi.middleware.cors import CORSMiddleware
from fastapi import FastAPI,Form,BackgroundTasks
from pydantic import BaseModel
from fastapi.responses import JSONResponse, FileResponse
from supertokens_fastapi import get_cors_allowed_headers
from fastapi import File, UploadFile, HTTPException
import uvicorn
import shutil
from fastapi import FastAPI
from typing import Union
from threading import Thread
import uuid
from src.constants import *
from src.utils import *
from src.knowledge_v2 import *

def main_text(text,number):
    kB_res = KB_main(text,number)
    return kB_res

app = FastAPI()
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"] + get_cors_allowed_headers(),
)


@app.post("/api/v1/query")
async def query(data: dict):
    question = data['question']
    user_uid = data['user_uid']

    response_message = main_text(question,user_uid)
    response_data = {"response": response_message}

    return JSONResponse(content=response_data)


if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0",port=5006)