space for iphone users

app.py

@@ -22,6 +22,8 @@ if BASE_URL is None:
 if DATABRICKS_API_TOKEN is None:
     raise ValueError("DATABRICKS_API_TOKEN environment variable must be set")
 
+st.set_page_config(layout="wide")
+
 # by default safety filter is not configured
 SAFETY_FILTER = False
 if SAFETY_FILTER_ENV is not None:
@@ -53,7 +55,6 @@ if TOKEN_CHUNK_SIZE_ENV is not None:
     TOKEN_CHUNK_SIZE = int(TOKEN_CHUNK_SIZE_ENV)
 
 MODEL_AVATAR_URL= "./icon.png"
-st.set_page_config(layout="wide")
 
 @st.cache_resource
 def get_global_semaphore():
@@ -219,23 +220,8 @@ def chat_completion(messages):
             max_token_warning = MSG_CLIPPED_AT_MAX_OUT_TOKENS
 
     yield {"content": partial_message, "error": None, "warning": max_token_warning}
-
 # if assistant is the last message, we need to prompt the user
 # if user is the last message, we need to retry the assistant.
-history = st.container(height=420)
-with history:
-    for message in st.session_state["messages"]:
-        avatar = None
-        if message["role"] == "assistant":
-            avatar = MODEL_AVATAR_URL
-        with st.chat_message(message["role"],avatar=avatar):
-            if message["content"] is not None:
-                st.markdown(message["content"])
-            if message["error"] is not None:
-                st.error(message["error"],icon="🚨")
-            if message["warning"] is not None:
-                st.warning(message["warning"],icon="⚠️")
-
 def handle_user_input(user_input):
     with history:
         response, stream_warning, stream_error = [None, None, None]
@@ -253,14 +239,29 @@ def handle_user_input(user_input):
         
         st.session_state["messages"].append({"role": "assistant", "content": response, "warning": stream_warning,"error": stream_error})
         print(st.session_state["messages"])
-
-
-if prompt := st.chat_input("Type a message!", max_chars=1000):
-    handle_user_input(prompt)
+    
+main = st.container()
+with main:
+    history = st.container(height=400)
+    with history:
+        for message in st.session_state["messages"]:
+            avatar = None
+            if message["role"] == "assistant":
+                avatar = MODEL_AVATAR_URL
+            with st.chat_message(message["role"],avatar=avatar):
+                if message["content"] is not None:
+                    st.markdown(message["content"])
+                if message["error"] is not None:
+                    st.error(message["error"],icon="🚨")
+                if message["warning"] is not None:
+                    st.warning(message["warning"],icon="⚠️")
+
+    if prompt := st.chat_input("Type a message!", max_chars=1000):
+        handle_user_input(prompt)
+    st.markdown("\n") #add some space for iphone users
 
 with st.sidebar:
     with st.container():
         st.title("Examples")
         for prompt in EXAMPLE_PROMPTS:
             st.button(prompt, args=(prompt,), on_click=handle_user_input)
-