muryshev commited on
Commit
a5177da
·
1 Parent(s): 1e3bb2e
Files changed (2) hide show
  1. app.py +2 -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.