from transformers import pipeline
import streamlit as st

def create_product_description_app():
    st.title("KI-gestützter Produktbeschreibungsgenerator")
    
    # Interface erstellen
    product_name = st.text_input("Produktname:")
    key_features = st.text_area("Hauptmerkmale (durch Kommas getrennt):")
    target_audience = st.text_input("Zielgruppe:")
    
    if st.button("Beschreibung generieren"):
        # Prompt zusammenbauen
        prompt = f"""
        Produkt: {product_name}
        Merkmale: {key_features}
        Zielgruppe: {target_audience}
        Erstelle eine verkaufsfördernde Produktbeschreibung.
        """
        
        # Generator initialisieren
        generator = pipeline('text-generation', 
                           model='gpt2')
        
        # Beschreibung generieren
        description = generator(prompt, 
                              max_length=200, 
                              num_return_sequences=1)[0]['generated_text']
        
        st.write("Generierte Beschreibung:")
        st.write(description)

if __name__ == "__main__":
    create_product_description_app()