app.py

import streamlit as st
import pandas as pd

# Define functions
def create_empty_csv_files():
    sem_df = pd.DataFrame(columns=["fact", "category", "source"])
    sem_df.to_csv("semantic_memory.csv", index=False)
    epi_df = pd.DataFrame(columns=["event", "sentiment", "date"])
    epi_df.to_csv("episodic_memory.csv", index=False)

def load_data():
    try:
        sem_df = pd.read_csv("semantic_memory.csv")
        sem_mem = sem_df.to_dict("records")
    except:
        create_empty_csv_files()
        sem_mem = [{"fact": "The Earth is round", "category": "science", "source": "NASA"},
                   {"fact": "Pizza is delicious", "category": "food", "source": "me"}]
    try:
        epi_df = pd.read_csv("episodic_memory.csv")
        epi_mem = epi_df.to_dict("records")
    except:
        create_empty_csv_files()
        epi_mem = [{"event": "I went to the beach", "sentiment": "happy", "date": "2022-02-28"},
                   {"event": "I had a fight with my friend", "sentiment": "sad", "date": "2022-02-25"}]
    return sem_mem, epi_mem

def save_data(sem_mem, epi_mem):
    sem_df = pd.DataFrame(sem_mem)
    sem_df.to_csv("semantic_memory.csv", index=False)
    epi_df = pd.DataFrame(epi_mem)
    epi_df.to_csv("episodic_memory.csv", index=False)

def view_semantic_memory(sem_mem):
    st.write("# Semantic Memory")
    for item in sem_mem:
        st.write(f"**{item['fact']}** ({item['category']}) - {item['source']}")

def view_episodic_memory(epi_mem):
    st.write("# Episodic Memory")
    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, epi_mem):
    fact = st.text_input("Enter a fact")
    category = st.text_input("Enter a category")
    source = st.text_input("Enter a source")
    if st.button("Add Fact"):
        add_fact(sem_mem, fact, category, source)
        save_data(sem_mem, epi_mem)
        st.success("Fact added to semantic memory!")
        st.sidebar.success("Fact added to semantic memory!")

def add_event_to_episodic_memory(epi_mem, sem_mem):
    event = st.text_input("Enter an event")
    sentiment = st.selectbox("Select a sentiment", ["happy", "sad", "neutral"])
    date = st.date_input("Select a date")
    if st.button("Add Event"):
        add_event(epi_mem, event, sentiment, date)
        save_data(sem_mem, epi_mem)
        st.success("Event added to episodic memory!")
        st.sidebar.success("Event added to episodic memory!")

def run_app():
    sem_mem, epi_mem = load_data()

    st.title("Cognitive Agent")
    option = st.sidebar.selectbox("Select an option",
                                  ["View Semantic Memory", "View Episodic Memory", "Add Fact to Semantic Memory",
                                   "Add Event to Episodic Memory"])

    if option == "View Semantic Memory":
        view_semantic_memory(sem_mem)
    elif option == "View Episodic Memory":
        view_episodic_memory(epi_mem)
    elif option == "Add Fact to Semantic Memory":
        add_fact_to_semantic_memory(sem_mem, epi_mem)
    elif option == "Add Event to Episodic Memory":
        add_event_to_episodic_memory(epi_mem, sem_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.


# This program uses Streamlit to create a web app that allows the user to view and add to both semantic and episodic memory. The semantic memory is stored as a list of dictionaries, where each dictionary represents a fact and includes the fact itself, the category it belongs to, and the source of the fact. The episodic memory is also stored as a list of dictionaries, where each dictionary represents an event and includes the event itself, the sentiment associated with the event, and the date the event occurred.
# The program allows the user to view both types of memory by selecting an option from the sidebar. If the user selects "View Semantic Memory", the program displays all of the facts stored in semantic memory. If the user selects "View Episodic Memory", the program displays all of the events stored in episodic memory.
# The program also allows the user to add new facts to semantic memory or new events to episodic memory by selecting an option from the sidebar and filling out a form with the relevant information. When the user clicks the "Add Fact" or "Add Event" button, the new fact or event is added to the appropriate list of dictionaries and saved to a CSV file. The program then displays a success message indicating that the fact or event was added to memory.
# Overall, this program demonstrates how semantic and episodic memory can be modeled using Python list dictionaries, and how these types of memory can be used to track both facts and observations, as well as sentiments associated with past experiences.