Update app.py

app.py

@@ -1,289 +1,339 @@
+"""
+🍕 Чат-бот для піцерії "Мама Міа" - Hugging Face версія
+Використовує файл menu.txt для завантаження меню
+"""
+
 import os
-import time
 import gradio as gr
-from google import genai
-from google.genai import types
-from dotenv import load_dotenv
+from openai import OpenAI
 
-# Завантаження змінних оточення
-load_dotenv()
+class PizzaChatBot:
+    """Основний клас чат-боту для піцерії"""
+    
+    def __init__(self):
+        """Ініціалізація чат-боту"""
+        # Отримуємо API ключ з змінних середовища (HF Secrets)
+        self.api_key = os.getenv('OPENROUTER_API_KEY')
+        self.base_url = "https://openrouter.ai/api/v1"
+        self.model = "qwen/qwen3-235b-a22b:free"
+        self.restaurant_name = "Піцерія 'Мама Міа'"
+        self.menu = self.load_menu()
+        
+        # Ініціалізація OpenAI клієнта для OpenRouter
+        if self.api_key and self.api_key != 'your_api_key_here':
+            self.client = OpenAI(
+                base_url=self.base_url,
+                api_key=self.api_key
+            )
+            print(f"✅ {self.restaurant_name} чат-бот ініціалізовано")
+        else:
+            self.client = None
+            print("⚠️ API ключ не налаштовано - використовується демо режим")
+    
+    def load_menu(self):
+        """Завантажує меню з файлу menu.txt"""
+        try:
+            with open('menu.txt', 'r', encoding='utf-8') as file:
+                content = file.read()
+                print(f"✅ Меню завантажено з файлу ({len(content)} символів)")
+                return content
+        except FileNotFoundError:
+            print("❌ Файл menu.txt не знайдено - використовується базове меню")
+            return self.get_fallback_menu()
+        except Exception as e:
+            print(f"❌ Помилка завантаження меню: {e}")
+            return self.get_fallback_menu()
+    
+    def get_fallback_menu(self):
+        """Резервне меню, якщо файл не завантажується"""
+        return """🍕 ПІЦЕРІЯ "МАМА МІА" 🍕
 
-# Отримання 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)
+🍕 Маргарита (30см) - 250 грн
+🍕 Пепероні (30см) - 320 грн  
+🍕 Барбекю Чікен (30см) - 450 грн
+🍕 Морська (30см) - 520 грн
 
-# Назва моделі
-MODEL_NAME = "gemini-2.5-flash-preview-04-17"
+🥗 Салат Цезар - 180 грн
+🥤 Кока-кола - 35 грн
+🍰 Тірамісу - 120 грн
 
-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 """# МЕНЮ ПІЦЕРІЇ "СМАЧНА ПІЦА" (РЕЗЕРВНА КОПІЯ)
+⏰ РЕЖИМ РОБОТИ: 11:00 - 23:00
+🚚 ДОСТАВКА: Безкоштовно від 400 грн
+📞 ТЕЛЕФОН: +38 (044) 123-45-67"""
+    
+    def get_system_prompt(self):
+        """Створює системний промпт для LLM"""
+        return f"""Ти - дружелюбний помічник піцерії "{self.restaurant_name}". 
 
-## КЛАСИЧНІ ПІЦИ
+Твоя роль:
+- Допомагати клієнтам з вибором піци та страв з меню
+- Надавати точну інформацію про ціни та інгредієнти
+- Розповідати про режим роботи, доставку та акції
+- Приймати замовлення та відповідати на запитання
+- Бути ввічливим, корисним та дружелюбним
 
-МАРГАРИТА
-Опис: Класична піца з томатним соусом, моцарелою та базиліком
-Інгредієнти: томатний соус, сир моцарела, свіжий базилік, оливкова олія
-Розмір: 30 см | Ціна: 160 грн
-Розмір: 40 см | Ціна: 220 грн
+Важливі правила:
+- Завжди говори українською мовою
+- Використовуй емодзі для дружнього спілкування 🍕😊
+- Базуйся тільки на інформації з меню
+- Якщо не знаєш відповіді - направляй до телефону
+- Пропонуй популярні страви та акції
 
