import gradio as gr
from transformers import pipeline

# Minimal stable configuration
def load_model():
    return pipeline(
        "text-generation",
        model="distilgpt2",  # Smallest reliable model
        device="cpu",
        framework="pt"
    )

model = load_model()

def generate_text(prompt):
    try:
        response = model(
            prompt,
            max_length=50,
            temperature=0.7,
            num_return_sequences=1
        )
        return response[0]['generated_text']
    except Exception as e:
        return f"Error: {str(e)}"

# Basic interface
iface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt..."),
    outputs=gr.Textbox(),
    title="Stable Text Generator",
    description="Basic but guaranteed-to-work version"
)

iface.launch(server_name="0.0.0.0")