try fragments

app.py

@@ -22,19 +22,15 @@ def load_model():
     return tokenizer, model
 
 # Function to load image from URL
-@st.cache_resource()
+@st.cache_data()
 def load_image_from_url(url):
     response = requests.get(url)
     img = Image.open(BytesIO(response.content))
     return img
 
-# Streamlit app
-st.title("Text Safety Evaluator")
-
-# User input
-user_input = st.text_area("Enter the text you want to evaluate:", height=100)
-
-if st.button("Evaluate"):
+# Evaluation fragment
+@st.experimental_fragment
+def evaluate_text(user_input, result_container):
     if user_input:
         # Load model and tokenizer
         tokenizer, model = load_model()
@@ -53,20 +49,45 @@ if st.button("Evaluate"):
         prediction = 'unsafe' if 'unsafe' in output_decoded.lower() else 'safe'
         
         # Display results
-        st.subheader("Evaluation Result:")
-        st.write(f"The text is evaluated as: **{prediction.upper()}**")
-        
+        with result_container:
+            st.subheader("Evaluation Result:")
+            st.write(f"The text is evaluated as: **{prediction.upper()}**")
     else:
-        st.warning("Please enter some text to evaluate.")
+        with result_container:
+            st.warning("Please enter some text to evaluate.")
 
-# Add logo at the bottom center
-col1, col2, col3 = st.columns([1,2,1])
-with col2:
+# Streamlit app
+st.title("Text Safety Evaluator")
+
+# User input
+user_input = st.text_area("Enter the text you want to evaluate:", height=100)
+
+# Create an empty container for the result
+result_container = st.empty()
+
+if st.button("Evaluate"):
+    evaluate_text(user_input, result_container)
+
+# Logo fragment
+@st.experimental_fragment
+def display_logo(logo_container):
     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
+    with logo_container:
+        st.image(logo, use_column_width=True, width=500)  # Adjust the width as needed
+
+# Info fragment
+@st.experimental_fragment
+def display_info(info_container):
+    with info_container:
+        st.info("For a more performant version, check out Walled Guard Advanced. Connect with us at [email protected] for more information.")
+
+# Add logo at the bottom center
+col1, col2, col3 = st.columns([1,2,1])
+logo_container = col2.empty()
+display_logo(logo_container)
 
 # 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 [email protected] for more information.")
+info_container = col2.empty()
+display_info(info_container)