import gradio as gr
from transformers import pipeline, set_seed

# Create text generation pipeline with GPT-2
generator = pipeline('text-generation', model='gpt2')

def generate_text(prompt, temperature, max_length, instruction):
    set_seed(42)
    result = generator(prompt + "\nAI: ", max_length=max_length, num_return_sequences=1, temperature=temperature)
    return result[0]['generated_text']

# Define Gradio interface
iface = gr.Interface(
    fn=generate_text, 
    inputs=[gr.inputs.Textbox(lines=3, label="Your Message"),
            gr.inputs.Slider(minimum=0.1, maximum=1.0, default=0.5, label="Temperature"),
            gr.inputs.Slider(minimum=10, maximum=200, default=100, label="Max Length"),
            gr.inputs.Textbox(default="This is a chat between the user and a truthful and intelligent AI. AI must answer very truthfully and accurately. The chat begins now.\nUser: hi\nAI: Hello, what do you need assistance with?\nUser: ", lines=2, label="AI Instruction")],
    outputs=gr.outputs.Textbox(label="AI Response"),
    layout="vertical",
    title="Chat with GPT-2",
    description="You can chat with the old GPT-2! Temperature and token limits are adjustable.",
)

iface.launch()