import gradio as gr
from transformers import pipeline

title = "Prompt Dich weiter mit Einfach.Prompt"
description = """
<p>
<center>
Big THX at Merve for this great and first Modell by Bart..
<img src="https://huggingface.co/spaces/EinfachAilex/einfach.prompt/blob/main/cover.png" alt="rick" width="250"/>
</center>
</p>
"""
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1907.06616' target='_blank'>Recipes for building an open-domain chatbot</a></p><p style='text-align: center'><a href='https://parl.ai/projects/recipes/' target='_blank'>Original PARLAI Code</a></p></center></p>"


generator = pipeline("text2text-generation", model="merve/chatgpt-prompts-bart-long")

def generate_prompt(prompt):
    output = generator(prompt, max_length=250)
    return output[0]['generated_text']

interface = gr.Interface(fn=generate_prompt, inputs="text", outputs="text", title="💯 Einfach.Prompt Generator 💯 made with ",
                         description="Wunsch prompt eingeben z.b. Web Designer und einen prompt für Chat.GPT erstellen lassen")

interface.launch()