import gradio as gr
from transformers import pipeline

messages = [
    {"role": "system", "content": "You are a chatbot that responds in 10 lines of shakespeare given a prompt"}
]
pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
    
def poet(input):
    messages.append(
    {"role": "user", "content": input})
    generated_text = pipe(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
    return generated_text
    

chat = gr.ChatInterface(
    fn=generate_shakespeare,
    title="Shakespearean Text Generator",
    description="Enter a prompt and let the DeepSeek model generate Shakespearean-style text.",
)

if __name__ == "__main__":
    chat.launch()