Server/kobart_server.py at master · PaperHelper/Server · GitHub

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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from flask import Flask, request, jsonify,make_response
import json
from paper_summarization import main
from data_crawling import get_top_10_papers,get_files,get_random_papers
import os
from threading import Thread

#from flask_restx import Api, Resource

app = Flask(__name__)


@app.route('/interest', methods=['POST'])
def postInterest():
    global interest

    data = request.get_json(silent=True, cache=False, force=True)
    print("Received data:", data)

    interest = data['interest']
    for key in cs.keys():
        interest = interest.replace(key, cs[key])

    print("Interests:", interest)
    return jsonify(interest)

def send_summary():

    papers = os.listdir('./paper_kor')
    papers = [f for f in papers if f.endswith('.txt')]
    paper = dict()

    for i,f in enumerate(papers):

        summary = open(f'./kobart_summarized/summarized_{f}','r').read()
        data = open(f'./kobart_paper_data/data_{f}','r').readlines()
        data = [d.strip() for d in data]

        paper['title'+str(i)] = data[0]
        paper['author'+str(i)] = data[1]
        paper['publication'+str(i)] = data[2]
        paper['year'+str(i)] = data[3]
        paper['summary'+str(i)] = summary
        paper['pdf'+str(i)] = data[4]
        paper['tag'+str(i)] = data[5]

    return paper


@app.route('/paper', methods=['GET'])
def getPaper():
    paper = {}

    fields = interest.strip().split('\t')
    print(fields)

    if os.path.exists('./kobart_summarized/'):
        paper = send_summary()

    else:
        for i in range(10):
            paper['title'+str(i)] = 'please wait'
            paper['author'+str(i)] = ''
            paper['publication'+str(i)] = ''
            paper['year'+str(i)] = ''
            paper['summary'+str(i)] = ''
            paper['pdf'+str(i)] = ''
            paper['tag'+str(i)] = ''

    response = make_response(json.dumps(paper,ensure_ascii=False).encode('utf-8'))
    print(paper)
    return response

if __name__ == "__main__":
    # paper = send_summary()
    # print(paper)
    app.run(debug=True, host="163.239.28.25", port=5000)