Update app.py

app.py

@@ -134,49 +134,49 @@ if 'DEEPSEEK_API_KEY' in st.session_state:
         for message in st.session_state.chat_history:
             with st.chat_message(message["role"]):
                 st.write(message["content"])
-        
-        # User input
-        user_input = st.chat_input("Type your message here...")
-        
-        if user_input:
-            # Add user message to chat history
-            st.session_state.chat_history.append({"role": "user", "content": user_input})
-            
-            # Display user message
-            with st.chat_message("user"):
-                st.write(user_input)
+
+# Move chat_input outside of any container
+user_input = st.chat_input("Type your message here...")
+
+if user_input:
+    # Add user message to chat history
+    st.session_state.chat_history.append({"role": "user", "content": user_input})
+    
+    # Display user message
+    with st.chat_message("user"):
+        st.write(user_input)
+    
+    # Get model response
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            try:
+                # Retrieve relevant context from vector database
+                context = retrieve_context(user_input, k=k_documents)
+                
+                # Prepare chat history for DeepSeek API
+                system_prompt = "You are a helpful assistant with access to a knowledge base."
+                if context:
+                    system_prompt += f"\n\nRelevant information from knowledge base:\n{context}\n\nUse this information to answer the user's question. If the information doesn't contain the answer, just say that you don't know based on the available information."
+                
+                messages = [{"role": "system", "content": system_prompt}]
+                for msg in st.session_state.chat_history:
+                    messages.append({"role": msg["role"], "content": msg["content"]})
+                
+                # Call DeepSeek API
+                response = st.session_state.client.chat.completions.create(
+                    model="deepseek-chat",
+                    messages=messages,
+                    stream=False
+                )
+                
+                assistant_response = response.choices[0].message.content
+                st.write(assistant_response)
+                
+                # Add assistant response to chat history
+                st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
             
-            # Get model response
-            with st.chat_message("assistant"):
-                with st.spinner("Thinking..."):
-                    try:
-                        # Retrieve relevant context from vector database
-                        context = retrieve_context(user_input, k=k_documents)
-                        
-                        # Prepare chat history for DeepSeek API
-                        system_prompt = "You are a helpful assistant with access to a knowledge base."
-                        if context:
-                            system_prompt += f"\n\nRelevant information from knowledge base:\n{context}\n\nUse this information to answer the user's question. If the information doesn't contain the answer, just say that you don't know based on the available information."
-                        
-                        messages = [{"role": "system", "content": system_prompt}]
-                        for msg in st.session_state.chat_history:
-                            messages.append({"role": msg["role"], "content": msg["content"]})
-                        
-                        # Call DeepSeek API
-                        response = st.session_state.client.chat.completions.create(
-                            model="deepseek-chat",
-                            messages=messages,
-                            stream=False
-                        )
-                        
-                        assistant_response = response.choices[0].message.content
-                        st.write(assistant_response)
-                        
-                        # Add assistant response to chat history
-                        st.session_state.chat_history.append({"role": "assistant", "content": assistant_response})
-                    
-                    except Exception as e:
-                        st.error(f"Error: {str(e)}")
+            except Exception as e:
+                st.error(f"Error: {str(e)}")
 
 # Sidebar with info
 with st.sidebar: