IdeaSpark / app.py
Abdallah-Ai's picture
Create app.py
afc9d60 verified
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()}")