Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 import langid  # مكتبة للكشف عن اللغة
 
-# تحميل نماذج التلخيص
+
 summarizer_en = pipeline("summarization", model="facebook/bart-large-cnn")
 summarizer_ar = pipeline("summarization", model="malmarjeh/t5-arabic-text-summarization")
 
@@ -10,21 +10,29 @@ def summarize_text(text, language, min_length, max_length):
     if not text.strip():
         return "Please enter text." if language == "English" else "الرجاء إدخال نص."
 
-    # الكشف عن لغة النص المدخل
     detected_language, _ = langid.classify(text)
 
-    # إذا كانت اللغة الإنجليزية
     if detected_language == "en":
-        summary = summarizer_en(text, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
-    # إذا كانت اللغة العربية
+        summary = summarizer_en(
+            text, 
+            max_length=int(max_length), 
+            min_length=int(min_length), 
+            do_sample=False, 
+            no_repeat_ngram_size=2
+        )[0]['summary_text']
     elif detected_language == "ar":
-        summary = summarizer_ar(text, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
+        summary = summarizer_ar(
+            text, 
+            max_length=int(max_length), 
+            min_length=int(min_length), 
+            do_sample=False, 
+            no_repeat_ngram_size=2
+        )[0]['summary_text']
     else:
         return "Unsupported language."
 
     return summary
 
-# gradio
 def translate_ui(language):
     return {
         "title": "👋 مرحبًا بك في أداة تلخيص النصوص!" if language == "العربية" else "👋 Welcome to the Text Summarization Tool!",
@@ -37,26 +45,97 @@ def translate_ui(language):
 
 def update_ui(language):
     texts = translate_ui(language)
-    return texts["title"], texts["summarize_btn"], texts["text_input_label"], texts["text_output_label"], texts["min_length_label"], texts["max_length_label"]
+    return (
+        texts["title"], 
+        texts["summarize_btn"], 
+        texts["text_input_label"], 
+        texts["text_output_label"], 
+        texts["min_length_label"], 
+        texts["max_length_label"]
+    )
 
 with gr.Blocks() as demo:
-    lang_toggle = gr.Radio(["العربية", "English"], label="🌍 اختر لغة الواجهة", value="العربية")
-    title = gr.Markdown("## 👋 مرحبًا بك في أداة تلخيص النصوص")
+    lang_toggle = gr.Radio(
+        ["العربية", "English"], 
+        label="🌍 اختر لغة الواجهة", 
+        value="العربية", 
+        elem_id="language-toggle"
+    )
+    title = gr.Markdown(
+        "## 👋 مرحبًا بك في أداة تلخيص النصوص", 
+        elem_id="title"
+    )
 
     with gr.Row():
+    
         with gr.Column():
-            text_input_label = gr.Markdown("أدخل النص")
-            text_input = gr.Textbox(label="")
-            min_length_slider = gr.Slider(10, 50, step=1, label="Minimum Length")
-            max_length_slider = gr.Slider(50, 150, step=1, label="Maximum Length")
-            summarize_btn = gr.Button("لخص النص", variant="primary")
+            text_input_label = gr.Markdown(
+                "أدخل النص", 
+                elem_id="text-input-label"
+            )
+            text_input = gr.Textbox(
+                label="", 
+                elem_id="text-input"
+            )
+
+            min_length_label = gr.Markdown(
+                "الحد الأدنى للطول", 
+                elem_id="min-length-label"
+            )
+            min_length = gr.Slider(
+                minimum=10, 
+                maximum=50, 
+                value=10, 
+                step=1, 
+                label="", 
+                elem_id="min-length-slider"
+            )
 
+            max_length_label = gr.Markdown(
+                "الحد الأقصى للطول", 
+                elem_id="max-length-label"
+            )
+            max_length = gr.Slider(
+                minimum=50, 
+                maximum=150, 
+                value=100, 
+                step=1, 
+                label="", 
+                elem_id="max-length-slider"
+            )
+
+            summarize_btn = gr.Button(
+                "لخص النص", 
+                elem_id="summarize-btn", variant="primary"
+            )
+        
         with gr.Column():
-            text_output_label = gr.Markdown("النتيجة")
-            text_output = gr.Textbox(label="")
+            text_output_label = gr.Markdown(
+                "النتيجة", 
+                elem_id="text-output-label"
+            )
+            text_output = gr.Textbox(
+                label="", 
+                elem_id="text-output"
+            )
+
+    lang_toggle.change(
+        fn=lambda lang: update_ui(lang), 
+        inputs=lang_toggle,
+        outputs=[
+            title, 
+            summarize_btn, 
+            text_input_label, 
+            text_output_label, 
+            min_length_label, 
+            max_length_label
+        ]
+    )
 
-    # تغيير النصوص عند تغيير اللغة
-    lang_toggle.change(fn=lambda lang: update_ui(lang), inputs=lang_toggle, outputs=[title, summarize_btn, text_input_label, text_output_label, min_length_slider, max_length_slider])
-    summarize_btn.click(fn=summarize_text, inputs=[text_input, lang_toggle, min_length_slider, max_length_slider], outputs=text_output)
+    summarize_btn.click(
+        fn=summarize_text, 
+        inputs=[text_input, lang_toggle, min_length, max_length], 
+        outputs=text_output
+    )
 
 demo.launch()