Update app.py

app.py

@@ -14,7 +14,7 @@ st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")
 st.title("🚗 Carfind.co.za AI Assistant")
 st.caption("Chat with Carfind.co.za and find your next car fast")
 
-# Custom CSS to fix the chat input at the bottom and make chat scrollable above it
+# Custom CSS for styling chat input and clear button inside it
 st.markdown(
     """
     <style>
@@ -36,9 +36,13 @@ st.markdown(
         padding: 10px;
         border-radius: 10px;
         box-shadow: 0 0 15px rgba(0,0,0,0.5);
+        display: flex;
+        align-items: center;
     }
-    .element-container:has(> .stChatMessage) {
-        margin-bottom: 20px;
+    .clear-chat-icon {
+        cursor: pointer;
+        margin-left: 10px;
+        font-size: 1.4em;
     }
     </style>
     """,
@@ -79,12 +83,28 @@ else:
                 st.markdown(message["content"], unsafe_allow_html=True)
         st.markdown('</div>', unsafe_allow_html=True)
 
-    # Input field fixed at the bottom
-    user_input = st.chat_input("Type your message here...")
-
-    if st.button("🗑️ Clear Chat", help="Clear chat history"):
+    # Custom chat input row with clear chat icon
+    with st.container():
+        user_input = st.chat_input("Type your message here...")
+        st.markdown(
+            """
+            <script>
+            function clearChat() {
+                fetch(window.location.href, {
+                    method: 'POST',
+                    headers: {'Content-Type': 'application/json'},
+                    body: JSON.stringify({"clear_chat": true})
+                }).then(() => location.reload());
+            }
+            </script>
+            <span class='clear-chat-icon' onclick='clearChat()'>🗑️</span>
+            """,
+            unsafe_allow_html=True
+        )
+
+    # Listen for clear chat POST
+    if st.experimental_get_query_params().get("clear_chat"):
         st.session_state.messages = []
-        st.rerun()
 
     if openai_key:
         client = OpenAI(api_key=openai_key)
@@ -140,4 +160,4 @@ else:
                 st.error(f"An error occurred: {str(e)}")
 
     else:
-        st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")
+        st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")