-
Notifications
You must be signed in to change notification settings - Fork 16
/
app.py
56 lines (39 loc) · 1.72 KB
/
app.py
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
import os
from flask import Flask, request, abort, make_response, jsonify
from flask_cors import CORS
from controllers import conversationController, legalController
from postgresql_db import database
# Flask Setup
app = Flask(__name__)
# DB Setup
db = database.connect(app, 'postgres', os.environ['POSTGRES_PASSWORD'], 'postgres')
# Cors Setup
CORS(app)
@app.route("/new", methods=['POST'])
def init_conversation():
init_request = request.get_json()
return conversationController.init_conversation(init_request['name'], init_request['person_type'])
@app.route("/conversation", methods=['POST'])
def chat():
chat_request = request.get_json()
return conversationController.receive_message(chat_request['conversation_id'], chat_request['message'])
@app.route("/conversation/<conversation_id>", methods=['GET'])
def get_conversation(conversation_id=None):
if conversation_id:
return conversationController.get_conversation(conversation_id)
else:
abort(make_response(jsonify(message="Invalid request"), 400))
@app.route("/conversation/<conversation_id>/files", methods=['GET', 'POST'])
def handle_files(conversation_id=None):
if conversation_id:
if request.method == 'GET':
return conversationController.get_file_list(conversation_id)
if request.method == 'POST':
if 'file' not in request.files:
abort(make_response(jsonify(message="No file provided"), 400))
return conversationController.upload_file(conversation_id, request.files['file'])
else:
abort(make_response(jsonify(message="Invalid request"), 400))
@app.route("/legal", methods=['GET'])
def get_legal_documents():
return legalController.get_legal_documents()