Update app.py

app.py

@@ -168,7 +168,11 @@ if st.session_state.query_submitted:
         if follow_up_query:
             with st.spinner("Generating response..."):
                 try:
-                    follow_up_response = generate_detailed_explanation(follow_up_query, "")  # Passing an empty context
+                    # Get the last 5 messages from the chat history
+                    last_5_messages = st.session_state.chat_history[-5:]
+                    context = "\n".join([f"Q: {q}\nA: {a}" for q, a in last_5_messages])
+                    
+                    follow_up_response = generate_detailed_explanation(follow_up_query, context)
                     st.session_state.chat_history.append((follow_up_query, follow_up_response))
                     st.rerun()  # Rerun to display the new response
                 except Exception as e: