import gradio as gr
from transformers import pipeline


model_name = "curiouscurrent/omnicode"
text_generator = pipeline("text-generation", model=model_name)

history = []

def generate_response(prompt):
    history.append(prompt)
    final_prompt = "\n".join(history)
    
    # Generate response
    response = text_generator(final_prompt, max_length=100)[0]['generated_text']
    
    return response

interface = gr.Interface(
    fn=generate_response,
    inputs=gr.inputs.Textarea(lines=4, placeholder="Enter your Prompt"),
    outputs="text",
    title="Text Generation App",
    description="Generate text based on the input prompt."
)
interface.launch()