Update app.py

app.py

@@ -33,20 +33,14 @@ if "messages" not in st.session_state:
 
 st.title("Mistral-7B Chatbot")
 
-# Function to get streamed response
-def get_streamed_response(message, history):
-    all_messages = []
-    # Include system prompt
-    all_messages.append({
-        "role": "system",
-        "content": "You are a knowledgeable and friendly assistant. Respond concisely and maintain a friendly tone."
-    })
-    # Add previous messages to history
-    for msg in history:
-        all_messages.append(msg)
-    # Add the latest user message
-    all_messages.append({"role": "user", "content": message})
-    return all_messages
+# Function to prepare messages for the API
+def get_messages_for_api():
+    messages = []
+    if len(st.session_state.messages) % 2 == 1:  # If the last message is from the user
+        messages.append({"role": "assistant", "content": "Your assistant is ready."})  # Ensure the assistant starts or resumes
+    for message in st.session_state.messages:
+        messages.append(message)
+    return messages
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
@@ -61,13 +55,12 @@ if prompt := st.chat_input("Type your message here..."):
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    # Prepare all messages for the conversation context
-    history = st.session_state.messages
+    # Prepare messages for the conversation context
+    all_messages = get_messages_for_api()
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
         try:
-            all_messages = get_streamed_response(prompt, history)
             response = client.chat.completions.create(
                 model=model_link,
                 messages=all_messages,