AdamAbde/code-generator-tool

1import streamlit as st2from transformers import pipeline3 4# تحميل نموذج توليد الكود من Hugging Face5code_generator = pipeline("text-generation", model="Salesforce/codegen-350M-mono")6 7def generate_code(description):8    """9    توليد الكود بناءً على الوصف باستخدام نموذج CodeGen.10    """11    prompt = f"# Python code for: {description}\n"12    generated_code = code_generator(prompt, max_length=300, num_return_sequences=1)13    return generated_code[0]['generated_text']14 15def save_code(code, filename="generated_code.py"):16    """17    حفظ الكود المولد في ملف.18    """19    with open(filename, "w", encoding="utf-8") as file:20        file.write(code)21    return filename22 23# واجهة المستخدم باستخدام Streamlit24st.title("أداة تصميم البرامج حسب الطلب")25st.write("أدخل وصف البرنامج الذي تريده، وسنقوم بإنشاء الكود لك.")26 27# إدخال وصف البرنامج28description = st.text_area("وصف البرنامج:", height=150)29 30if st.button("إنشاء البرنامج"):31    if description:32        with st.spinner("جاري توليد الكود..."):33            # توليد الكود34            code = generate_code(description)35            st.code(code)36            37            # حفظ الكود38            filename = save_code(code)39            st.success(f"تم حفظ الكود في ملف: {filename}")40    else:41        st.error("الرجاء إدخال وصف للبرنامج.")