import gradio as gr
from transformers import pipeline
import pyttsx3

# Inicializace modelu Hugging Face pro generování textu
generator = pipeline("text-generation", model="gpt2")

# TTS: Funkce pro syntézu hlasu
engine = pyttsx3.init()
def speak(text):
    engine.say(text)
    engine.runAndWait()

# Funkce pro zpracování uživatelského vstupu
def chatbot(user_input):
    response = generator(user_input, max_length=50, num_return_sequences=1)
    generated_text = response[0]["generated_text"]
    speak(generated_text)  # Přečte odpověď
    return generated_text

# Vytvoření jednoduchého uživatelského rozhraní
interface = gr.Interface(
    fn=chatbot,
    inputs=gr.Textbox(label="Zadej svou otázku:"),
    outputs=gr.Textbox(label="Odpověď od chatbota:"),
    live=True
)

# Spuštění aplikace
interface.launch()