Update app.py

app.py

@@ -41,6 +41,12 @@ def view_episodic_memory(epi_mem):
     for item in epi_mem:
         st.write(f"**{item['event']}** ({item['sentiment']}) - {item['date']}")
 
+def add_fact(sem_mem, fact, category, source):
+    sem_mem.append({"fact": fact, "category": category, "source": source})
+
+def add_event(epi_mem, event, sentiment, date):
+    epi_mem.append({"event": event, "sentiment": sentiment, "date": date})
+
 def add_fact_to_semantic_memory(sem_mem):
     fact = st.text_input("Enter a fact")
     category = st.text_input("Enter a category")
@@ -78,9 +84,12 @@ def run_app():
     elif option == "Add Event to Episodic Memory":
         add_event_to_episodic_memory(epi_mem)
 
+    save_data(sem_mem, epi_mem)
+
 if __name__ == '__main__':
     run_app()
 
+
 # AW: Restructure the code listing into four functions.  shorten the code by eliminating comments and unnecessary whitespace and empty lines.
 # AI: This revised code splits the app into four functions: load_data, save_data, add_fact, and add_event. The run_app function handles the logic of the Streamlit app and calls these other functions as necessary. The code has been shortened by removing unnecessary whitespace and comments, but retains its functionality.