Update app.py

app.py

@@ -129,6 +129,11 @@ if view == "Artwork Explorer":
         st.markdown(f"> {generate_description(query_art)}")
 
     with st.expander("💬 Ask a question about this artwork"):
+        if st.button("🧹 Clear Conversation"):
+            st.session_state.chat_history = []
+        if "chat_history" not in st.session_state:
+            st.session_state.chat_history = []
+
         user_question = st.text_input("Your question")
         if user_question:
             qa_prompt = (
@@ -150,11 +155,18 @@ if view == "Artwork Explorer":
             qa_prompt += f"
 Question: {user_question}
 Answer:"
+
             try:
                 answer = describer(qa_prompt, max_length=100, truncation=True, do_sample=True)[0]["generated_text"]
-                st.markdown(f"**Answer:** {answer.strip()}")
+                st.session_state.chat_history.append((user_question, answer.strip()))
             except Exception as e:
-                st.error(f"[Answer unavailable: {e}]")
+                answer = f"[Answer unavailable: {e}]"
+                st.session_state.chat_history.append((user_question, answer))
+
+        if st.session_state.chat_history:
+            for question, response in reversed(st.session_state.chat_history[-5:]):
+                st.markdown(f"**You:** {question}")
+                st.markdown(f"> {response}")
 
     sim_scores = cosine_similarity([features[query_idx]], features)[0]
     ranked = np.argsort(sim_scores)[::-1][1:6]