Upload app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import torch
 import random
+import arabic_reshaper
+from bidi.algorithm import get_display
 
 # تهيئة النموذج - سيتم تحميله من Hugging Face
 try:
@@ -14,45 +16,64 @@ except Exception as e:
     model = None
     tokenizer = None
 
-def generate_article(topic, article_type, length, style):
+def format_arabic_text(text):
     """
-    توليد مقال كامل بناءً على المدخلات
+    تنسيق النص العربي بشكل صحيح
     """
-    # تحديد طول المقال
-    max_length = {
-        "قصير": 200,
-        "متوسط": 400,
-        "طويل": 800
-    }.get(length, 400)
-    
-    # إنشاء المطلب للنموذج
-    prompt = f"""اكتب مقالاً {article_type} عن {topic}. 
-    يجب أن يكون المقال {style} ويتضمن مقدمة وعرض وخاتمة.
+    try:
+        reshaped_text = arabic_reshaper.reshape(text)
+        bidi_text = get_display(reshaped_text)
+        return bidi_text
+    except:
+        return text
+
+def generate_article(topic, article_type, length, style):
+    """
+    توليد مقال كامل بناءً على المدخلات مع التأكد من صحة النص العربي
     """
-    
     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)
+        # إنشاء هيكل المقال
+        sections = {
+            "مقدمة": [
+                f"يُعد {topic} من أهم المجالات التي تشغل العالم في العصر الحديث.",
+                f"في ظل التطورات التكنولوجية المتسارعة، يبرز {topic} كموضوع محوري يستحق الدراسة والبحث.",
+                f"يشهد العالم اهتماماً متزايداً بـ{topic} نظراً لتأثيره الكبير على مختلف مجالات الحياة."
+            ],
+            "عرض": [
+                f"يمكن تحليل {topic} من خلال عدة محاور رئيسية:",
+                "أولاً: الأسس النظرية والمفاهيم الأساسية",
+                "ثانياً: التطبيقات العملية والتجارب الحديثة",
+                "ثالثاً: التحديات والفرص المستقبلية",
+                f"\nمن الجوانب المهمة في {topic}:",
+                "• التطور التاريخي والنشأة",
+                "• الأدوات والتقنيات المستخدمة",
+                "• الآثار الاجتماعية والاقتصادية",
+                "\nوتشير الدراسات الحديثة إلى أن..."
+            ],
+            "خاتمة": [
+                f"وختاماً، يمكننا القول أن {topic} يمثل تحولاً جوهرياً في مجال التكنولوجيا والعلوم.",
+                f"في النهاية، نؤكد على أهمية مواصلة البحث والتطوير في مجال {topic}.",
+                "وبناءً على ما سبق، نوصي بتكثيف الجهود البحثية وتعزيز التعاون الدولي في هذا المجال."
+            ]
+        }
+
+        # اختيار محتوى المقال بناءً على النوع والطول
+        intro = random.choice(sections["مقدمة"])
+        body = "\n".join(sections["عرض"][:5 if length == "قصير" else 8 if length == "متوسط" else -1])
+        conclusion = random.choice(sections["خاتمة"])
+
+        # تجميع المقال
+        article = f"""
+{intro}
+
+{body}
+
+{conclusion}
+"""
+        # تنسيق النص العربي
+        return format_arabic_text(article)
     except Exception as e:
-        return f"حدث خطأ أثناء توليد المقال: {str(e)}"
+        return f"حدث خطأ أثناء إنشاء المقال: {str(e)}"
 
 def generate_simple_article(topic, article_type, length, style):
     """