Upload app.py

app.py

@@ -454,12 +454,20 @@ def check_grammar_and_style(text):
     
     return text
 
-def generate_article(topic, article_type):
+def generate_article(topic, article_type, word_count=1500, formality="رسمي"):
     template = get_article_template(article_type)
     structure = template["structure"]
     style = template["style"]
     transitions = template["transitions"]
     
+    # تعديل الأسلوب حسب مستوى الرسمية
+    if formality == "رسمي":
+        style = "أكاديمي رسمي"
+    elif formality == "شبه رسمي":
+        style = "إخباري موضوعي"
+    else:  # عادي
+        style = "تعليمي تفاعلي"
+    
     # إنشاء المقال الأساسي
     article = {
         "title": generate_title(topic, style),
@@ -469,14 +477,61 @@ def generate_article(topic, article_type):
         "faq": generate_faq(topic, style)
     }
     
-    # تحسين المحتوى
+    # تنسيق المقال
     formatted_text = format_article(article)
+    
+    # تحسين المحتوى
     enhanced_text = enhance_text_variation(formatted_text)
     humanized_text = add_human_touch(enhanced_text)
     final_text = check_grammar_and_style(humanized_text)
     
+    # ضبط عدد الكلمات
+    current_word_count = len(final_text.split())
+    if current_word_count < word_count:
+        # إضافة محتوى إضافي
+        additional_content = generate_additional_content(topic, style, word_count - current_word_count)
+        final_text = insert_additional_content(final_text, additional_content)
+    elif current_word_count > word_count:
+        # تقليص المحتوى مع الحفاظ على التماسك
+        final_text = trim_content(final_text, word_count)
+    
     return final_text
 
+def generate_additional_content(topic, style, words_needed):
+    """توليد محتوى إضافي للوصول إلى عدد الكلمات المطلوب"""
+    sections = [
+        f"وفي سياق متصل، يمكن الإشارة إلى أن {topic} يشهد تطورات مستمرة",
+        f"ومن الجدير بالذكر أن دراسات حديثة في مجال {topic} قد أظهرت نتائج مثيرة للاهتمام",
+        f"وعلاوة على ذلك، فإن التطبيقات العملية لـ {topic} تتنوع بشكل كبير"
+    ]
+    return " ".join(sections[:words_needed//50])  # تقريب تقديري لعدد الكلمات المطلوبة
+
+def trim_content(text, target_word_count):
+    """تقليص المحتوى للوصول إلى عدد الكلمات المطلوب"""
+    words = text.split()
+    if len(words) <= target_word_count:
+        return text
+    
+    # الحفاظ على المقدمة والخاتمة
+    paragraphs = text.split("\n\n")
+    intro = paragraphs[0]
+    conclusion = paragraphs[-1]
+    
+    # تقليص المحتوى الرئيسي
+    main_content = "\n\n".join(paragraphs[1:-1])
+    main_words = main_content.split()
+    words_to_keep = target_word_count - len(intro.split()) - len(conclusion.split())
+    trimmed_main = " ".join(main_words[:words_to_keep])
+    
+    return f"{intro}\n\n{trimmed_main}\n\n{conclusion}"
+
+def insert_additional_content(text, additional_content):
+    """إدراج المحتوى الإضافي في المكان المناسب من المقال"""
+    paragraphs = text.split("\n\n")
+    # إدراج المحتوى قبل الخاتمة
+    paragraphs.insert(-2, additional_content)
+    return "\n\n".join(paragraphs)
+
 def format_article(article):
     """تنسيق المقال بشكل جميل"""
     formatted_text = f"""# {article['title']}
@@ -568,12 +623,18 @@ def generate_faq(topic, style):
 
 # إنشاء واجهة المستخدم
 with gr.Blocks(title="مساعد كتابة المقالات العربية") as demo:
-    gr.Markdown("# 📝 مساعد كتابة المقالات العربية")
-    gr.Markdown("### منصة ذكية لإنشاء وتحسين المقالات العربية")
+    gr.Markdown("""
+    # 📝 مساعد كتابة المقالات العربية
+    ### نظام متطور لإنشاء مقالات عربية احترافية ومتنوعة
+    """)
     
     with gr.Tab("إنشاء مقال احترافي"):
         with gr.Row():
-            topic_input = gr.Textbox(label="موضوع المقال", placeholder="أدخل موضوع المقال هنا...")
+            topic_input = gr.Textbox(
+                label="موضوع المقال",
+                placeholder="أدخل موضوع المقال هنا...",
+                lines=2
+            )
             article_type = gr.Dropdown(
                 choices=["علمي", "صحفي", "تقني", "تعليمي", "تسويقي", "اجتماعي", "رأي", "ثقافي"],
                 label="نوع المقال",
@@ -582,7 +643,7 @@ with gr.Blocks(title="مساعد كتابة المقالات العربية") as
         
         with gr.Row():
             word_count_slider = gr.Slider(
-                minimum=1000,
+                minimum=500,
                 maximum=3000,
                 value=1500,
                 step=100,
@@ -606,16 +667,17 @@ with gr.Blocks(title="مساعد كتابة المقالات العربية") as
         
         word_count_output = gr.Textbox(label="عدد الكلمات الفعلي")
         
+        def generate_and_format(topic, article_type, word_count, formality):
+            article = generate_article(topic, article_type, word_count, formality)
+            return article
+        
         def count_words(text):
             words = text.split()
             return f"عدد الكلمات: {len(words)}"
         
-        def copy_article(text):
-            return text
-        
         generate_btn.click(
-            fn=generate_article,
-            inputs=[topic_input, article_type],
+            fn=generate_and_format,
+            inputs=[topic_input, article_type, word_count_slider, formality_level],
             outputs=[output]
         ).then(
             fn=count_words,
@@ -624,7 +686,7 @@ with gr.Blocks(title="مساعد كتابة المقالات العربية") as
         )
         
         copy_btn.click(
-            fn=copy_article,
+            fn=lambda x: x,
             inputs=[output],
             outputs=[output]
         )