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.
|