Upload 4 files

app.py

@@ -1,83 +1,162 @@
 import gradio as gr
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import torch
+import random
 
-def generate_outline(topic, article_type, length):
+# تهيئة النموذج - سيتم تحميله من Hugging Face
+try:
+    model_name = "aubmindlab/aragpt2-medium"  # نموذج GPT-2 المدرب على اللغة العربية
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+except Exception as e:
+    print(f"Error loading model: {e}")
+    # استخدام نموذج بسيط في حالة الفشل
+    model = None
+    tokenizer = None
+
+def generate_article(topic, article_type, length, style):
     """
-    توليد هيكل المقال بناءً على الموضوع ونوع المقال وطوله
+    توليد مقال كامل بناءً على المدخلات
     """
-    # سيتم تحسين هذه الوظيفة لاحقاً باستخدام نموذج مدرب
-    sections = {
-        "short": 3,
-        "medium": 5,
-        "long": 7
-    }
+    # تحديد طول المقال
+    max_length = {
+        "قصير": 200,
+        "متوسط": 400,
+        "طويل": 800
+    }.get(length, 400)
     
-    num_sections = sections.get(length, 3)
+    # إنشاء المطلب للنموذج
+    prompt = f"""اكتب مقالاً {article_type} عن {topic}. 
+    يجب أن يكون المقال {style} ويتضمن مقدمة وعرض وخاتمة.
+    """
     
-    if article_type == "أكاديمي":
-        outline = [
-            "مقدمة وخلفية نظرية",
-            "مراجعة الأدبيات",
-            "المنهجية",
-            "النتائج والمناقشة",
-            "الخاتمة والتوصيات"
-        ][:num_sections]
-    else:
-        outline = [
-            "مقدمة جذابة",
-            "عرض الفكرة الرئيسية",
-            "التفاصيل والأمثلة",
-            "وجهات النظر المختلفة",
-            "الخاتمة"
-        ][:num_sections]
+    try:
+        if model and tokenizer:
+            # توليد النص باستخدام النموذج
+            inputs = tokenizer.encode(prompt, return_tensors="pt", max_length=100, truncation=True)
+            outputs = model.generate(
+                inputs,
+                max_length=max_length,
+                num_return_sequences=1,
+                no_repeat_ngram_size=2,
+                do_sample=True,
+                top_k=50,
+                top_p=0.95,
+                temperature=0.7,
+                pad_token_id=tokenizer.eos_token_id
+            )
+            
+            generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            return generated_text
+        else:
+            # نموذج بسيط في حالة عدم توفر النموذج الرئيسي
+            return generate_simple_article(topic, article_type, length, style)
+    except Exception as e:
+        return f"حدث خطأ أثناء توليد المقال: {str(e)}"
+
+def generate_simple_article(topic, article_type, length, style):
+    """
+    توليد مقال بسيط في حالة عدم توفر النموذج الرئيسي
+    """
+    # قوالب للمقدمات
+    intros = [
+        f"يعد موضوع {topic} من المواضيع الهامة التي تستحق الدراسة والبحث.",
+        f"في ظل التطورات المتسارعة، يبرز موضوع {topic} كقضية محورية.",
+        f"يثير {topic} اهتماماً متزايداً في الآونة الأخيرة."
+    ]
+    
+    # قوالب للخاتمة
+    conclusions = [
+        f"وختاماً، يمكننا القول أن {topic} يستحق المزيد من الاهتمام والدراسة.",
+        f"في النهاية، نؤكد على أهمية {topic} في حياتنا المعاصرة.",
+        "وبناءً على ما سبق، نوصي بمزيد من البحث والدراسة في هذا المجال."
+    ]
     
-    return "\n".join([f"{i+1}. {section}" for i, section in enumerate(outline)])
+    # إنشاء المقال
+    article = f"""
+    {random.choice(intros)}
+    
+    إن دراسة {topic} تكتسب أهمية خاصة في وقتنا الحاضر، حيث تتعدد جوانبه وتتشعب تأثيراته.
+    ومن الجدير بالذكر أن هناك العديد من العوامل التي تؤثر في هذا الموضوع.
+    
+    يمكننا تحليل {topic} من عدة جوانب:
+    أولاً: الجانب النظري والمفاهيمي
+    ثانياً: التطبيقات العملية
+    ثالثاً: التحديات والفرص المستقبلية
+    
+    {random.choice(conclusions)}
+    """
+    
+    return article
 
 def improve_style(text):
     """
     تحسين أسلوب النص وتقديم اقتراحات
     """
