try3

app.py

@@ -28,9 +28,8 @@ def load_image_from_url(url):
     img = Image.open(BytesIO(response.content))
     return img
 
-# Evaluation fragment
-@st.experimental_fragment
-def evaluate_text(user_input, result_container):
+# Evaluation function
+def evaluate_text(user_input):
     if user_input:
         # Get model and tokenizer from session state
         tokenizer, model = st.session_state.model_and_tokenizer
@@ -48,13 +47,8 @@ def evaluate_text(user_input, result_container):
         # Determine prediction
         prediction = 'unsafe' if 'unsafe' in output_decoded.lower() else 'safe'
         
-        # Display results
-        with result_container:
-            st.subheader("Evaluation Result:")
-            st.write(f"The text is evaluated as: **{prediction.upper()}**")
-    else:
-        with result_container:
-            st.warning("Please enter some text to evaluate.")
+        return prediction
+    return None
 
 # Streamlit app
 st.title("Text Safety Evaluator")
@@ -70,16 +64,25 @@ user_input = st.text_area("Enter the text you want to evaluate:", height=100)
 result_container = st.empty()
 
 if st.button("Evaluate"):
-    evaluate_text(user_input, result_container)
+    prediction = evaluate_text(user_input)
+    if prediction:
+        result_container.subheader("Evaluation Result:")
+        result_container.write(f"The text is evaluated as: **{prediction.upper()}**")
+    else:
+        result_container.warning("Please enter some text to evaluate.")
 
-# Add logo at the bottom center
-col1, col2, col3 = st.columns([1,2,1])
-with col2:
-    logo_url = "https://github.com/walledai/walledeval/assets/32847115/d8b1d14f-7071-448b-8997-2eeba4c2c8f6"
-    logo = load_image_from_url(logo_url)
-    st.image(logo, use_column_width=True, width=500)  # Adjust the width as needed
+# Add logo at the bottom center (only once)
+if 'logo_displayed' not in st.session_state:
+    col1, col2, col3 = st.columns([1,2,1])
+    with col2:
+        logo_url = "https://github.com/walledai/walledeval/assets/32847115/d8b1d14f-7071-448b-8997-2eeba4c2c8f6"
+        logo = load_image_from_url(logo_url)
+        st.image(logo, use_column_width=True, width=500)  # Adjust the width as needed
+    st.session_state.logo_displayed = True
 
-# Add information about Walled Guard Advanced
-col1, col2, col3 = st.columns([1,2,1])
-with col2:
-    st.info("For a more performant version, check out Walled Guard Advanced. Connect with us at admin@walled.ai for more information.")
+# Add information about Walled Guard Advanced (only once)
+if 'info_displayed' not in st.session_state:
+    col1, col2, col3 = st.columns([1,2,1])
+    with col2:
+        st.info("For a more performant version, check out Walled Guard Advanced. Connect with us at admin@walled.ai for more information.")
+    st.session_state.info_displayed = True