import streamlit as st
from transformers import pipeline

st.title("مولد أفكار المحتوى الإبداعي")
st.header("أدخل فكرتك وسأقوم بتوليد أفكار جذابة للمحتوى الخاص بك")

# هذا نموذج لغوي عربي قوي من Hugging Face
generator = pipeline('text-generation', model='aubmindlab/aragpt2-large')

user_input = st.text_input("اكتب فكرتك هنا:")

if user_input:
    if st.button("توليد الأفكار"):
        with st.spinner('جاري توليد الأفكار...'):
            generated_text = generator(
                f"أفكار لمحتوى عن {user_input}:",
                max_length=150,
                num_return_sequences=3,
                pad_token_id=generator.tokenizer.eos_token_id
            )
            st.success("تم توليد الأفكار بنجاح!")
            for i, idea in enumerate(generated_text):
                st.write(f"**فكرة {i+1}:** {idea['generated_text'].replace(f'أفكار لمحتوى عن {user_input}:', '', 1).strip()}")