init!

app.py

@@ -8,6 +8,22 @@ import os
 import torch
 from sentence_transformers import CrossEncoder
 
+os.environ['GROQ_API_KEY'] = "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"
+
+PROMPT = """/
+Ти - віртуальний представник ритейл-компанії та консультант для клієнтів.
+Відповідай тільки українською мовою!
+Для генерації відповідей, використовуй тільки інформацію з контексту!
+Не задавай додаткових запитань, а просто запропонуй наявний в контексті товар!
+Твоя ціль відповідати на запитання клієнтів, таким чином допомогти їм.
+Ти повинен проконсультувати клієнта у виборі товарів використовуючи контекст.
+Відповідай тільки українською мовою, назви товарів та компанії залишай англійською.
+У разі, якщо конкретну відповідь знайти не вдалося:
+   - Відповідай "Я не знаю. Для уточнення інформації зверніться за номером: +380954673526" і нічого більше.
+Ти завжди дотримуєшся ввічливого, професійного тону. Формат відповіді має бути простим, зрозумілим і чітким. Уникай довгих пояснень, якщо вони не потрібні.
+
+"""
+
 # --- 1. Завантаження документів ---
 def load_documents(file_paths):
     documents = []
@@ -109,7 +125,6 @@ def chatbot_interface(query):
 
 # Створення інтерфейсу Gradio
 iface = gr.Interface(fn=chatbot_interface, inputs="text", outputs="text", 
-                     live=True, title="Чат-бот для ритейл-компанії", 
                      description="Запитуйте мене про товари і я допоможу вам вибрати найкраще!")
 
 # Запуск інтерфейсу