Update app.py

app.py

@@ -7,13 +7,13 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_community.chat_message_histories import SQLChatMessageHistory
 from langchain_core.runnables.history import RunnableWithMessageHistory
 
-# Load API key from secrets
+# Load API key
 GOOGLE_API_KEY = st.secrets.get("GOOGLE_API_KEY")
 
 # Set up the Gemini 1.5 Pro model
 llm = GoogleGenerativeAI(api_key=GOOGLE_API_KEY, model="gemini-1.5-pro")
 
-# Initialize SQLite database for chat history
+# Initialize SQLite database
 conn = sqlite3.connect("chat_history.db", check_same_thread=False)
 cursor = conn.cursor()
 cursor.execute("""
@@ -43,7 +43,7 @@ def chat_history(session_id):
         connection="sqlite:///chat_history.db"
     )
 
-# Generate unique session ID for each user
+# Generate unique session ID
 if "session_id" not in st.session_state:
     st.session_state.session_id = str(uuid.uuid4())
 
@@ -92,30 +92,34 @@ if "messages" not in st.session_state:
 
 chat_container = st.container()
 
-# Display chat history in a sticky format
+# Display chat history
 with chat_container:
     for role, content in st.session_state.messages:
         with st.chat_message(role):
             st.markdown(content)
 
-# User input stays at the bottom
+# User input at the bottom
 user_input = st.text_input("Type your message here:", key="user_message")
 
+# If user submits a message
 if user_input:
+    # Save user message
     save_message(session_id, "user", user_input)
     st.session_state.messages.append(("user", user_input))
     
-    # Invoke the AI model
+    # Invoke AI model
     config = {'configurable': {'session_id': session_id}}
     response = chat.invoke({'prompt': user_input}, config)
     
+    # Save AI response
     save_message(session_id, "assistant", response)
     st.session_state.messages.append(("assistant", response))
     
-    # Refresh chat UI
+    # Display AI response
     with chat_container:
         with st.chat_message("assistant"):
             st.markdown(response)
 
-    # Clear input field after message submission
+    #  Clear the input field after message submission
     st.session_state.pop("user_message")
+    st.rerun()  # Refresh the app