Update app.py

app.py

@@ -81,7 +81,7 @@ if "metrics" not in st.session_state:
 if "query_dataset" not in st.session_state:
     st.session_state.query_dataset = ''
 if "rmse" not in st.session_state:
-    st.session_state.rmse = {}
+    st.session_state.rmse = 0.0
 
 recent_questions = load_recent_questions()
 
@@ -149,6 +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)
 
 # # Calculate Metrics Button
 # with col1:
@@ -172,16 +178,15 @@ with col1:
 with col2:
     #st.text_area("Metrics:", value=metrics, height=100, disabled=True)
     st.json(st.session_state.metrics)
-    
-# Calculate RMSE Button
+
+# Show RMSE Button
 with col1:
     if st.button("Show RMSE"):     
-        st.session_state.rmse = compute_rmse( st.session_state.metrics.values(),st.session_state.response)
-    else:
-        metrics_ = {}
+        if st.session_state.metrics:  # Check if metrics exist
+            st.session_state.rmse = compute_rmse(st.session_state.metrics.values(), st.session_state.response)
+        else:
+            st.session_state.rmse = "Metrics not available"
 
 with col2:
-    #st.text_area("RMSE:", value=rmse, height=100, disabled=True)
     st.json(st.session_state.rmse)
 
-