Update app.py

app.py

@@ -34,6 +34,12 @@ if "sselected_language" not in st.session_state:
     st.session_state["sselected_language"] = options[0]
 if "tselected_language" not in st.session_state:
     st.session_state["tselected_language"] = options[1]
+if "model_name" not in st.session_state:
+    st.session_state["model_name"] = models[0]
+
+# Model selection FIRST
+model_name = st.selectbox("Select a model:", models, 
+                          index=models.index(st.session_state["model_name"]))
 
 # Create columns for language selection
 scol, swapcol, tcol = st.columns([3, 1, 3])
@@ -43,24 +49,19 @@ 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["model_name"] = model_name  # Preserve model
         st.session_state["sselected_language"], st.session_state["tselected_language"] = \
             st.session_state["tselected_language"], st.session_state["sselected_language"]
-        st.rerun()  # Rerun to update UI
+        st.rerun()
 with tcol:
     tselected_language = st.selectbox("Target language:", options, 
                                       index=options.index(st.session_state["tselected_language"]))
 
-# Model selection
-model_name = st.selectbox("Select a model:", models, index=0)
-
 # Language codes
 sl = langs[st.session_state["sselected_language"]]
 tl = langs[st.session_state["tselected_language"]]
 
-# Store selections in session state
+# Store selections
 st.session_state["model_name"] = model_name
 
 if model_name == 'Helsinki-NLP':