from transformers import pipeline
import gradio as gr

# تحميل النموذج
pipe = pipeline("text-generation", model="nlpcloud/instruct-gpt-j-fp16")

# دالة توليد النص
def generate_text(prompt, max_length=200, temp=0.7):
    result = pipe(prompt, max_length=max_length, temperature=temp, do_sample=True)
    return result[0]["generated_text"]

# واجهة Gradio
iface = gr.Interface(
    fn=generate_text,
    inputs=[
        gr.Textbox(label="سؤالك أو البرومبت"),
        gr.Slider(10, 500, value=200, step=10, label="Max Length"),
        gr.Slider(0.1, 1.5, value=0.7, step=0.1, label="Temperature")
    ],
    outputs="text",
    title="🧠 GPT-J Ethical Test",
    description="اختبر النموذج وشوف هل يجاوب على أسئلة غير قانونية أم لا"
)

iface.launch()