Update app.py

app.py

@@ -45,7 +45,8 @@ models = {
 }
 
 # Layout for model selection and max_tokens slider
-col1, col2 = st.columns(2)
+col1, col2 = st.columns([1, 3])  # Adjust the ratio to make the first column smaller
+
 
 with col1:
     model_option = st.selectbox(
@@ -54,26 +55,21 @@ with col1:
         format_func=lambda x: models[x]["name"],
         index=0,  # Default to the first model in the list
     )
-
-# Detect model change and clear chat history if model has changed
-if st.session_state.selected_model != model_option:
-    st.session_state.messages = []
-    st.session_state.selected_model = model_option
-
-max_tokens_range = models[model_option]["tokens"]
-
-with col2:
-    # Adjust max_tokens slider dynamically based on the selected model
+    max_tokens_range = models[model_option]["tokens"]
     max_tokens = st.slider(
         "Max Tokens:",
-        min_value=512,  # Minimum value to allow some flexibility
+        min_value=512,
         max_value=max_tokens_range,
-        # Default value or max allowed if less
         value=min(32768, max_tokens_range),
         step=512,
         help=f"Adjust the maximum number of tokens (words) for the model's response. Max for selected model: {max_tokens_range}",
     )
 
+# Detect model change and clear chat history if model has changed
+if st.session_state.selected_model != model_option:
+    st.session_state.messages = []
+    st.session_state.selected_model = model_option
+
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
     avatar = "🤖" if message["role"] == "assistant" else "🕺"