add option to reset LLM analysis

MannLabs · Oct 23, 2024 · 2e04da8 · 2e04da8
1 parent e8ff6c2
commit 2e04da8
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/alphastats/gui/pages/05_LLM.py b/alphastats/gui/pages/05_LLM.py
@@ -169,10 +169,19 @@ def llm_config():
 
 st.markdown(f"##### LLM Analysis with {model_name}")
 
-llm_submitted = st.button(
+c1, c2, _ = st.columns((0.2, 0.2, 0.6))
+llm_submitted = c1.button(
     "Run LLM analysis ...", disabled=llm_integration_set_for_model
 )
 
+llm_reset = c2.button(
+    "Reset LLM analysis ...", disabled=not llm_integration_set_for_model
+)
+if llm_reset:
+    del st.session_state[StateKeys.LLM_INTEGRATION]
+    st.rerun()
+
+
 if st.session_state[StateKeys.LLM_INTEGRATION].get(model_name) is None:
     if not llm_submitted:
         st.stop()