Update app.py

app.py

@@ -31,7 +31,8 @@ import tqdm
 # Streamlit app layout
 st.title("🦜🔗 Reminder AI")
 
-messages = st.container(height=300)
+
+
 
 tagging_prompt = ChatPromptTemplate.from_template(
     """
@@ -263,7 +264,7 @@ Current time: {current_time}
 Database sqlite:
 {context}
 
-If no date speicify assume is today, if user say one hour later or other, adding it into the current time to get the time (for example: Current time: 10pm, I said 1 hour later then 10pm + 1 hour is 11pm) 
+If no date speicify assume is today, if user say one hour later or other, adding it into the current time to get the time (for example: Current time: 10pm, I said 1 hour later then 10pm + 1 hour is 11pm)
 Don't say anything extra, if I am asking to remind something then only reply in this format:
 
 INSERT INTO reminders (date, time, reason)
@@ -284,7 +285,7 @@ prompt_tt = ChatPromptTemplate.from_messages([("human", message_thr)])
 rag_chain_tt = prompt_tt | llm_chat
 
 # Load the check_reminder data
-with open("important_classification_data.json", 'r') as f:
+with open("/content/drive/MyDrive/Colab_storage/important_classification_data.json", 'r') as f:
     important_classification_data = json.load(f)
 
 # Convert the intent classification data into Document format
@@ -299,12 +300,20 @@ vectorstore = Chroma.from_documents(documents=splits, embedding=HuggingFaceEmbed
 # Set up the retriever to fetch relevant phrases based on the user's query
 retriever = vectorstore.as_retriever()
 
-if prompt := st.chat_input("Say something"):
-    messages.chat_message("user").write(prompt)
-    ai_response = handle_user_input(prompt)
-    messages.chat_message("assistant").write(ai_response)
 
-def handle_user_input(user_input):
+# Initialize a list to store chat messages
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Text input for user message
+user_input = st.text_input("You:", key="input")
+
+
+# Loop for continuous interaction
+if user_input:
+    try:
+        # Take user input
+        st.session_state.messages.append(f"You: {user_input}")
         time_now = get_current_time()
         current_date = get_date()
 
@@ -320,7 +329,7 @@ def handle_user_input(user_input):
                 new_remind += new_reminder.content
             print(f"New reminder: {new_remind}")
             if new_remind == "repeated_reminder":
-                return("Reminder already existed!")
+                st.session_state.messages.append("Reminder already existed!")
             else:
                 try:
                     with sqlite3.connect('reminders.db') as conn:
@@ -332,12 +341,12 @@ def handle_user_input(user_input):
 
                     # Check if the update was successful
                     if cursor.rowcount > 0:
-                        return("New reminder created.")
+                        st.session_state.messages.append("New reminder created.")
                     else:
-                        return("No reminder created, Errors occur")
+                        st.session_state.messages.append("No reminder created, Errors occur")
 
                 except sqlite3.Error as e:
-                    return(f"An error occurred while updating the reminder: {e}")
+                    st.session_state.messages.append(f"An error occurred while updating the reminder: {e}")
         elif sentimenttocheck == "update reminder":
             reminders_sec = get_reminders()
             database_sec = format_reminders_for_context(reminders_sec)
@@ -346,7 +355,7 @@ def handle_user_input(user_input):
                 updated_remind += updated_cont.content
             print(f"Database: {updated_remind}")
             if updated_remind == "reminder_x":
-                return("No reminder found to change!")
+                st.session_state.messages.append("No reminder found to change!")
             else:
                 try:
                     with sqlite3.connect('reminders.db') as conn:
@@ -358,22 +367,31 @@ def handle_user_input(user_input):
 
                     # Check if the update was successful
                     if cursor.rowcount > 0:
-                        return("Reminder updated successfully.")
+                        st.session_state.messages.append("Reminder updated successfully.")
                     else:
-                        return("No reminder found to update with the given details.")
+                        st.session_state.messages.append("No reminder found to update with the given details.")
 
                 except sqlite3.Error as e:
-                    print(f"An error occurred while updating the reminder: {e}")
+                    st.session_state.messages.append(f"An error occurred while updating the reminder: {e}")
         elif sentimenttocheck == "check reminder":
             reminders = get_reminders()
             database = format_reminders_for_context(reminders)
             response_max = ""
             for max in rag_chain.stream({"question": user_input, "context": database, "date_tt": get_date()}):
                 response_max += max.content
-            return(f"Database remind: {response_max}")
+            st.session_state.messages.append(f"Database remind: {response_max}")
         elif sentimenttocheck == "remove reminder":
-          return("Remove Reminder")
+          st.session_state.messages.append("Remove Reminder")
         else:
-          ai_response, error_message = get_ai_response(user_input)
-          return(f"{ai_response}")
+          ai_response = get_ai_response(user_input)
+          st.session_state.messages.append(f"{ai_response}")
+
+    except Exception as e:
+        # Print the error and continue
+        st.session_state.messages.append(f"\nAn error occurred: {e}. Please try again.")
+
+    st.session_state.input = ""
 
+# Display the chat history
+for message in st.session_state.messages:
+    st.write(message)