Update app.py

app.py

@@ -105,11 +105,11 @@ def chat_ui(tab_name, chat_history, system_prompt, input_key):
                         st.write(msg)
 
     # --- Input bar + send button in one row ---
-    user_input = st.text_input(
-        "Type your message...", key=input_key, label_visibility="collapsed"
-    )
+    col1, col2 = st.columns([10, 1])
+with col1:
+    user_input = st.text_input("Type your message...", key=input_key, label_visibility="collapsed")
+with col2:
     send_btn = st.button("➤", key=input_key + "_send")
-
     # --- Handle input ---
     if send_btn and user_input.strip():
         chat_history.append(("user", user_input.strip()))