import gradio as gr
from transformers import pipeline

# Titel, Beschreibung und Artikel für die Schnittstelle
title = "Prompt Dich weiter mit Einfach.Prompt"
description = """
<p>
<center>

</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>"

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

# Funktion, die die Textgenerierung durchführt
def generate_prompt(prompt):
    output = generator(prompt, max_length=250)
    return output[0]['generated_text']

# Erstellung der Gradio-Schnittstelle
interface = gr.Interface(
    fn=generate_prompt,
    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter your prompt here..."),
    outputs="text",
    title=title,
    description=description,
    article=article,
    layout="vertical",  # Setzt die Anordnung der Elemente auf vertikal
    theme='ParityError/Interstellar'
)

# Starten der Schnittstelle
interface.launch()