Update app.py

app.py

@@ -205,6 +205,9 @@ if "conversation_memory" not in st.session_state:
         memory_key="chat_history",
         return_messages=True
     )
+if "saved_conversations" not in st.session_state:
+    st.session_state.saved_conversations = []
+    
 start_time = time.time()
 if st.session_state["documents"] is None or st.session_state["vector_db"] is None:
     with st.spinner("Loading data..."):
@@ -217,27 +220,18 @@ else:
 
 retriever = vector_db.as_retriever()
 with st.sidebar:
-    st.title("Conversation History")
+    st.title("Past Conversations")
     
-    if st.button("New Session 🔄"):
-        # Clear all conversation related session states
-        st.session_state.messages = []
-        st.session_state.messages.append({
-            "role": "assistant",
-            "content": "👋 Hello! I'm Dr. Radha, your AI-powered Organic Farming Consultant. How can I assist you today?"
-        })
-        st.session_state.conversation_memory.clear()
-        st.session_state.session_id = str(uuid.uuid4())
-        st.rerun()
-        # Display conversation history
-    st.subheader("Previous Conversations")
-    for message in st.session_state.messages[1:]:  # Skip the initial greeting
-        if message["role"] == "user":
-            st.text("You:")
-            st.text_area("", message["content"], height=70, disabled=True, key=f"hist_{uuid.uuid4()}")
-        else:
-            st.text("Dr. Radha:")
-            st.text_area("", message["content"], height=100, disabled=True, key=f"hist_{uuid.uuid4()}")
+    # Display saved conversations
+    for idx, conv in enumerate(st.session_state.saved_conversations):
+        # Get the first message from user in the conversation
+        first_user_msg = next((msg["content"] for msg in conv if msg["role"] == "user"), "")
+        # Take first 30 characters of the message
+        preview = first_user_msg[:30] + "..." if len(first_user_msg) > 30 else first_user_msg
+        
+        if st.button(f"Conversation {idx + 1}: {preview}", key=f"conv_{idx}"):
+            st.session_state.messages = conv.copy()
+            st.rerun()
 
 prompt_template = """As an expert organic farming consultant with specialization in Agro-Homeopathy, analyze the following context and question to provide a clear, structured response.
 Context: {context}
@@ -348,8 +342,25 @@ with st.form(key='query_form', clear_on_submit=True):
         placeholder="Describe your plant issue here...",
         label_visibility="collapsed"
     )
-    submit_button = st.form_submit_button(label='Submit 📤')
-
+    col1, col2 = st.columns([1, 1])
+    with col1:
+        submit_button = st.form_submit_button(label='Submit 📤')
+    with col2:
+        new_conv_button = st.form_submit_button(label='New Conversation 🔄')
+
+if new_conv_button and len(st.session_state.messages) > 1:
+    # Save current conversation
+    st.session_state.saved_conversations.append(st.session_state.messages.copy())
+    
+    # Clear current conversation
+    st.session_state.messages = []
+    st.session_state.messages.append({
+        "role": "assistant",
+        "content": "👋 Hello! I'm Dr. Radha, your AI-powered Organic Farming Consultant. How can I assist you today?"
+    })
+    st.session_state.conversation_memory.clear()
+    st.rerun()
+    
 human_image = "human.png"
 robot_image = "bot.jpg"