from flask import Flask,render_template,request,jsonify
from flask_cors import CORS
from checkvectordb import getdata,trainpdf1,loadIPCsections,loadmiyapurcourtorders
app = Flask(__name__)
CORS(app)

@app.get("/")
def index_get():
  print("just started...")
  #response=loadIPCsections("./uploads/ipc.json")
  #response=loadmiyapurcourtorders("./uploads/court_history.json")

  #print("------",response)
  return render_template("base.html")


@app.post("/trainpdf")
def trainpdf():
  try:
   pdf_file = request.files['file']
   response=trainpdf1(pdf_file)
   print("response :",response);
   message={"success":response}
  except Exception as we:
   message={"error":response}
   print("-error -", we)


  return response



@app.post("/predict")
def predict():
   
   text=request.get_json().get("message")
   print("from web ",text);
   response=getdata(text)
   message={"answer":response}
   print("message ",message)
   return message

if __name__ == "__main__":
        app.run(host='0.0.0.0', port=5000)