Update app.py

app.py

@@ -38,28 +38,30 @@ if not st.session_state.authenticated:
 else:
     st.divider()
 
-    # Clear chat button positioned near chat window
-    if st.button("🧹 Clear Chat"):
-        st.session_state.messages = []
-        st.success("Chat history cleared.")
+    # Display chat history
+    if "messages" not in st.session_state:
+        st.session_state["messages"] = []
+
+    for message in st.session_state.messages:
+        role = message["role"]
+        content = message["content"]
+        st.chat_message(role).write(content)
+
+    # Add Clear Chat button next to input field
+    clear_col, input_col = st.columns([1, 8])
+    with clear_col:
+        if st.button("🧹 Clear Chat"):
+            st.session_state.messages = []
+            st.success("Chat history cleared.")
+
+    with input_col:
+        user_input = st.chat_input("Type your message here...")
 
     # Check for OpenAI key
-    if not openai_key:
-        st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")
-    else:
+    if openai_key:
         client = OpenAI(api_key=openai_key)
         ASSISTANT_ID = "asst_5gQR21fOsmHil11FGBzEArA7"
 
-        # Initialize message history
-        if "messages" not in st.session_state:
-            st.session_state["messages"] = []
-
-        # Display chat history
-        for message in st.session_state.messages:
-            role = message["role"]
-            content = message["content"]
-            st.chat_message(role).write(content)
-
         # Save chat transcript (backend only)
         def save_transcript(messages):
             with open("chat_logs.txt", "a") as log:
@@ -68,10 +70,10 @@ else:
                     log.write(f"{msg['role'].capitalize()}: {msg['content']}\n")
                 log.write("--- End Chat ---\n")
 
-        # Handle user input
-        if prompt := st.chat_input("Type your message here..."):
-            st.session_state.messages.append({"role": "user", "content": prompt})
-            st.chat_message("user").write(prompt)
+        # Handle user input and keep sticky experience
+        if user_input:
+            st.session_state.messages.append({"role": "user", "content": user_input})
+            st.chat_message("user").write(user_input)
 
             try:
                 # Start OpenAI chat thread
@@ -82,7 +84,7 @@ else:
                 client.beta.threads.messages.create(
                     thread_id=thread_id,
                     role="user",
-                    content=prompt
+                    content=user_input
                 )
 
                 # Trigger assistant response
@@ -114,4 +116,6 @@ else:
 
             except Exception as e:
                 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.")