update
Browse files- app.py +2 -2
- lib/llm_2.py +16 -7
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import os
|
2 |
import uuid
|
3 |
import json
|
4 |
-
from flask import Flask, request, jsonify
|
5 |
import pytesseract
|
6 |
from pdf2image import convert_from_bytes
|
7 |
from flask_cors import CORS
|
@@ -61,7 +61,7 @@ def analize():
|
|
61 |
text_data = request.json.get('text')
|
62 |
app_info = llm_2.getApplicationInfo(text_data)
|
63 |
|
64 |
-
return json.dumps(app_info, sort_keys=False)
|
65 |
|
66 |
if __name__ == '__main__':
|
67 |
app.run(debug=False)
|
|
|
1 |
import os
|
2 |
import uuid
|
3 |
import json
|
4 |
+
from flask import Flask, request, jsonify, Response
|
5 |
import pytesseract
|
6 |
from pdf2image import convert_from_bytes
|
7 |
from flask_cors import CORS
|
|
|
61 |
text_data = request.json.get('text')
|
62 |
app_info = llm_2.getApplicationInfo(text_data)
|
63 |
|
64 |
+
return Response(json.dumps(app_info, sort_keys=False, ensure_ascii=False), content_type='application/json; charset=utf-8')
|
65 |
|
66 |
if __name__ == '__main__':
|
67 |
app.run(debug=False)
|
lib/llm_2.py
CHANGED
@@ -171,7 +171,7 @@ def getSector(application) :
|
|
171 |
if verbose :
|
172 |
print(i, ':', sector)
|
173 |
|
174 |
-
return sector
|
175 |
|
176 |
def getProduct(application, sector) :
|
177 |
product = ''
|
@@ -196,7 +196,7 @@ def getProduct(application, sector) :
|
|
196 |
if verbose :
|
197 |
print(product)
|
198 |
|
199 |
-
return product
|
200 |
|
201 |
def getProblem(application, sector, product) :
|
202 |
problem = ''
|
@@ -220,7 +220,7 @@ def getProblem(application, sector, product) :
|
|
220 |
if verbose :
|
221 |
print(problem)
|
222 |
|
223 |
-
return problem
|
224 |
|
225 |
def getAuthor(application) :
|
226 |
prefix = '''Ты мой помощник. Ты отвечаешь только на РУССКОМ языке. Ты не отвечаешь на вопросы, не комментируешь,
|
@@ -416,9 +416,9 @@ def getPersons(application) :
|
|
416 |
return names
|
417 |
|
418 |
def getApplicationInfo(application) :
|
419 |
-
sector = getSector(application)
|
420 |
-
product = getProduct(application, sector)
|
421 |
-
problem = getProblem(application, sector, product)
|
422 |
author = getAuthor(application)
|
423 |
persons = getPersons(application)
|
424 |
contractdata = getContractData(application)
|
@@ -440,6 +440,15 @@ def getApplicationInfo(application) :
|
|
440 |
app_info['Указан ли в обращении номер договора?'] = ifcontract
|
441 |
app_info['Жалоба касается комиссии за обслуживание рублевого счета?'] = ifcomission
|
442 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
443 |
if verbose :
|
444 |
print()
|
445 |
print(i)
|
@@ -454,7 +463,7 @@ def getApplicationInfo(application) :
|
|
454 |
print('Указан ли в обращении номер договора?', ifcontract)
|
455 |
print('Жалоба касается комиссии за обслуживание рублевого счета?', ifcomission)
|
456 |
|
457 |
-
return app_info
|
458 |
|
459 |
# application = '''Я, Кристенгоф Оксана Михайловна, заключила кредитный договор с ПАО Камабанк № 59875 и № 456835645645654
|
460 |
# и договор на выдачу кредитной карты с ПАО "БСТ-Банк" №4567652124534235 и №4561254153412.
|
|
|
171 |
if verbose :
|
172 |
print(i, ':', sector)
|
173 |
|
174 |
+
return sector, prompt, response
|
175 |
|
176 |
def getProduct(application, sector) :
|
177 |
product = ''
|
|
|
196 |
if verbose :
|
197 |
print(product)
|
198 |
|
199 |
+
return product, prompt, response
|
200 |
|
201 |
def getProblem(application, sector, product) :
|
202 |
problem = ''
|
|
|
220 |
if verbose :
|
221 |
print(problem)
|
222 |
|
223 |
+
return problem, prompt, response
|
224 |
|
225 |
def getAuthor(application) :
|
226 |
prefix = '''Ты мой помощник. Ты отвечаешь только на РУССКОМ языке. Ты не отвечаешь на вопросы, не комментируешь,
|
|
|
416 |
return names
|
417 |
|
418 |
def getApplicationInfo(application) :
|
419 |
+
sector, sector_prompt, sector_response = getSector(application)
|
420 |
+
product, product_prompt, product_response = getProduct(application, sector)
|
421 |
+
problem, problem_prompt, problem_response = getProblem(application, sector, product)
|
422 |
author = getAuthor(application)
|
423 |
persons = getPersons(application)
|
424 |
contractdata = getContractData(application)
|
|
|
440 |
app_info['Указан ли в обращении номер договора?'] = ifcontract
|
441 |
app_info['Жалоба касается комиссии за обслуживание рублевого счета?'] = ifcomission
|
442 |
|
443 |
+
debug_info = {
|
444 |
+
'sector_prompt': sector_prompt,
|
445 |
+
'sector_response': sector_response,
|
446 |
+
'product_prompt': product_prompt,
|
447 |
+
'product_response': product_response,
|
448 |
+
'problem_prompt': problem_prompt,
|
449 |
+
'problem_response': problem_response,
|
450 |
+
}
|
451 |
+
|
452 |
if verbose :
|
453 |
print()
|
454 |
print(i)
|
|
|
463 |
print('Указан ли в обращении номер договора?', ifcontract)
|
464 |
print('Жалоба касается комиссии за обслуживание рублевого счета?', ifcomission)
|
465 |
|
466 |
+
return {'application': app_info, 'debug': debug_info}
|
467 |
|
468 |
# application = '''Я, Кристенгоф Оксана Михайловна, заключила кредитный договор с ПАО Камабанк № 59875 и № 456835645645654
|
469 |
# и договор на выдачу кредитной карты с ПАО "БСТ-Банк" №4567652124534235 и №4561254153412.
|