Update app.py

app.py

@@ -40,12 +40,13 @@ if prompt := st.chat_input("Type your message here..."):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # Prepare context as a plain string (no schema)
     context = ""
     for msg in st.session_state.messages:
         role = "User" if msg["role"] == "user" else "Assistant"
         context += f"{role}: {msg['content']}\n"
-        context += "Assistant: (Please answer in 50 words or fewer.)"
+
+    # Add explicit instruction to limit response
+    context += "Assistant: (Please answer in 50 words or fewer.)"
 
     # Generate response from LLM
     with st.chat_message("assistant"):