from kivy.app import App
from kivy.uix.boxlayout import BoxLayout
from transformers import pipeline

# Load the GPT-2 model
generator = pipeline("text-generation", model="gpt2")

class TextGenLayout(BoxLayout):
    def generate(self):
        user_prompt = self.ids.prompt_input.text
        if not user_prompt.strip():
            self.ids.output_box.text = "⚠️ Please enter some text."
            return
        result = generator(user_prompt, max_length=80, num_return_sequences=1)
        generated_text = result[0]["generated_text"]
        self.ids.output_box.text = generated_text

class TextGenApp(App):
    def build(self):
        return TextGenLayout()