AdamAbde's picture
app.py
d3c380e verified
import streamlit as st
from transformers import pipeline
# تحميل نموذج توليد الكود من Hugging Face
code_generator = pipeline("text-generation", model="Salesforce/codegen-350M-mono")
def generate_code(description):
"""
توليد الكود بناءً على الوصف باستخدام نموذج CodeGen.
"""
prompt = f"# Python code for: {description}\n"
generated_code = code_generator(prompt, max_length=300, num_return_sequences=1)
return generated_code[0]['generated_text']
def save_code(code, filename="generated_code.py"):
"""
حفظ الكود المولد في ملف.
"""
with open(filename, "w", encoding="utf-8") as file:
file.write(code)
return filename
# واجهة المستخدم باستخدام Streamlit
st.title("أداة تصميم البرامج حسب الطلب")
st.write("أدخل وصف البرنامج الذي تريده، وسنقوم بإنشاء الكود لك.")
# إدخال وصف البرنامج
description = st.text_area("وصف البرنامج:", height=150)
if st.button("إنشاء البرنامج"):
if description:
with st.spinner("جاري توليد الكود..."):
# توليد الكود
code = generate_code(description)
st.code(code)
# حفظ الكود
filename = save_code(code)
st.success(f"تم حفظ الكود في ملف: {filename}")
else:
st.error("الرجاء إدخال وصف للبرنامج.")