Update app.py

app.py

@@ -4,10 +4,8 @@ import gradio as gr
 from typing import List, Optional, Tuple, Dict, Iterator
 
 # 🛠 Безопасная загрузка API-ключа
-GROQ_API_KEY = os.getenv("GROQ_API_KEY", "").strip()
-if not GROQ_API_KEY:
-    raise ValueError("API ключ не найден! Укажите его в коде или в переменной окружения.")
-
+#GROQ_API_KEY = "ВАШ_КЛЮЧ"  # Укажите API-ключ явно
+GROQ_API_KEY = os.environ.get('GROQ_API_KEY')
 # 🛠 Инициализация клиента
 client = OpenAI(
     api_key=GROQ_API_KEY,
@@ -32,8 +30,8 @@ def history_to_messages(history: History, system: str) -> Messages:
         messages.append({'role': 'assistant', 'content': ai_response})
     return messages
 
-# 🛠 Функция общения с моделью (с исправленным потоком)
-def model_chat(query: Optional[str], history: Optional[History], system: str, model_name: str) -> Iterator[Tuple[str, History, str]]:
+# 🛠 Функция общения с моделью
+def model_chat(query: Optional[str], history: Optional[History], system: str, model_name: str) -> Iterator[List[Dict[str, str]]]:
     if not query:
         return
     if history is None:
@@ -56,12 +54,12 @@ def model_chat(query: Optional[str], history: Optional[History], system: str, mo
                 content = getattr(delta, "content", "")
                 if content:
                     full_response += content
-                    yield full_response, history + [(query, full_response)], system
+                    yield [{"role": "assistant", "content": full_response}], history + [(query, full_response)], system
 
     except Exception as e:
         import traceback
         error_message = traceback.format_exc()
-        yield f"Ошибка: {error_message}", history, system
+        yield [{"role": "system", "content": f"Ошибка: {error_message}"}], history, system
 
 # 🛠 Корректное отображение выбора модели
 def choose_radio(model_name: str, system: str):
@@ -101,8 +99,8 @@ def main():
                     clear_history = gr.Button("🧹 Очистить историю")
                     submit = gr.Button("🚀 Отправить")
 
-                textbox.submit(model_chat, inputs=[textbox, chatbot, system_state, radio], outputs=[textbox, chatbot, system_input])
-                submit.click(model_chat, inputs=[textbox, chatbot, system_state, radio], outputs=[textbox, chatbot, system_input])
+                textbox.submit(model_chat, inputs=[textbox, chatbot, system_state, radio], outputs=[chatbot, chatbot, system_input])
+                submit.click(model_chat, inputs=[textbox, chatbot, system_state, radio], outputs=[chatbot, chatbot, system_input])
                 clear_history.click(fn=clear_session, inputs=[], outputs=[textbox, chatbot, system_input])
                 modify_system.click(fn=modify_system_session, inputs=[system_input], outputs=[system_state, system_input, chatbot])
         
@@ -111,5 +109,14 @@ def main():
     demo.queue(api_open=False)
     demo.launch()
 
+# Функция теста API
+def test_api():
+    chat_completion = client.chat.completions.create(
+        model="deepseek-r1-distill-llama-70b",
+        messages=[{"role": "user", "content": "What's the purpose of Generative AI?"}],
+    )
+    print(chat_completion.choices[0].message.content)
+
 if __name__ == "__main__":
     main()
+    test_api()