Upload app.py

app.py

@@ -78,7 +78,7 @@ if st.session_state.recent_questions:
     st.sidebar.title("Overall RMSE")
     rmse_values = [q["metrics"]["rmse"] for q in recent_qns if "metrics" in q and "rmse" in q["metrics"]]
     average_rmse = sum(rmse_values) / len(rmse_values) if rmse_values else 0
-    st.sidebar.write(f"📊 **Average RMSE:** {average_rmse:.4f}")
+    st.sidebar.write(f"📊 **Average RMSE:** {average_rmse:.4f} for {len(rmse_values)} questions")
 
     st.sidebar.markdown("---")
 
@@ -144,8 +144,8 @@ st.text_area("Generated Response:", value=st.session_state.response, height=150,
 col1, col2 = st.columns([1, 3])  # Creating two columns for button and metrics display
 
 # Calculate Metrics Button
-with col1:
-    if st.button("Show Metrics"):     
+if st.button("Show Metrics"): 
+    with col1:
         st.session_state.metrics = calculate_metrics(question, st.session_state.query_dataset, st.session_state.response, st.session_state.retrieved_documents, st.session_state.time_taken_for_response)
         for q in st.session_state.recent_questions:
             if q["question"] == st.session_state.last_question:
@@ -153,13 +153,14 @@ with col1:
 
         # Save updated data to file
         save_recent_questions(st.session_state.recent_questions)
-
-with col2:
-    if st.session_state.metrics is not None:
-        metrics_ = st.session_state.metrics
-    else:
-        metrics_ ={}
+    with col2:
+        if st.session_state.metrics is not None:
+            metrics_ = st.session_state.metrics
+        else:
+            metrics_ ={}
         
-    st.text_area("Metrics:", value=metrics_, height=100, disabled=True)
+        st.text_area("Metrics:", value=metrics_, height=100, disabled=True)
+
+    st.rerun()
 
-st.experimental_rerun()
+