Create app.py

app.py

@@ -0,0 +1,289 @@
+import os
+import time
+import gradio as gr
+from google import genai
+from google.genai import types
+from dotenv import load_dotenv
+
+# Завантаження змінних оточення
+load_dotenv()
+
+# Отримання API ключа (спершу з HF_ENVIRON, потім з локального .env)
+api_key = os.environ.get("GEMINI_API_KEY")
+if not api_key:
+    api_key = os.getenv("GEMINI_API_KEY")
+    if not api_key:
+        raise ValueError("API ключ не знайдено. Перевірте змінні оточення")
+
+# Налаштування клієнта API
+client = genai.Client(api_key=api_key)
+
+# Назва моделі
+MODEL_NAME = "gemini-2.5-flash-preview-04-17"
+
+def load_menu():
+    """Функція для зчитування меню з файлу з покращеною обробкою шляхів"""
+    try:
+        # Спроба знайти файл відносно поточної директорії
+        if os.path.exists('menu.txt'):
+            print("Знайдено menu.txt в поточній директорії")
+            with open('menu.txt', 'r', encoding='utf-8') as file:
+                menu_content = file.read()
+            return menu_content
+        
+        # Спроба знайти файл відносно директорії скрипту
+        script_dir = os.path.dirname(os.path.abspath(__file__))
+        menu_path = os.path.join(script_dir, 'menu.txt')
+        if os.path.exists(menu_path):
+            print(f"Знайдено menu.txt в директорії скрипту: {menu_path}")
+            with open(menu_path, 'r', encoding='utf-8') as file:
+                menu_content = file.read()
+            return menu_content
+        
+        # Пошук в різних можливих директоріях (для Hugging Face)
+        for possible_dir in ['/app', '/home/user', '/content', '/workspace']:
+            menu_path = os.path.join(possible_dir, 'menu.txt')
+            if os.path.exists(menu_path):
+                print(f"Знайдено menu.txt в: {menu_path}")
+                with open(menu_path, 'r', encoding='utf-8') as file:
+                    menu_content = file.read()
+                return menu_content
+        
+        # Якщо файл не знайдено
+        print("ПОМИЛКА: Файл menu.txt не знайдено в жодній з директорій")
+        # Резервний вміст меню для демонстрації
+        return """# МЕНЮ ПІЦЕРІЇ "СМАЧНА ПІЦА" (РЕЗЕРВНА КОПІЯ)
+
+## КЛАСИЧНІ ПІЦИ
+
+МАРГАРИТА
+Опис: Класична піца з томатним соусом, моцарелою та базиліком
+Інгредієнти: томатний соус, сир моцарела, свіжий базилік, оливкова олія
+Розмір: 30 см | Ціна: 160 грн
+Розмір: 40 см | Ціна: 220 грн
+
+ПЕПЕРОНІ
+Опис: Піца з пікантною ковбаскою пепероні
+Інгредієнти: томатний соус, сир моцарела, пепероні
+Розмір: 30 см | Ціна: 180 грн
+Розмір: 40 см | Ціна: 250 грн"""
+    
+    except Exception as e:
+        print(f"Помилка при зчитуванні меню: {str(e)}")
+        return "Меню тимчасово недоступне. Будь ласка, зверніться до адміністратора."
+
+# Глобальна змінна для зберігання меню
+menu_data = None
+
+def get_menu():
+    """Повертає поточне меню, з можливістю оновлення"""
+    global menu_data
+    # Перше завантаження або оновлення меню кожну годину
+    if menu_data is None or not hasattr(get_menu, "last_update") or time.time() - get_menu.last_update > 3600:
+        menu_data = load_menu()
+        get_menu.last_update = time.time()
+        
+        # Перевірка завантаження меню
+        if "Меню тимчасово недоступне" in menu_data:
+            print("УВАГА: Меню не завантажено коректно! Перевірте файл menu.txt")
+        else:
+            print(f"Меню успішно завантажено. Розмір: {len(menu_data)} символів")
+    
+    return menu_data
+
+def create_system_prompt():
+    """Створює системний промпт для моделі з меню піцерії"""
+    current_menu = get_menu()
+    return f"""
+    Ти дружній чат-бот української піцерії "Смачна Піца".
+    Твоє завдання - допомагати клієнтам дізнатися про меню, ціни, інгредієнти та доставку.
+    
+    Дотримуйся цих правил:
+    1. Завжди будь ввічливим і дружнім
+    2. Відповідай ТІЛЬКИ на основі наданого меню
+    3. Якщо клієнт запитує про щось, чого немає в меню, скажи, що такого у меню немає, але можеш запропонувати схожі опції
+    4. Завжди пиши українською мовою
+    5. Твої відповіді мають бути короткими і по суті
+    6. Пам'ятай всю попередню розмову з клієнтом і використовуй цю інформацію
+    7. Якщо клієнт раніше цікавився певними піцами, використовуй цю інформацію для рекомендацій
+    
+    Ось актуальне меню:
+    
+    {current_menu}
+    """
+
+def extract_client_preferences(history):
+    """Аналізує історію розмови та виявляє уподобання клієнта"""
+    preferences = {
+        "піци": [],
+        "розмір": "",
+        "додатки": [],
+        "напої": []
+    }
+    
+    # Аналіз історії повідомлень
+    for human_msg, _ in history:
+        human_msg = human_msg.lower()
+        
+        # Виявлення вподобань щодо піц
+        for pizza in ["маргарита", "пепероні", "гавайська", "чотири сири", "карбонара", "барбекю", "вегетаріанська", "грибна"]:
+            if pizza in human_msg and pizza not in preferences["піци"]:
+                preferences["піци"].append(pizza)
+        
+        # Виявлення вподобань щодо розміру
+        if "30 см" in human_msg:
+            preferences["розмір"] = "30 см"
+        if "40 см" in human_msg:
+            preferences["розмір"] = "40 см"
+        
+        # Виявлення вподобань щодо додатків
+        for addition in ["часниковий соус", "кетчуп", "сирний соус"]:
+            if addition in human_msg and addition not in preferences["додатки"]:
+                preferences["додатки"].append(addition)
+        
+        # Виявлення вподобань щодо напоїв
+        for drink in ["кола", "спрайт", "мінеральна вода"]:
+            if drink in human_msg and drink not in preferences["напої"]:
+                preferences["напої"].append(drink)
+    
+    return preferences
+
+def enhance_system_prompt(history):
+    """Створює розширений системний промпт з урахуванням уподобань клієнта"""
+    base_prompt = create_system_prompt()
+    
+    if not history:
+        return base_prompt
+    
+    preferences = extract_client_preferences(history)
+    
+    # Додаємо інформацію про уподобання клієнта до промпту
+    preferences_prompt = "\nІнформація про уподобання клієнта на основі історії розмови:\n"
+    
+    if preferences["піци"]:
+        preferences_prompt += f"- Клієнт цікавився такими піцами: {', '.join(preferences['піци'])}\n"
+    
+    if preferences["розмір"]:
+        preferences_prompt += f"- Клієнт цікавився розміром: {preferences['розмір']}\n"
+    
+    if preferences["додатки"]:
+        preferences_prompt += f"- Клієнт цікавився такими додатками: {', '.join(preferences['додатки'])}\n"
+    
+    if preferences["напої"]:
+        preferences_prompt += f"- Клієнт цікавився такими напоями: {', '.join(preferences['напої'])}\n"
+    
+    enhanced_prompt = base_prompt + preferences_prompt
+    return enhanced_prompt
+
+def chat_with_history(message, history):
+    """Функція для чату з історією розмови"""
+    try:
+        # Перетворюємо історію розмови у формат для API
+        conversation = []
+        
+        # Додаємо попередні повідомлення з історії
+        for human_msg, bot_msg in history:
+            conversation.append(
+                types.Content(
+                    role="user",
+                    parts=[types.Part.from_text(text=human_msg)]
+                )
+            )
+            conversation.append(
+                types.Content(
+                    role="model",
+                    parts=[types.Part.from_text(text=bot_msg)]
+                )
+            )
+        
+        # Додаємо поточне повідомлення
+        conversation.append(
+            types.Content(
+                role="user",
+                parts=[types.Part.from_text(text=message)]
+            )
+        )
+        
+        # Створюємо розширений системний промпт на основі історії
+        enhanced_prompt = enhance_system_prompt(history)
+        
+        # Конфігурація для запиту
+        generate_content_config = types.GenerateContentConfig(
+            response_mime_type="text/plain",
+            system_instruction=[
+                types.Part.from_text(text=enhanced_prompt),
+            ],
+        )
+        
+        # Відправляємо запит до моделі
+        response = client.models.generate_content(
+            model=MODEL_NAME,
+            contents=conversation,
+            config=generate_content_config,
+        )
+        
+        return response.text
+        
+    except Exception as e:
+        error_message = str(e)
+        print(f"Помилка при зверненні до API: {error_message}")
+        
+        # Повертаємо зрозуміле повідомлення про помилку
+        if "rate limit" in error_message.lower():
+            return "Вибачте, зараз я отримую забагато запитів. Будь ласка, спробуйте знову через кілька хвилин."
+        elif "api key" in error_message.lower():
+            return "Виникла помилка з аутентифікацією. Будь ласка, повідомте адміністратора."
+        else:
+            return f"Вибачте, виникла технічна проблема. Спробуйте повторити запитання або зв'яжіться з піцерією за телефоном."
+
+def display_menu():
+    """Функція для відображення поточного меню"""
+    return get_menu()
+
+# Створюємо розширений інтерфейс з вкладками
+with gr.Blocks(theme="soft") as demo:
+    gr.Markdown("# Чат-бот піцерії 'Смачна Піца'")
+    gr.Markdown("Вітаю! Я чат-бот піцерії 'Смачна Піца'. Запитайте мене про меню, ціни, інгредієнти або доставку.")
+    
+    with gr.Tab("Чат"):
+        chatbot = gr.ChatInterface(
+            fn=chat_with_history,
+            examples=[
+                "Які у вас є піци?",
+                "Розкажи про піцу Пепероні",
+                "Скільки коштує доставка?",
+                "Які у вас є вегетаріанські піци?",
+                "Що порадиш з напоїв?",
+                "О котрій годині ви працюєте?"
+            ]
+        )
+    
+    with gr.Tab("Меню"):
+        gr.Markdown(get_menu())
+    
+    with gr.Tab("Про нас"):
+        gr.Markdown("""
+        # Піцерія "Смачна Піца"
+        
+        Ми пропонуємо найсмачнішу піцу в місті, приготовану з найсвіжіших інгредієнтів!
+        
+        ## Контактна інформація
+        
+        📞 Телефон: 067-123-4567  
+        🕒 Години роботи: щодня з 10:00 до 22:00  
+        🏠 Адреса: вул. Піцерійна, 1  
+        
+        ## Доставка
+        
+        Безкоштовна доставка при замовленні від 300 грн  
+        Час доставки: 30-60 хвилин  
+        Зона доставки: в межах міста  
+        """)
+
+# Завантажуємо меню перед стартом
+print("Ініціалізація програми...")
+get_menu()
+
+if __name__ == "__main__":
+    print("Запуск чат-бота піцерії...")
+    demo.launch(share=True)  # share=True для генерації публічного посилання