import gradio as gr
from transformers import pipeline

# Завантаження моделі
def load_model():
    try:
        generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M")
        return generator
    except Exception as e:
        return f"Помилка завантаження моделі: {e}"

# Ініціалізація моделі
generator = load_model()

# Функція для генерації відповіді
def chat_with_npc(user_input):
    if isinstance(generator, str):  # Якщо модель не завантажилась
        return generator
    response = generator(user_input, max_length=100, do_sample=True, temperature=0.7)
    return response[0]["generated_text"]

# Інтерфейс Gradio
interface = gr.Interface(
    fn=chat_with_npc,
    inputs="text",
    outputs="text",
    title="Alt Помічник",
    description="Віртуальний помічник для спілкування та підтримки."
)

# Запуск програми
if __name__ == "__main__":
    interface.launch()