Update app.py

app.py

@@ -126,10 +126,11 @@ if submit_button and user_input:
     st.session_state['generated'].append(output)
 
 # Display chat history within the container
-with chat_history_container:
-   for i in range(len(st.session_state['generated'])):
-        message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="big-smile")
-        message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
+if st.session_state['generated']:
+    with chat_history_container:
+       for i in range(len(st.session_state['generated'])):
+            message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="big-smile")
+            message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
 
 # Ensure chat history is always scrollable
 st.markdown("""