Update app.py

app.py

@@ -101,39 +101,39 @@ if "OPENAI_API_KEY" in os.environ:
             return retriever, document_chain
 
         # Load components
-        with st.spinner("Initializing Assistant..."):
-            retriever, document_chain = initialize_components()
+    with st.spinner("Initializing Assistant..."):
+        retriever, document_chain = initialize_components()
 
         # Initialize memory for each session
-        if "memory" not in st.session_state:
-            st.session_state.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
+    if "memory" not in st.session_state:
+        st.session_state.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
         # Chat interface
-        st.subheader("Chat AI Assistant")
+    st.subheader("Chat AI Assistant")
 
         # Initialize chat history
-        if "messages" not in st.session_state:
-            st.session_state.messages = []
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
 
         # Display chat messages from history on app rerun
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
+    for message in st.session_state.messages:
+         with st.chat_message(message["role"]):
+            st.markdown(message["content"])
 
         # React to user input
-        if prompt := st.chat_input("What would you like to know about Document?"):
-            # Display user message in chat message container
+    if prompt := st.chat_input("What would you like to know about Document?"):
+                # Display user message in chat message container
             st.chat_message("user").markdown(prompt)
-            # Add user message to chat history
+                # Add user message to chat history
             st.session_state.messages.append({"role": "user", "content": prompt})
-
+    
             with st.chat_message("assistant"):
                 message_placeholder = st.empty()
-
-                # Retrieve relevant documents
+    
+                    # Retrieve relevant documents
                 docs = retriever.get_relevant_documents(prompt)
-
-                # Generate response
+    
+                    # Generate response
                 response = document_chain.invoke(
                     {
                         "context": docs,
@@ -142,17 +142,17 @@ if "OPENAI_API_KEY" in os.environ:
                             HumanMessage(content=prompt)
                         ],
                     }
-                )
-
-                # The response is already a string, so we can use it directly
-                full_response = response
-                message_placeholder.markdown(full_response)
-
-            # Add assistant response to chat history
-            st.session_state.messages.append({"role": "assistant", "content": full_response})
-
-            # Update memory
-            st.session_state.memory.save_context({"input": prompt}, {"output": full_response})
+                    )
+    
+                    # The response is already a string, so we can use it directly
+            full_response = response
+            message_placeholder.markdown(full_response)
+    
+                # Add assistant response to chat history
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
+    
+                # Update memory
+        st.session_state.memory.save_context({"input": prompt}, {"output": full_response})
 
 else:
     st.warning("Please enter your OpenAI API Key in the sidebar to start the chatbot.")