import gradio as gr
from transformers import pipeline

models = ["gpt2", "microsoft/phi-2","meta-llama/Llama-2-7b-chat-hf" ]

def generate_text(prompt, model, max_length, temperature):
    pipe = pipeline("text-generation", model=model,trust_remote_code = True)
    try:
        generated_text = pipe(prompt, max_length=max_length, temperature=temperature)[0]["generated_text"]
        return generated_text
    except Exception as e:
        return f"An error occurred: {str(e)}"

gr.Interface(
   generate_text,   
   [
        gr.Textbox(label="Enter your Prompt"),
        gr.Dropdown(models, label="choose your Model" ),
        gr.Slider(minimum=50, maximum=650, step=50),
        gr.Slider(minimum=0.1, maximum=1, step=0.1),
   ],
   outputs="text"
).launch()