-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
62 lines (51 loc) · 2.24 KB
/
server.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
57
58
59
60
61
62
from flask import Flask, request
from flask_cors import CORS, cross_origin
from factchecker import CheckFact, initialiseES
from os.path import exists
import requests
import zipfile
import shutil
from tqdm import tqdm
from dotenv import load_dotenv
import os
load_dotenv()
def download_file(path, link):
response = requests.get(link, stream=True)
total = int(response.headers.get('content-length', 0))
with open(path, 'wb') as file, tqdm(
desc=path,
total=total,
unit='iB',
unit_scale=True,
unit_divisor=1024,
) as bar:
for data in response.iter_content(chunk_size=1024):
size = file.write(data)
bar.update(size)
if not os.path.exists(os.environ.get("MODEL_EXTRACTION_PATH")):
os.makedirs(os.environ.get("MODEL_EXTRACTION_PATH"))
if(not exists(os.environ.get("MODEL_PATH"))):
download_file(os.environ.get("MODEL_PATH"), os.environ.get("MODEL_LINK"))
with zipfile.ZipFile(os.environ.get("MODEL_PATH"), 'r') as zip_ref:
zip_ref.extractall(os.environ.get("MODEL_EXTRACTION_PATH"))
if os.path.exists(os.path.join(os.environ.get("MODEL_EXTRACTION_PATH"), "__MACOSX/")):
shutil.rmtree(os.path.join(os.environ.get("MODEL_EXTRACTION_PATH"), "__MACOSX/"))
for file in os.listdir(os.environ.get("MODEL_EXTRACTION_PATH")):
if not file.endswith(".zip") and not file.endswith(".DS_Store") and not file.endswith(".json"):
os.rename(os.path.join(os.environ.get("MODEL_EXTRACTION_PATH"), file), os.environ.get("EXTRACTED_MODEL_PATH"))
if(not exists(os.environ.get("CORPUS_PATH"))):
download_file(os.environ.get("CORPUS_PATH"), os.environ.get("CORPUS_LINK"))
initialiseES()
app = Flask(__name__, static_url_path='', static_folder=os.environ.get("FRONTEND_PATH"))
CORS(app, support_credentials=True)
@app.route("/myapi/fact_check", methods=['GET'])
@cross_origin(supports_credentials=True)
def explanation():
claim = request.args.get('claim')
mode = request.args.get('mode')
if(mode == "google" or mode == "stored" ):
return CheckFact(claim, mode)
else:
return {"claim": claim, "explanations":[], "evidence": [], "status": "Invalid mode selected."}
if(__name__ == "__main__"):
app.run(debug=True)