Update app.py

app.py

@@ -1,37 +1,58 @@
 import gradio as gr
 from transformers import pipeline
+import langid  # مكتبة للكشف عن اللغة
 
-summarizer_en = pipeline("summarization", model="facebook/bart-large-cnn")
-summarizer_ar = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en")
-translator_ar = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar")
+# تحميل نماذج التلخيص
+summarizer_en = pipeline("summarization", model="facebook/bart-large-cnn")  
+summarizer_ar = pipeline("summarization", model="eslamxm/mt5-base-finetuned-persian-finetuned-persian-arabic") 
 
 def summarize_text(text, language):
     if not text.strip():
-        return " الرجاء إدخال نص للتلخيص."
+        return "Please enter text." if language == "English" else "الرجاء إدخال نص."
     
-    if language == "العربية":
-
-        translated_text = summarizer_ar(text)[0]['translation_text']
-        summary = summarizer_en(translated_text, min_length=10, max_length=100, do_sample=False)
-
-        final_summary = translator_ar(summary[0]['summary_text'])[0]['translation_text']
+    # الكشف عن لغة النص المدخل
+    detected_language, _ = langid.classify(text)
+    
+    # إذا كانت اللغة الإنجليزية
+    if detected_language == "en":
+        summary = summarizer_en(text, max_length=100, min_length=10, do_sample=False)[0]['summary_text']
+    # إذا كانت اللغة العربية
+    elif detected_language == "ar":
+        summary = summarizer_ar(text, max_length=100, min_length=10, do_sample=False)[0]['summary_text']
     else:
-        summary = summarizer_en(text, min_length=10, max_length=100, do_sample=False)
-        final_summary = summary[0]['summary_text']
+        return "Unsupported language."
     
-    return final_summary
+    return summary
+
+# gradio
+def translate_ui(language):
+    return {
+        "title": "👋 مرحبًا بك في أداة تلخيص النصوص!" if language == "العربية" else "👋 Welcome to the Text Summarization Tool!",
+        "summarize_btn": " لخص النص" if language == "العربية" else " Summarize Text",
+        "text_input_label": "أدخل النص" if language == "العربية" else "Enter text",
+        "text_output_label": "النتيجة" if language == "العربية" else "Result"
+    }
+
+def update_ui(language):
+    texts = translate_ui(language)
+    return texts["title"], texts["summarize_btn"], texts["text_input_label"], texts["text_output_label"]
 
-# Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("# Text Summarization App")
-    gr.Markdown("Enter a long text below, choose the language, and get a summarized version!")
-    
-    input_text = gr.Textbox(label="Input Text", placeholder="Enter your text here ^^ ...")
-    language = gr.Dropdown(choices=["العربية", "English"], value="English", label="Select Language")
-    summarize_btn = gr.Button("Summarize")
-    output_text = gr.Textbox(label="Summarized Text")
+    lang_toggle = gr.Radio(["العربية", "English"], label="🌍 اختر لغة الواجهة", value="العربية")
+    title = gr.Markdown("## 👋 مرحبًا بك في أداة تلخيص النصوص")
     
-    summarize_btn.click(summarize_text, inputs=[input_text, language], outputs=output_text)
+    with gr.Row():
+        with gr.Column():
+            text_input_label = gr.Markdown("أدخل النص") 
+            text_input = gr.Textbox(label="") 
+            summarize_btn = gr.Button("لخص النص")
+        
+        with gr.Column():
+            text_output_label = gr.Markdown("النتيجة")  
+            text_output = gr.Textbox(label="")
+
+    # تغيير النصوص عند تغيير اللغة
+    lang_toggle.change(fn=lambda lang: update_ui(lang), inputs=lang_toggle, outputs=[title, summarize_btn, text_input_label, text_output_label])
+    summarize_btn.click(fn=summarize_text, inputs=[text_input, lang_toggle], outputs=text_output)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()