Update app.py

app.py

@@ -82,17 +82,27 @@ def generate_message_gpt4o(prompt, temperature):
             "Authorization": f"Bearer {openai_api_key}"
         }
         data = {
-            "model": "chatgpt-4o-latest",
+            "model": "chatgpt-4o-latest",  # убедитесь, что модель корректна
             "messages": [{"role": "system", "content": prompt}],
             "max_tokens": 101,
             "temperature": temperature
         }
         response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
         response_data = response.json()
-        return clean_message(response_data["choices"][0]["message"]["content"].strip())
+
+        # Проверка на наличие ошибок в ответе
+        if response.status_code != 200:
+            return f"Ошибка от OpenAI: {response_data.get('error', {}).get('message', 'Неизвестная ошибка')}"
+
+        # Проверка наличия поля 'choices' в ответе
+        if "choices" in response_data and len(response_data["choices"]) > 0:
+            return clean_message(response_data["choices"][0]["message"]["content"].strip())
+        else:
+            return "Ошибка: ответ от API не содержит необходимых данных."
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
+
 def clean_message(message):
     # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
     if not message.endswith(('.', '!', '?')):
@@ -229,9 +239,9 @@ def perform_personalization_with_retry(standard_message, personalization_prompt)
     return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
 
-def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model, temperature):
+def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model):
     for _ in range(10):
-        message = perform_personalization_gigachat(standard_message, personalization_prompt, model, temperature)
+        message = perform_personalization_gigachat(standard_message, personalization_prompt, model)
         if len(message) <= 250:
             return message
     return message