Update app.py

app.py

@@ -149,12 +149,12 @@ st.subheader("Response")
 st.text_area("Generated Response:", value=st.session_state.response, height=150, disabled=True)
 
 col1, col2 = st.columns([1, 3])  # Creating two columns for button and metrics display
-col3, col4 =st.columns([3, 4]) #for RMSE button
+
 
 
 
 # Display RMSE
-st.json(st.session_state.rmse)
+#st.json(st.session_state.rmse)
 
 # # Calculate Metrics Button
 # with col1:
@@ -178,21 +178,16 @@ with col1:
 with col2:
     #st.text_area("Metrics:", value=metrics, height=100, disabled=True)
     st.json(st.session_state.metrics)
+
+col3, col4 =st.columns([3, 4]) #for RMSE button
 # Show RMSE Button
 with col3:
     if st.button("Show RMSE"):     
         if st.session_state.metrics:  # Check if metrics exist
             # Assuming compute_rmse returns a numeric value or dictionary
             st.session_state.rmse = compute_rmse(st.session_state.metrics.values(), st.session_state.response)
-            
-            # Ensure that rmse is valid before displaying
-            if isinstance(st.session_state.rmse, (dict, list, int, float)):
-                st.json({"rmse": st.session_state.rmse})  # Pass the RMSE as part of a valid JSON
-            else:
-                st.json({"error": "Invalid RMSE format"})  # Error in case RMSE is not valid
-        else:
-            st.session_state.rmse = "Metrics not available"
-            st.json({"error": "Metrics not available"})
+    else:
+            st.session_state.rmse = 0
 
 # Display RMSE (whether metrics were calculated or not)
 with col4: