add app.py

app.py

@@ -0,0 +1,59 @@
+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)
+            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()