fruitpicker01 commited on
Commit
249f265
·
verified ·
1 Parent(s): f8f69c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -15,9 +15,9 @@ gc_key = os.getenv('GC_KEY')
15
  token = os.getenv('GITHUB_TOKEN')
16
 
17
  # Авторизация в сервисе GigaChat
18
- chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=60, verify_ssl_certs=False)
19
- chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=60, verify_ssl_certs=False)
20
- chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=60, verify_ssl_certs=False)
21
 
22
  # Загрузка данных из Excel-файла
23
  try:
@@ -85,7 +85,7 @@ def generate_message_gpt4o(prompt):
85
  data = {
86
  "model": "chatgpt-4o-latest",
87
  "messages": [{"role": "system", "content": prompt}],
88
- "max_tokens": 110
89
  }
90
  response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
91
  response_data = response.json()
@@ -135,26 +135,26 @@ def generate_messages(description, advantages, *selected_values):
135
 
136
  results["gpt4o"] = generate_message_gpt4o(standard_prompt)
137
  gpt4o_length = len(results["gpt4o"])
138
- gpt4o_display = f"{results['gpt4o']}\n\n------\nКоличество знаков: {gpt4o_length}"
139
  yield results["prompt"], gpt4o_display, "", "", "", "Сообщение GPT-4o сгенерировано"
140
 
141
  results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
142
  gigachat_pro_length = len(results["gigachat_pro"])
143
- gigachat_pro_display = f"{results['gigachat_pro']}\n\n------\nКоличество знаков: {gigachat_pro_length}"
144
  yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "Сообщение GigaChat-Pro сгенерировано"
145
 
146
  time.sleep(2)
147
 
148
  results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
149
  gigachat_lite_length = len(results["gigachat_lite"])
150
- gigachat_lite_display = f"{results['gigachat_lite']}\n\n------\nКоличество знаков: {gigachat_lite_length}"
151
  yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Сообщение GigaChat-Lite сгенерировано"
152
 
153
  time.sleep(2)
154
 
155
  results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
156
  gigachat_plus_length = len(results["gigachat_plus"])
157
- gigachat_plus_display = f"{results['gigachat_plus']}\n\n------\nКоличество знаков: {gigachat_plus_length}"
158
  yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все сообщения сгенерированы"
159
 
160
  return results
@@ -195,22 +195,22 @@ def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigacha
195
 
196
  personalized_message_gpt4o = perform_personalization(gpt4o_message, personalization_prompt)
197
  gpt4o_length = len(personalized_message_gpt4o)
198
- gpt4o_display = f"{personalized_message_gpt4o}\n\n------\nКоличество знаков: {gpt4o_length}"
199
  yield personalization_prompt, gpt4o_display, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
200
 
201
  personalized_message_gigachat_pro = perform_personalization_gigachat(gigachat_pro_message, personalization_prompt, "gigachat_pro")
202
  gigachat_pro_length = len(personalized_message_gigachat_pro)
203
- gigachat_pro_display = f"{personalized_message_gigachat_pro}\n\n------\nКоличество знаков: {gigachat_pro_length}"
204
  yield personalization_prompt, gpt4o_display, gigachat_pro_display, "", "", "Персонализированное сообщение GigaChat-Pro сгенерировано"
205
 
206
  personalized_message_gigachat_lite = perform_personalization_gigachat(gigachat_lite_message, personalization_prompt, "gigachat_lite")
207
  gigachat_lite_length = len(personalized_message_gigachat_lite)
208
- gigachat_lite_display = f"{personalized_message_gigachat_lite}\n\n------\nКоличество знаков: {gigachat_lite_length}"
209
  yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Персонализированное сообщение GigaChat-Lite сгенерировано"
210
 
211
  personalized_message_gigachat_plus = perform_personalization_gigachat(gigachat_plus_message, personalization_prompt, "gigachat_plus")
212
  gigachat_plus_length = len(personalized_message_gigachat_plus)
