Update app.py

app.py

@@ -128,9 +128,10 @@ def main():
 
 def health_check():
     """Simple health check endpoint that returns JSON"""
-    if 'health' in st.experimental_get_query_params():
-        st.json({"status": "OK"})
-        st.stop()
+    params = st.experimental_get_query_params()
+    if 'health' in params and params['health'][0] == 'true':
+        st.write('{"status": "OK"}')
+        st.cache_data.clear()  # Clear cache to ensure fresh state
         return True
     return False