Update app.py

app.py

@@ -5,7 +5,7 @@ from typing import Generator
 from groq import Groq
 
 _ = load_dotenv(find_dotenv())
-st.set_page_config(page_icon="💬", layout="wide", page_title="Groq & LLaMA3...")
+st.set_page_config(page_icon="💬", layout="wide", page_title="Groq Chat Bot...")
 
 
 def icon(emoji: str):
@@ -18,31 +18,7 @@ def icon(emoji: str):
 
 icon("📣")
 
-# Header layout with title and model/token components on the same line
-header_col1, header_col2 = st.columns([2, 1])
-
-with header_col1:
-    st.subheader("Groq & LLaMA3 Chat")
-
-with header_col2:
-    model_option = st.selectbox(
-        "Choose a model:",
-        options=list(models.keys()),
-        format_func=lambda x: models[x]["name"],
-        index=0,  # Default to the first model in the list
-    )
-    max_tokens_range = models[model_option]["tokens"]
-    max_tokens = st.slider(
-        "Max Tokens:",
-        min_value=512,
-        max_value=max_tokens_range,
-        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}",
-    )
-
-# Place separator bar after header
-st.markdown("---")
+st.subheader("Groq Chat Streamlit App", divider="rainbow", anchor=False)
 
 client = Groq(
     api_key=os.environ['GROQ_API_KEY'],
@@ -68,6 +44,27 @@ models = {
     },
 }
 
+# Layout for model selection and max_tokens slider
+col1, col2 = st.columns([1, 3])  # Adjust the ratio to make the first column smaller
+
+
+with col1:
+    model_option = st.selectbox(
+        "Choose a model:",
+        options=list(models.keys()),
+        format_func=lambda x: models[x]["name"],
+        index=0,  # Default to the first model in the list
+    )
+    max_tokens_range = models[model_option]["tokens"]
+    max_tokens = st.slider(
+        "Max Tokens:",
+        min_value=512,
+        max_value=max_tokens_range,
+        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 = []