-ПЕПЕРОНІ
-Опис: Піца з пікантною ковбаскою пепероні
-Інгредієнти: томатний соус, сир моцарела, пепероні
-Розмір: 30 см | Ціна: 180 грн
-Розмір: 40 см | Ціна: 250 грн"""
-    
-    except Exception as e:
-        print(f"Помилка при зчитуванні меню: {str(e)}")
-        return "Меню тимчасово недоступне. Будь ласка, зверніться до адміністратора."
+Ось повне меню та інформація про піцерію:
 
-# Глобальна змінна для зберігання меню
-menu_data = None
+{self.menu}
 
-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()
+Починай кожну розмову привітно та будь готовий допомогти!"""
+    
+    def chat_response(self, message, history):
+        """Обробляє повідомлення користувача та повертає відповідь"""
+        # Якщо немає API ключа - демо режим
+        if not self.client:
+            return self.demo_response(message)
         
-        # Перевірка завантаження меню
-        if "Меню тимчасово недоступне" in menu_data:
-            print("УВАГА: Меню не завантажено коректно! Перевірте файл menu.txt")
-        else:
-            print(f"Меню успішно завантажено. Розмір: {len(menu_data)} символів")
+        try:
+            # Формуємо історію розмови для API
+            messages = [{"role": "system", "content": self.get_system_prompt()}]
+            
+            # Додаємо історію розмови (останні 10 повідомлень для економії токенів)
+            recent_history = history[-10:] if len(history) > 10 else history
+            for human, assistant in recent_history:
+                if human and assistant:
+                    messages.append({"role": "user", "content": human})
+                    messages.append({"role": "assistant", "content": assistant})
+            
+            # Додаємо поточне повідомлення
+            messages.append({"role": "user", "content": message})
+            
+            # Отримуємо відповідь від LLM через OpenRouter
+            response = self.client.chat.completions.create(
+                model=self.model,
+                messages=messages,
+                max_tokens=800,
+                temperature=0.7
+            )
+            
+            return response.choices[0].message.content
+            
+        except Exception as e:
+            print(f"❌ Помилка API: {e}")
+            return f"Вибачте, виникла технічна проблема 😔 Зателефонуйте нам: +38 (044) 123-45-67"
     
-    return menu_data
+    def demo_response(self, message):
+        """Демо відповіді без API"""
+        message_lower = message.lower()
+        
+        if any(word in message_lower for word in ['меню', 'піца', 'страв']):
+            return """🍕 Ось наше меню:
 
-def create_system_prompt():
-    """Створює системний промпт для моделі з меню піцерії"""
-    current_menu = get_menu()
-    return f"""
-    Ти дружній чат-бот української піцерії "Смачна Піца".
-    Твоє завдання - допомагати клієнтам дізнатися про меню, ціни, інгредієнти та доставку.
-    
-    Дотримуйся цих правил:
-    1. Завжди будь ввічливим і дружнім
-    2. Відповідай ТІЛЬКИ на основі наданого меню
-    3. Якщо клієнт запитує про щось, чого немає в меню, скажи, що такого у меню немає, але можеш запропонувати схожі опції
-    4. Завжди пиши українською мовою
-    5. Твої відповіді мають бути короткими і по суті
-    6. Пам'ятай всю попередню розмову з клієнтом і використовуй цю інформацію
-    7. Якщо клієнт раніше цікавився певними піцами, використовуй цю інформацію для рекомендацій
-    
-    Ось актуальне меню:
-    
-    {current_menu}
-    """
+**ПІЦИ:**
+• Маргарита (30см) - 250 грн
+• Пепероні (30см) - 320 грн  
+• Барбекю Чікен (30см) - 450 грн
+• Морська (30см) - 520 грн
 
-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)
+        elif any(word in message_lower for word in ['ціна', 'скільки', 'коштує']):
+            return "💰 Ціни на піци від 250 до 520 грн. Безкоштовна доставка від 400 грн! Що саме вас цікавить?"
         
-        # Виявлення вподобань щодо розміру
-        if "30 см" in human_msg:
-            preferences["розмір"] = "30 см"
-        if "40 см" in human_msg:
-            preferences["розмір"] = "40 см"
+        elif any(word in message_lower for word in ['доставка', 'час', 'швидко']):
+            return "🚚 Доставка 30-45 хвилин. Безкоштовно від 400 грн, інакше 50 грн. Працюємо з 11:00 до 23:00!"
         
-        # Виявлення вподобань щодо додатків
-        for addition in ["часниковий соус", "кетчуп", "сирний соус"]:
-            if addition in human_msg and addition not in preferences["додатки"]:
-                preferences["додатки"].append(addition)
+        elif any(word in message_lower for word in ['телефон', 'контакт', 'адреса']):
+            return "📞 Телефон: +38 (044) 123-45-67\n📍 Адреса: вул. Хрещатик, 25, Київ\n⏰ Режим: 11:00 - 23:00"
         
-        # Виявлення вподобань щодо напоїв
-        for drink in ["кола", "спрайт", "мінеральна вода"]:
-            if drink in human_msg and drink not in preferences["напої"]:
-                preferences["напої"].append(drink)
-    
-    return preferences
+        else:
+            return f"""Привіт! 👋 Я помічник піцерії "Мама Міа". 
 
