Update app.py

app.py

@@ -93,12 +93,8 @@ st.markdown('<h1 class="title">Tagba Translator</h1>', unsafe_allow_html=True)
 
 col1, col_swap, col2 = st.columns([1, 0.05, 1], gap="small")
 
-with col_swap:
-    st.write("")
-    swap = st.button("⇄")
 
-    if swap:
-        st.session_state.source_lang, st.session_state.target_lang = st.session_state.target_lang, st.session_state.source_lang
+
 
 if "translation" not in st.session_state:
     st.session_state.translation = "Traduction"
@@ -135,5 +131,16 @@ with col2:
         value=st.session_state.translation,
         height=200
     )
+
+with col_swap:
+    st.write("")
+    swap = st.button("⇄")
+
+    if swap:
+        st.session_state.source_lang, st.session_state.target_lang = st.session_state.target_lang, st.session_state.source_lang
+        temp = text_to_translate
+        text_to_translate = st.session_state.translation
+        st.session_state.translation = temp
+        
 st.markdown('</div>', unsafe_allow_html=True)