Sabbah13 commited on
Commit
8fb0be5
1 Parent(s): 2285943

Added env variables

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import requests
2
  import base64
 
3
  import json
4
  import streamlit as st
5
  from speechlib import Transcriptor
@@ -30,7 +31,7 @@ if uploaded_file is not None:
30
 
31
  log_folder = "logs"
32
  language = "ru"
33
- modelSize = "medium"
34
  voices_folder = ""
35
  quantization = False
36
 
@@ -49,35 +50,35 @@ if uploaded_file is not None:
49
  auth_str = f'{username}:{password}'
50
  auth_bytes = auth_str.encode('utf-8')
51
  auth_base64 = base64.b64encode(auth_bytes).decode('utf-8')
52
- url = 'https://ngw.devices.sberbank.ru:9443/api/v2/oauth'
53
 
54
  headers = {
55
  'Authorization': f'Basic {auth_base64}', # вставляем базовую авторизацию
56
- 'RqUID': '6f0b1291-c7f3-43c6-bb2e-9f3efb2dc98f',
57
  'Content-Type': 'application/x-www-form-urlencoded',
58
  'Accept': 'application/json'
59
  }
60
 
61
  data = {
62
- 'scope': 'GIGACHAT_API_PERS'
63
  }
64
 
65
  response = requests.post(url, headers=headers, data=data, verify=False)
66
  access_token = response.json()['access_token']
67
  print('Got access token')
68
 
69
- url_completion = "https://gigachat.devices.sberbank.ru/api/v1/chat/completions"
70
 
71
  data_copm = json.dumps({
72
- "model": "GigaChat",
73
  "messages": [
74
  {
75
  "role": "user",
76
- "content": "Напиши резюме транскрибации звонка, текст которого приложен в ниже. Выдели самостоятельно цель встречи, потом описать ключевые моменты всей встречи. Потом выделить отдельные темы звонка и выделить ключевые моменты в них. Напиши итоги того, о чем договорились говорящие, если такое возможно выделить из текста. Транскрибация: " + transcript
77
  }
78
  ],
79
  "stream": False,
80
- "max_tokens": 1024,
81
  })
82
 
83
  headers_comp = {
 
1
  import requests
2
  import base64
3
+ import os
4
  import json
5
  import streamlit as st
6
  from speechlib import Transcriptor
 
31
 
32
  log_folder = "logs"
33
  language = "ru"
34
+ modelSize = os.getenv('WHISPER_MODEL_SIZE')
35
  voices_folder = ""
36
  quantization = False
37
 
 
50
  auth_str = f'{username}:{password}'
51
  auth_bytes = auth_str.encode('utf-8')
52
  auth_base64 = base64.b64encode(auth_bytes).decode('utf-8')
53
+ url = os.getenv('GIGA_AUTH_URL')
54
 
55
  headers = {
56
  'Authorization': f'Basic {auth_base64}', # вставляем базовую авторизацию
57
+ 'RqUID': os.getenv('GIGA_rquid'),
58
  'Content-Type': 'application/x-www-form-urlencoded',
59
  'Accept': 'application/json'
60
  }
61
 
62
  data = {
63
+ 'scope': os.getenv('GIGA_SCOPE')
64
  }
65
 
66
  response = requests.post(url, headers=headers, data=data, verify=False)
67
  access_token = response.json()['access_token']
68
  print('Got access token')
69
 
70
+ url_completion = os.getenv('GIGA_COMPLETION_URL')
71
 
72
  data_copm = json.dumps({
73
+ "model": os.getenv('GIGA_MODEL'),
74
  "messages": [
75
  {
76
  "role": "user",
77
+ "content": os.getenv('GIGA_BASE_PROMPT') + transcript
78
  }
79
  ],
80
  "stream": False,
81
+ "max_tokens": os.getenv('GIGA_MAX_TOKENS'),
82
  })
83
 
84
  headers_comp = {