xzxyx's picture
Duplicate from malmarjeh/arabic-text-summarization
cfa1e90
from urllib.parse import unquote
import arabic_reshaper
import streamlit as st
from bidi.algorithm import get_display
st.set_page_config(
page_title="Arabic Text Summarization",
page_icon="📖",
initial_sidebar_state="expanded"
# layout="wide"
)
from summarize import get_results
rtl = lambda w: get_display(f"{arabic_reshaper.reshape(w)}")
st.title("تَلْخِيصُ اَلنُّصُوصِ بِاللُّغَةِ اَلْعَرَبِيَّةِ")
st.markdown(
"""
<style>
@import url(https://fonts.googleapis.com/earlyaccess/scheherazade.css);
section.main {
background-color: beige;
}
.stMarkdown h1, .main .element-container.css-o7ulmj.e1tzin5v3 {
text-align: right;
}
.stMarkdown div.css-nlntq9.e16nr0p33 {
font-weight: bold;
}
textarea {
direction: rtl;
height: 140px;
}
.stTextArea .css-qrbaxs {
float: right;
font-size: 23px;
}
h1 {
font-family: 'Scheherazade', serif;
}
.main div.css-nlntq9.e16nr0p33 > p {
direction: rtl;
}
.main .stMarkdown div.css-nlntq9 p {
font-size: 22px;
}
.main .stMarkdown div.css-nlntq9 {
direction: rtl;
}
.main p, .main div, .main input, .main label {
text-align: right;
direction: rtl;
}
.main div>h1>div {
left: 0;
}
.main button {
font-size: 22px;
}
</style>
""",
unsafe_allow_html=True,
)
st.sidebar.write("Arabic Text Summarization")
st.sidebar.write("Contact: banimarje@gmail.com")
st.sidebar.write("\n")
model_selected = st.sidebar.selectbox(
'Select a model',
('T5','BERT2BERT', 'GPT-2', 'mBERT2mBERT','Transformer'))
st.sidebar.write("\n")
num_beams = st.sidebar.slider(
"Number of beams", min_value=1, max_value=10, value=3, step=1
)
length_pe_slider_disabled = False
if model_selected == "GPT-2":
length_pe_slider_disabled = True
st.sidebar.write("\n")
length_penalty = st.sidebar.slider(
"Length penalty ", min_value=0.1, max_value=3.0, value=1.0, step=0.1, disabled=length_pe_slider_disabled
)
txt = """يجري علماء في بريطانيا تجربة لاختبار فعالية عقار إيبوبروفين لمساعدة المصابين بفيروس كورونا. وذكرت هيئة الإذاعة البريطانية "بي بي سي" أن فريق مشترك من أطباء مستشفيات "جاي" و"سانت توماس" و"كينغز كوليدج" في لندن يعتقد أن إيبوبروفين، وهو مضاد للالتهابات ومسكن للألم، يمكن أن يعالج صعوبات التنفس.
ويأمل العلماء أن يساعد هذا العلاج المنخفض التكلفة المرضى في الاستغناء عن أجهزة التنفس الصناعي. وذكرت أنه خلال فترة الاختبار، سيحصل نصف المرضى على إيبوبروفين بالإضافة إلى الرعاية المعتادة، حيث سيتم استخدام تركيبة خاصة من إيبوبروفين بدلا من الأقراص العادية التي قد يشتريها الناس عادة."""
text = st.text_area("أدخل نص ليتم تلخيصه", value=txt)
run_query = st.button("لخّص")
if run_query:
# https://discuss.streamlit.io/t/showing-a-gif-while-st-spinner-runs/5084
with st.spinner("جاري التلخيص ..."):
result = get_results(text, model_selected, num_beams, length_penalty)
if len(result) > 0:
st.write(result)
else:
st.write("")