import gradio as gr
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

# تحميل النموذج من Hugging Face
model_name = "Salesforce/codegen-350M-mono"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

# دالة توليد الكود
def generate_code(prompt):
    inputs = tokenizer(prompt, return_tensors="pt")
    outputs = model.generate(**inputs, max_length=200, do_sample=True, top_k=50)
    return tokenizer.decode(outputs[0], skip_special_tokens=True)

# واجهة Gradio
interface = gr.Interface(
    fn=generate_code,
    inputs=gr.Textbox(lines=5, label="اكتب وصف الكود هنا"),
    outputs=gr.Textbox(lines=10, label="الكود الناتج"),
    title="مولّد كود Python",
    description="اكتب وصفًا للكود الذي تريده، وسنولّد كودًا باستخدام نموذج ذكاء صناعي."
)

# تشغيل التطبيق
interface.launch()