fix: Move widget commands outside cached function to avoid unexpected behavior

alwrity.py

@@ -218,7 +218,19 @@ def main():
     setup_environment_paths()
     sidebar_configuration()
 
-    if check_api_keys():
+    missing_keys = check_api_keys()
+    if missing_keys:
+        st.warning(f"API keys not found: {', '.join(missing_keys)}. Please provide them below. Restart the app after saving the keys.")
+        with st.form(key='api_keys_form'):
+            for key in missing_keys:
+                st.text_input(f"{key}:", type="password", key=key)
+            if st.form_submit_button("Save Keys"):
+                with open(".env", "a") as env_file:
+                    for key in missing_keys:
+                        key_value = st.session_state[key]
+                        env_file.write(f"{key}={key_value}\n")
+                st.success("API keys saved successfully! Please restart the application.")
+    else:
         setup_tabs()
         modify_prompts_sidebar()