-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 = []
+*Примітка: Зараз працює демо-режим. Для повного функціоналу потрібен API ключ.*"""
+    
+    def create_interface(self):
+        """Створює Gradio інтерфейс для Hugging Face"""
         
-        # Додаємо попередні повідомлення з історії
-        for human_msg, bot_msg in history:
-            conversation.append(
-                types.Content(
-                    role="user",
-                    parts=[types.Part.from_text(text=human_msg)]
-                )
+        # Кастомні стилі
+        custom_css = """
+        .gradio-container {
+            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+        }
+        .header-banner {
+            background: linear-gradient(135deg, #ff6b6b, #feca57, #48cab2);
+            border-radius: 15px;
+            padding: 25px;
+            text-align: center;
+            margin-bottom: 20px;
+            box-shadow: 0 4px 15px rgba(0,0,0,0.1);
+        }
+        .info-card {
+            background-color: #f8f9fa;
+            border-radius: 10px;
+            padding: 15px;
+            margin-top: 15px;
+            border-left: 4px solid #ff6b6b;
+        }
+        """
+        
+        with gr.Blocks(
+            css=custom_css,
+            title=f"{self.restaurant_name} - AI Чат-бот",
+            theme=gr.themes.Soft()
+        ) as interface:
+            
+            # Заголовок з градієнтом
+            gr.HTML(f"""
+                <div class="header-banner">
+                    <h1 style="color: white; margin: 0; font-size: 2.5em;">🍕 {self.restaurant_name}</h1>
+                    <h3 style="color: white; margin: 15px 0; font-size: 1.3em;">
+                        Привіт! Я AI-помічник, готовий допомогти з замовленням 😊
+                    </h3>
+                    <p style="color: white; margin: 0; font-size: 1.1em;">
+                        Запитайте про меню, ціни, доставку або зробіть замовлення!
+                    </p>
+                </div>
+            """)
+            
+            # Основний чат
+            chatbot = gr.Chatbot(
+                height=500,
+                placeholder="🍕 Привіт! Що вас цікавить сьогодні?",
+                label="💬 Розмова з піца-ботом",
+                avatar_images=("👤", "🤖")
             )
-            conversation.append(
-                types.Content(
-                    role="model",
-                    parts=[types.Part.from_text(text=bot_msg)]
+            
+            # Поле вводу та кнопка відправки
+            with gr.Row():
+                msg = gr.Textbox(
+                    placeholder="Введіть ваше повідомлення тут...",
+                    label="",
+                    lines=2,
+                    scale=4,
+                    container=False
                 )
+                send_btn = gr.Button(
+                    "Відправити 📤", 
+                    variant="primary", 
+                    scale=1,
+                    size="lg"
+                )
+            
+            # Додаткові кнопки
+            with gr.Row():
+                clear_btn = gr.Button("Очистити чат 🗑️", variant="secondary")
+                
+            # Швидкі запитання
+            gr.Examples(
+                examples=[
+                    "Покажіть меню піци",
+                    "Яка найпопулярніша піца?", 
+                    "Скільки коштує доставка?",
+                    "Режим роботи піцерії",
+                    "Хочу замовити 2 піци Маргарита",
+                    "Які у вас акції?"
+                ],
+                inputs=msg,
+                label="💡 Швидкі запитання (натисніть для вибору):"
             )
-        
-        # Додаємо поточне повідомлення
-        conversation.append(
-            types.Content(
-                role="user",
-                parts=[types.Part.from_text(text=message)]
+            
+            # Інформаційна картка
+            gr.HTML("""
+                <div class="info-card">
+                    <h4 style="margin-top: 0; color: #333;">📋 Основна інформація:</h4>
+                    <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(250px, 1fr)); gap: 15px;">
+                        <div>
+                            <strong>📍 Адреса:</strong><br>
+                            вул. Хрещатик, 25, Київ
+                        </div>
+                        <div>
+                            <strong>📞 Телефон:</strong><br>
+                            +38 (044) 123-45-67
+                        </div>
+                        <div>
+                            <strong>⏰ Режим роботи:</strong><br>
+                            Щодня 11:00 - 23:00
+                        </div>
+                        <div>
+                            <strong>🚚 Доставка:</strong><br>
+                            Безкоштовно від 400 грн
+                        </div>
+                    </div>
+                </div>
+            """)
+            
+            # Обробники подій
+            def user_message(message, history):
+                """Додає повідомлення користувача до чату"""
+                if message.strip():
+                    return "", history + [[message, None]]
+                return message, history
+            
+            def bot_response(history):
+                """Генерує відповідь бота"""
+                if history and history[-1][0]:
+                    message = history[-1][0]
+                    response = self.chat_response(message, history[:-1])
+                    history[-1][1] = response
+                return history
+            
+            def clear_chat():
+                """Очищає історію чату"""
+                return []
+            
+            # Прив'язування подій
+            msg.submit(
+                user_message, 
+                [msg, chatbot], 
+                [msg, chatbot], 
+                queue=False
+            ).then(
+                bot_response, 
+                chatbot, 
+                chatbot
             )
-        )
-        
-        # Створюємо розширений системний промпт на основі історії
-        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}")
+            
+            send_btn.click(
+                user_message, 
+                [msg, chatbot], 
+                [msg, chatbot], 
+                queue=False
+            ).then(
+                bot_response, 
+                chatbot, 
+                chatbot
+            )
+            
+            clear_btn.click(clear_chat, None, chatbot, queue=False)
         
-        # Повертаємо зрозуміле повідомлення про помилку
-        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()
+        return interface
 
-# Створюємо розширений інтерфейс з вкладками
-with gr.Blocks(theme="soft") as demo:
-    gr.Markdown("# Чат-бот піцерії 'Смачна Піца'")
-    gr.Markdown("Вітаю! Я чат-бот піцерії 'Смачна Піца'. Запитайте мене про меню, ціни, інгредієнти або доставку.")
+def main():
+    """Головна функція для запуску на Hugging Face Spaces"""
+    print("🚀 Ініціалізація піца-чат-боту для Hugging Face...")
     
-    with gr.Tab("Чат"):
-        chatbot = gr.ChatInterface(
-            fn=chat_with_history,
-            examples=[
-                "Які у вас є піци?",
-                "Розкажи про піцу Пепероні",
-                "Скільки коштує доставка?",
-                "Які у вас є вегетаріанські піци?",
-                "Що порадиш з напоїв?",
-                "О котрій годині ви працюєте?"
-            ]
-        )
+    # Створення екземпляру чат-боту
+    bot = PizzaChatBot()
     
-    with gr.Tab("Меню"):
-        gr.Markdown(get_menu())
+    # Створення інтерфейсу
+    interface = bot.create_interface()
     
-    with gr.Tab("Про нас"):
-        gr.Markdown("""
-        # Піцерія "Смачна Піца"
-        
-        Ми пропонуємо найсмачнішу піцу в місті, приготовану з найсвіжіших інгредієнтів!
-        
-        ## Контактна інформація
-        
-        📞 Телефон: 067-123-4567  
-        🕒 Години роботи: щодня з 10:00 до 22:00  
-        🏠 Адреса: вул. Піцерійна, 1  
-        
-        ## Доставка
-        
-        Безкоштовна доставка при замовленні від 300 грн  
-        Час доставки: 30-60 хвилин  
-        Зона доставки: в межах міста  
-        """)
-
-# Завантажуємо меню перед стартом
-print("Ініціалізація програми...")
-get_menu()
+    print("✅ Інтерфейс створено, запускаємо...")
+    return interface
 
+# Запуск для Hugging Face Spaces
 if __name__ == "__main__":
-    print("Запуск чат-бота піцерії...")
-    demo.launch(share=True)  # share=True для генерації публічного посилання
+    demo = main()
+    demo.launch(
+        server_name="0.0.0.0",  # Для Hugging Face
+        server_port=7860,       # Стандартний порт HF
+        share=False,            # HF автоматично надає публічний доступ
+        debug=False             # Вимкнути debug в продакшені
+    )