import gradio as gr
import random

# История для контекста
conversation_history = []

def bounty_hunter_response(message, history):
    # Добавляем системный промпт
    prompt = f"""Ты - охотник за головами Боба Фетт из Звездных Войн. 
Отвечай кратко, грубо, в стиле наемника.
Используй термины: кредиты, контракт, Империя, корабль, гиперпространство.
На русском языке.

Сообщение: {message}
Ответ охотника:"""
    
    # Здесь будет интеграция с моделью
    # Пока используем заглушку
    responses = [
        "5000 кредитов. Без торга.",
        "Мой корабль готов. Координаты?",
        "Видел цель в секторе Татуин.",
        "Контракт принят. Детали?",
        "Не трать моё время. Какая работа?",
        "...",
        "Что-что? Говори о контракте."
    ]
    
    # Простая логика ответа
    if any(word in message.lower() for word in ['цена', 'стоимость', 'credit']):
        response = "10000 кредитов. Аванс 50%."
    elif any(word in message.lower() for word in ['корабль', 'ship', 'полёт']):
        response = "Мой 'Славный-1' готов к прыжку."
    elif any(word in message.lower() for word in ['привет', 'hello', 'hi']):
        response = "Канал безопасен. Докладывай."
    else:
        response = random.choice(responses)
    
    # Сохраняем историю
    conversation_history.append({"user": message, "bot": response})
    
    return response

# Создаем красивый интерфейс в стиле Звездных Войн
css = """
.background {
    background: #000;
    color: #39ff14;
    font-family: 'Courier New';
}
"""

with gr.Blocks(css=css, theme=gr.themes.Default(primary_hue="green")) as demo:
    gr.Markdown("# ⍟ BOUNTY HUNTER COMMS ⍟")
    gr.Markdown("**Зашифрованный канал охотника за головами**")
    
    chatbot = gr.ChatInterface(
        bounty_hunter_response,
        title="",
        description="Общайтесь с наемником из далекой-далекой галактики..."
    )

if __name__ == "__main__":
    demo.launch(share=True)