Spaces:
Sleeping
Sleeping
File size: 2,760 Bytes
6f7e68b abab12b 6f7e68b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import streamlit as st
import models
def main():
st.set_page_config(layout="wide", page_title="Review Toxicity Checker")
st.title('Review Toxicity Checker')
left_col, right_col = st.columns(2)
content = left_col.empty()
with st.sidebar:
model_id = st.radio(
"Select a model option:",
["stabilityai/stablelm-zephyr-3b"]
)
hf_api_key = st.text_input('HF API Key\nhttps://huggingface.co/settings/tokens')
color = st.color_picker('Highlight Color', '#F44336')
with right_col:
st.markdown("<p style='text-align: left; color: white; margin-bottom: 5px; font-size: 14px;'>Revised review:</p>", unsafe_allow_html=True)
with left_col:
input_text = st.text_area('Enter your review here:', height=500)
left_col2, right_col2 = st.columns([1,3])
with left_col2:
check = st.button('Check Review')
with right_col2:
clear = st.button('Clear Text')
if check:
with st.spinner('Processing review...'):
revision = models.revise_review(input_text, hf_api_key, model_id, color)
print(revision)
try:
with right_col:
print(revision['data']['revision'])
st.markdown(f""" <div style="border:1px solid white; padding:10px; height:500px; overflow:auto; border-radius:7px;"> {revision['data']['revision']} </div> """, unsafe_allow_html=True)
num_revised = int(revision['data']['revised_sentences'])/int(revision['data']['sentence_count'])
left_col2, right_col2 = st.columns([1,2])
st.write(f"Sentences Revised: {revision['data']['revised_sentences']}/{revision['data']['sentence_count']}")
st.progress(num_revised)
score = revision['data']['score']
if score == 1:
st.write(f"Score: 1 (toxic)")
if score == 0:
st.write(f"Score: 0 (non-toxic)")
st.progress(int(revision['data']['score'])/1)
except Exception as e:
with right_col:
st.error("An error occured.")
st.error(e)
elif clear:
with right_col:
st.markdown(f""" <div style="border:1px solid white; padding:10px; height:500px; overflow:auto; border-radius:7px;"> </div> """, unsafe_allow_html=True)
else:
with right_col:
st.markdown(f""" <div style="border:1px solid white; padding:10px; height:500px; overflow:auto; border-radius:7px;"> </div> """, unsafe_allow_html=True)
if __name__ == "__main__":
main()
|