Update app.py

app.py

@@ -11,12 +11,29 @@ import time
 from openai import OpenAI
 import string
 import math
+import langchain_gigachat
+
+from langchain_gigachat.chat_models import GigaChat
 
 XAI_API_KEY = os.getenv("XAI_API_KEY")
 client = OpenAI(
     api_key=XAI_API_KEY,
     base_url="https://api.x.ai/v1",
 )
+
+# Авторизация в GigaChat Pro
+gc_key = os.getenv('GC_KEY')
+chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, temperature=1.15, verify_ssl_certs=False, scope="GIGACHAT_API_CORP")
+
+# Функция для генерации сообщения с GigaChat Pro
+def call_model(prompt):
+    try:
+        messages = [SystemMessage(content=prompt)]
+        res = chat_pro.invoke(messages)
+        return res
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+
 token = os.getenv('GITHUB_TOKEN')
 repo = "fruitpicker01/Storage_Anastasia"
 current_request_index = -1
@@ -189,7 +206,8 @@ def generate_display_prompts(description, product_name, benefits, key_message, c
 Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
 
     return prompt_1, prompt_2
-
+    
+'''
 def call_model(model_prompt):
     completion = client.chat.completions.create(
         model="grok-2-1212",
@@ -199,6 +217,7 @@ def call_model(model_prompt):
         ],
     )
     return completion.choices[0].message.content.strip()
+'''
 
 def correct_dash_usage(text):
     morph = pymorphy3.MorphAnalyzer()