-    # سيتم تحسين هذه الوظيفة لاحقاً باستخدام نموذج مدرب
     suggestions = [
-        "حاول استخدام جمل أقصر وأكثر وضوحاً",
-        "تجنب تكرار الكلمات",
-        "استخدم روابط منطقية بين الفقرات",
-        "أضف أمثلة توضيحية",
+        "✍️ حاول استخدام جمل أقصر وأكثر وضوحاً",
+        "🎯 تجنب تكرار الكلمات واستخدم مرادفات",
+        "🔄 استخدم روابط منطقية بين الفقرات",
+        "💡 أضف أمثلة توضيحية لتعزيز الفكرة",
+        "📝 استخدم علامات الترقيم بشكل صحيح",
+        "🎨 نوع في الأساليب اللغوية",
+        "📚 استشهد بمصادر موثوقة"
     ]
     
     return "\n".join(suggestions)
 
 # إنشاء واجهة المستخدم
 with gr.Blocks(title="مساعد كتابة المقالات العربية") as demo:
-    gr.Markdown("# مساعد كتابة المقالات العربية 📝")
+    gr.Markdown("# 📝 مساعد كتابة المقالات العربية")
+    gr.Markdown("### منصة ذكية لإنشاء وتحسين المقالات العربية")
     
-    with gr.Tab("توليد هيكل المقال"):
+    with gr.Tab("إنشاء مقال جديد"):
         with gr.Row():
-            topic_input = gr.Textbox(label="موضوع المقال")
-            article_type = gr.Dropdown(
-                choices=["أكاديمي", "صحفي", "مدونة"],
-                label="نوع المقال",
-                value="أكاديمي"
-            )
-            length = gr.Dropdown(
-                choices=["short", "medium", "long"],
-                label="طول المقال",
-                value="medium"
-            )
-        generate_btn = gr.Button("توليد الهيكل")
-        outline_output = gr.Textbox(label="هيكل المقال", lines=10)
+            with gr.Column():
+                topic_input = gr.Textbox(
+                    label="موضوع المقال",
+                    placeholder="أدخل موضوع المقال هنا..."
+                )
+                article_type = gr.Dropdown(
+                    choices=["أكاديمي", "صحفي", "مدونة", "تقني"],
+                    label="نوع المقال",
+                    value="أكاديمي"
+                )
+                length = gr.Radio(
+                    choices=["قصير", "متوسط", "طويل"],
+                    label="طول المقال",
+                    value="متوسط"
+                )
+                style = gr.Dropdown(
+                    choices=["رسمي", "أدبي", "تقني", "صحفي"],
+                    label="الأسلوب",
+                    value="رسمي"
+                )
+        
+        generate_btn = gr.Button("إنشاء المقال", variant="primary")
+        article_output = gr.Textbox(
+            label="المقال المُنشأ",
+            lines=10,
+            placeholder="سيظهر المقال هنا..."
+        )
         
     with gr.Tab("تحسين الأسلوب"):
-        text_input = gr.Textbox(label="النص المراد تحسينه", lines=5)
+        text_input = gr.Textbox(
+            label="النص المراد تحسينه",
+            lines=5,
+            placeholder="أدخل النص هنا..."
+        )
         improve_btn = gr.Button("تحسين الأسلوب")
-        suggestions_output = gr.Textbox(label="الاقتراحات", lines=5)
+        suggestions_output = gr.Textbox(
+            label="الاقتراحات",
+            lines=7
+        )
     
     # ربط الأزرار بالوظائف
     generate_btn.click(
-        generate_outline,
-        inputs=[topic_input, article_type, length],
-        outputs=outline_output
+        generate_article,
+        inputs=[topic_input, article_type, length, style],
+        outputs=article_output
     )
     
     improve_btn.click(

requirements.txt

@@ -1,6 +1,7 @@
 transformers==4.35.2
 torch==2.1.1
 gradio==3.50.2
-uvicorn==0.24.0
 python-dotenv==1.0.0
-pydantic==2.5.2
+sentencepiece==0.1.99
+arabic-reshaper==3.0.0
+python-bidi==0.4.2