fix clear history issue

app.py

@@ -112,15 +112,10 @@ if "selected_model" not in st.session_state:
     st.session_state.selected_model = None
 
 
-# --- Clear history if model changes ---
+# --- Track selected model, but do not clear chat or image state on model change ---
 if st.session_state.selected_model != model_option:
-    st.session_state.messages = []
-    st.session_state.current_image_prompt_text = ""
-    # --- MODIFICATION START ---
-    # Clear the list of generated images when model changes
-    st.session_state.generated_images_list = []
-    # --- MODIFICATION END ---
     st.session_state.selected_model = model_option
+    # Optionally rerun to update UI, but do not clear messages or images
     st.rerun()
 
 # --- Define Main Columns ---