Shatha2030 commited on
Commit
ae79b2b
·
verified ·
1 Parent(s): ebbd0a6

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +62 -0
app.py ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+ import langid # مكتبة للكشف عن اللغة
4
+
5
+ # تحميل نماذج التلخيص
6
+ summarizer_en = pipeline("summarization", model="facebook/bart-large-cnn")
7
+ summarizer_ar = pipeline("summarization", model="malmarjeh/t5-arabic-text-summarization")
8
+
9
+ def summarize_text(text, language, min_length, max_length):
10
+ if not text.strip():
11
+ return "Please enter text." if language == "English" else "الرجاء إدخال نص."
12
+
13
+ # الكشف عن لغة النص المدخل
14
+ detected_language, _ = langid.classify(text)
15
+
16
+ # إذا كانت اللغة الإنجليزية
17
+ if detected_language == "en":
18
+ summary = summarizer_en(text, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
19
+ # إذا كانت اللغة العربية
20
+ elif detected_language == "ar":
21
+ summary = summarizer_ar(text, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
22
+ else:
23
+ return "Unsupported language."
24
+
25
+ return summary
26
+
27
+ # gradio
28
+ def translate_ui(language):
29
+ return {
30
+ "title": "👋 مرحبًا بك في أداة تلخيص النصوص!" if language == "العربية" else "👋 Welcome to the Text Summarization Tool!",
31
+ "summarize_btn": " لخص النص" if language == "العربية" else " Summarize Text",
32
+ "text_input_label": "أدخل النص" if language == "العربية" else "Enter text",
33
+ "text_output_label": "النتيجة" if language == "العربية" else "Result",
34
+ "min_length_label": "الحد الأدنى للطول" if language == "العربية" else "Minimum Length",
35
+ "max_length_label": "الحد الأقصى للطول" if language == "العربية" else "Maximum Length"
36
+ }
37
+
38
+ def update_ui(language):
39
+ texts = translate_ui(language)
40
+ return texts["title"], texts["summarize_btn"], texts["text_input_label"], texts["text_output_label"], texts["min_length_label"], texts["max_length_label"]
41
+
42
+ with gr.Blocks() as demo:
43
+ lang_toggle = gr.Radio(["العربية", "English"], label="🌍 اختر لغة الواجهة", value="العربية")
44
+ title = gr.Markdown("## 👋 مرحبًا بك في أداة تلخيص النصوص")
45
+
46
+ with gr.Row():
47
+ with gr.Column():
48
+ text_input_label = gr.Markdown("أدخل النص")
49
+ text_input = gr.Textbox(label="")
50
+ min_length_slider = gr.Slider(10, 50, step=1, label="Minimum Length")
51
+ max_length_slider = gr.Slider(50, 150, step=1, label="Maximum Length")
52
+ summarize_btn = gr.Button("لخص النص")
53
+
54
+ with gr.Column():
55
+ text_output_label = gr.Markdown("النتيجة")
56
+ text_output = gr.Textbox(label="")
57
+
58
+ # تغيير النصوص عند تغيير اللغة
59
+ 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])
60
+ summarize_btn.click(fn=summarize_text, inputs=[text_input, lang_toggle, min_length_slider, max_length_slider], outputs=text_output)
61
+
62
+ demo.launch()