213
- gigachat_plus_display = f"{personalized_message_gigachat_plus}\n\n------\nКоличество знаков: {gigachat_plus_length}"
214
  yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все персонализированные сообщения сгенерированы"
215
 
216
 
 
15
  token = os.getenv('GITHUB_TOKEN')
16
 
17
  # Авторизация в сервисе GigaChat
18
+ chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=63, verify_ssl_certs=False)
19
+ chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=63, verify_ssl_certs=False)
20
+ chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=63, verify_ssl_certs=False)
21
 
22
  # Загрузка данных из Excel-файла
23
  try:
 
85
  data = {
86
  "model": "chatgpt-4o-latest",
87
  "messages": [{"role": "system", "content": prompt}],
88
+ "max_tokens": 105
89
  }
90
  response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
91
  response_data = response.json()
 
135
 
136
  results["gpt4o"] = generate_message_gpt4o(standard_prompt)
137
  gpt4o_length = len(results["gpt4o"])
138
+ gpt4o_display = f"{results['gpt4o']}\n------\nКоличество знаков: {gpt4o_length}"
139
  yield results["prompt"], gpt4o_display, "", "", "", "Сообщение GPT-4o сгенерировано"
140
 
141
  results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
142
  gigachat_pro_length = len(results["gigachat_pro"])
143
+ gigachat_pro_display = f"{results['gigachat_pro']}\n------\nКоличество знаков: {gigachat_pro_length}"
144
  yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "Сообщение GigaChat-Pro сгенерировано"
145
 
146
  time.sleep(2)
147
 
148
  results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
149
  gigachat_lite_length = len(results["gigachat_lite"])
150
+ gigachat_lite_display = f"{results['gigachat_lite']}\n------\nКоличество знаков: {gigachat_lite_length}"
151
  yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Сообщение GigaChat-Lite сгенерировано"
152
 
153
  time.sleep(2)
154
 
155
  results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
156
  gigachat_plus_length = len(results["gigachat_plus"])
157
+ gigachat_plus_display = f"{results['gigachat_plus']}\n------\nКоличество знаков: {gigachat_plus_length}"
158
  yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все сообщения сгенерированы"
159
 
160
  return results
 
195
 
196
  personalized_message_gpt4o = perform_personalization(gpt4o_message, personalization_prompt)
197
  gpt4o_length = len(personalized_message_gpt4o)
198
+ gpt4o_display = f"{personalized_message_gpt4o}\n------\nКоличество знаков: {gpt4o_length}"
199
  yield personalization_prompt, gpt4o_display, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
200
 
201
  personalized_message_gigachat_pro = perform_personalization_gigachat(gigachat_pro_message, personalization_prompt, "gigachat_pro")
202
  gigachat_pro_length = len(personalized_message_gigachat_pro)
203
+ gigachat_pro_display = f"{personalized_message_gigachat_pro}\n------\nКоличество знаков: {gigachat_pro_length}"
204
  yield personalization_prompt, gpt4o_display, gigachat_pro_display, "", "", "Персонализированное сообщение GigaChat-Pro сгенерировано"
205
 
206
  personalized_message_gigachat_lite = perform_personalization_gigachat(gigachat_lite_message, personalization_prompt, "gigachat_lite")
207
  gigachat_lite_length = len(personalized_message_gigachat_lite)
208
+ gigachat_lite_display = f"{personalized_message_gigachat_lite}\n------\nКоличество знаков: {gigachat_lite_length}"
209
  yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Персонализированное сообщение GigaChat-Lite сгенерировано"
210
 
211
  personalized_message_gigachat_plus = perform_personalization_gigachat(gigachat_plus_message, personalization_prompt, "gigachat_plus")
212
  gigachat_plus_length = len(personalized_message_gigachat_plus)
213
+ gigachat_plus_display = f"{personalized_message_gigachat_plus}\n------\nКоличество знаков: {gigachat_plus_length}"
214
  yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все персонализированные сообщения сгенерированы"
215
 
216