Update app.py

app.py

@@ -199,7 +199,7 @@ class GEOSEOApp:
                     vectorstore = create_vectorstore_from_text(documents, self.embeddings)
                     st.session_state.rag_chain = create_rag_chain(self.llm, vectorstore)
 
-                    result = st.session_state.rag_chain.invoke({"question": user_query})
+                    result = st.session_state.rag_chain.invoke({"query": user_query})
 
                 
                 # Display results
@@ -306,7 +306,7 @@ class GEOSEOApp:
                     vectorstore = create_vectorstore_from_text([new_doc], self.embeddings)
                     st.session_state.rag_chain = create_rag_chain(self.llm, vectorstore)
 
-                    result = st.session_state.rag_chain.invoke({"question": user_query})
+                    result = st.session_state.rag_chain.invoke({"query": user_query})
                     st.success("Answer:")
                     st.write(result["result"])
 
@@ -401,7 +401,7 @@ class GEOSEOApp:
                 user_query = st.text_input("Ask here:", key="website_q")
 
                 if user_query:
-                     result = st.session_state.rag_chain.invoke({"question": user_query})
+                     result = st.session_state.rag_chain.invoke({"query": user_query})
                      st.success("Answer:")
                      st.write(result["result"])