Update app.py

app.py

@@ -40,19 +40,25 @@ llm = ChatGoogleGenerativeAI(
 
 
 # Response function
-def load_answer(question):
-    st.session_state.sessionMessages.append(HumanMessage(content=question))
-    assistant_response = llm.invoke(st.session_state.sessionMessages)
+# def load_answer(question):
+#     st.session_state.sessionMessages.append(HumanMessage(content=question))
+#     assistant_response = llm.invoke(st.session_state.sessionMessages)
     
-    # Assuming assistant_response is an object with a 'content' attribute
-    if hasattr(assistant_response, 'content') and isinstance(assistant_response.content, str):
-        processed_content = assistant_response.content
-        st.session_state.sessionMessages.append(AIMessage(content=processed_content))
-    else:
-        st.error("Invalid response received from AI.")
-        processed_content = "Sorry, I couldn't process your request."
-
-    return processed_content
+#     # Assuming assistant_response is an object with a 'content' attribute
+#     if hasattr(assistant_response, 'content') and isinstance(assistant_response.content, str):
+#         processed_content = assistant_response.content
+#         st.session_state.sessionMessages.append(AIMessage(content=processed_content))
+#     else:
+#         st.error("Invalid response received from AI.")
+#         processed_content = "Sorry, I couldn't process your request."
+
+#     return processed_content
+
+def load_answer(question):
+    st.session_state.sessionMessages.append(HumanMessage(content = question))
+    assistant_answer = llm.invoke(st.session_state.sessionMessages)
+    st.session_state.sessionMessages.append(AIMessage(content = assistant_answer))
+    return assistant_answer.content
 
 # User message
 def get_text():