app improvement again and again

app.py

@@ -14,6 +14,11 @@ st.set_page_config(
     layout="wide"
 )
 
+# Initialize session state
+if 'history' not in st.session_state:
+    st.session_state.history = []
+if 'api_health' not in st.session_state:
+    st.session_state.api_health = None
 
 # Custom CSS
 st.markdown("""
@@ -99,7 +104,7 @@ with st.expander("📝 Example Inputs"):
     if st.button("Use Example 1"):
         st.session_state.user_input = "Have a great day!"
     if st.button("Use Example 2"):
-        st.session_state.user_input = "I disagree with your opinion."
+        st.session_state.user_input = "You are the worst person in the world."
     if st.button("Use Example 3"):
         st.session_state.user_input = "You're amazing!"
 
@@ -148,26 +153,31 @@ if user_input:
                 st.error(f"Error: {error}")
             else:
                 # Format probability as percentage
-                probability_pct = result['probability']
+                probability = result['probability']
                 
                 # Create prediction box with appropriate severity class
                 severity_class = get_severity_class(result['probability'])
                 
+                if result['prediction'] == 'offensive':
+                    final_prediction = "Offensive"
+                else : 
+                    final_prediction = "Not Offensive"
+
                 st.markdown(f"""
                 <div class="prediction-box {severity_class}">
                     <h3>Analysis Results</h3>
-                    <p><strong>Prediction:</strong> {result['prediction']}</p>
-                    <p><strong>Confidence:</strong> {probability_pct:.1f}%</p>
+                    <p><strong>Prediction:</strong> {final_prediction}</p>
+                    <p><strong>Prediction Value:</strong> {probability:.2f}%</p>
                 </div>
                 """, unsafe_allow_html=True)
                 
                 # Confidence meter using Plotly
                 fig = go.Figure(go.Indicator(
                     mode = "gauge+number",
-                    value = probability_pct,
+                    value = probability,
                     title = {'text': "Confidence Level"},
                     gauge = {
-                        'axis': {'range': [0, 100]},
+                        'axis': {'range': [0, 1]},
                         'bar': {'color': "darkblue"},
                         'steps': [
                             {'range': [0, 0.3], 'color': "lightgreen"},
@@ -183,8 +193,8 @@ if user_input:
                 st.session_state.history.append({
                     'timestamp': datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                     'text': user_input,
-                    'prediction': result['prediction'],
-                    'confidence': probability_pct
+                    'prediction': final_prediction,
+                    'confidence': probability
                 })
 
 # History Section