Update app.py

app.py

@@ -43,6 +43,8 @@ with scol:
                                       index=options.index(st.session_state["sselected_language"]))
 with swapcol:
     if st.button("🔄 Swap"):
+        # Preserve current model selection
+        st.session_state["model_name"] = model_name
         # Swap languages in session state
         st.session_state["sselected_language"], st.session_state["tselected_language"] = \
             st.session_state["tselected_language"], st.session_state["sselected_language"]