import gradio as gr
from transformers import pipeline

# تحميل نموذج GPT-Neo 125M
generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", device=-1)

# ذاكرة افتراضية للبوت
memory = []

def generate_post():
    # توليد نص قصير
    prompt = "Write a short English text (<=100 chars):"
    output = generator(prompt, max_new_tokens=30, do_sample=True, temperature=1.0)
    text = output[0]['generated_text'].strip()
    
    # قص النص إذا كان طويل
    if len(text) > 100:
        text = text[:100]
    
    # إضافة للذاكرة
    memory.append(text)
    return text

def stop_and_clear():
    memory.clear()
    return "🛑 Bot stopped and memory cleared!"

with gr.Blocks() as demo:
    gr.Markdown("## 🤖 Auto-writing Bot")
    post_box = gr.Textbox(label="Generated Post", lines=2)
    generate_btn = gr.Button("Generate Post")
    stop_btn = gr.Button("Stop & Clear Memory")
    
    generate_btn.click(fn=generate_post, outputs=post_box)
    stop_btn.click(fn=stop_and_clear, outputs=post_box)

demo.launch()