Upload app.py

app.py

@@ -0,0 +1,50 @@
+import os
+import gradio as gr
+from openai import OpenAI
+
+# Проверка ключа
+api_key = os.getenv("OPENROUTER_API_KEY")
+if not api_key:
+    raise ValueError("OPENROUTER_API_KEY не найден в Secrets!")
+
+# Создаем клиента
+client = OpenAI(
+    base_url="https://openrouter.ai/api/v1",
+    api_key=api_key
+)
+
+# Функция обработки сообщений
+def ask_ai(user_input, history=[]):
+    if not user_input.strip():
+        return history, "Введите сообщение!"
+    try:
+        messages = [{"role": "system", "content": "Ты HR-бот, задавай вопросы."}]
+        # Преобразуем историю в правильный формат словарей
+        for h in history[-6:]:
+            messages.append({"role": "user", "content": str(h[0])})
+            messages.append({"role": "assistant", "content": str(h[1])})
+        # Добавляем текущее сообщение
+        messages.append({"role": "user", "content": str(user_input)})
+
+        # Вызов LLM с ограничением токенов
+        response = client.chat.completions.create(
+            model="anthropic/claude-3.5-sonnet",
+            messages=messages,
+            max_tokens=1000
+        )
+
+        answer = response.choices[0].message.content
+        history.append((user_input, answer))
+        return history, ""
+    except Exception as e:
+        return history, f"Ошибка при вызове AI: {e}"
+
+# Интерфейс Gradio
+with gr.Blocks() as demo:
+    chat = gr.Chatbot()
+    msg = gr.Textbox(placeholder="Напишите сообщение...")
+    state = gr.State([])  # история чата
+    msg.submit(ask_ai, inputs=[msg, state], outputs=[chat, msg])
+
+# Запуск
+demo.launch()