first commit

.gitignore

@@ -0,0 +1,160 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/

app.py

@@ -0,0 +1,366 @@
+import streamlit as st
+import plotly.express as px
+import pandas as pd
+import streamlit_authenticator as stauth
+import yaml
+from yaml.loader import SafeLoader
+
+# from langchain.chat_models import ChatAnthropic
+# from langchain.callbacks.base import BaseCallbackHandler
+# from langchain.prompts import PromptTemplate
+# from langchain.chains import LLMChain
+
+
+# Function to fetch simulated fly situation data
+def get_fly_situation(canteen):
+    if canteen == "Deck":
+        # Sample fly situation data
+        fly_situation = {
+            "temperature": 28,
+            "humidity": 60,
+            "fly_count": 9,
+            "last_updated": "2023-11-10 12:00:00"
+        }
+        delta1 = '0.2'
+        delta2 = '2'
+        delta3 = '1'
+    elif canteen == "Frontier":
+        # Sample fly situation data
+        fly_situation = {
+            "temperature": 28.1,
+            "humidity": 62,
+            "fly_count": 21,
+            "last_updated": "2023-11-10 12:00:00"
+        }
+        delta1 = '0.1' 
+        delta2 = '1' 
+        delta3 = '3' 
+    return fly_situation, delta1, delta2, delta3
+
+# Function to generate a sample fly situation dataset with time series
+def get_fly_situation_history(canteen):
+    if canteen == "Deck":
+        # Sample fly situation time series data
+        fly_situation_history = [
+            {"timestamp": "2023-11-10 11:00:00", "fly_count": 2, "sensor":1},
+            {"timestamp": "2023-11-10 11:05:00", "fly_count": 1, "sensor": 1},
+            {"timestamp": "2023-11-10 11:10:00", "fly_count": 2, "sensor": 1},
+            {"timestamp": "2023-11-10 11:15:00", "fly_count": 2, "sensor": 1},
+            {"timestamp": "2023-11-10 11:20:00", "fly_count": 3, "sensor": 1},
+            {"timestamp": "2023-11-10 11:25:00", "fly_count": 1, "sensor": 1},
+            {"timestamp": "2023-11-10 11:30:00", "fly_count": 2, "sensor": 1},
+            {"timestamp": "2023-11-10 11:35:00", "fly_count": 1, "sensor": 1},
+            {"timestamp": "2023-11-10 11:40:00", "fly_count": 3, "sensor": 1},
+            {"timestamp": "2023-11-10 11:45:00", "fly_count": 1, "sensor": 1},
+            {"timestamp": "2023-11-10 11:50:00", "fly_count": 2, "sensor": 1},
+            {"timestamp": "2023-11-10 11:55:00", "fly_count": 3, "sensor": 1},
+            {"timestamp": "2023-11-10 12:00:00", "fly_count": 1, "sensor": 1},
+            {"timestamp": "2023-11-10 11:00:00", "fly_count": 1, "sensor": 2},
+            {"timestamp": "2023-11-10 11:05:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 11:10:00", "fly_count": 3, "sensor": 2},
+            {"timestamp": "2023-11-10 11:15:00", "fly_count": 1, "sensor": 2},
+            {"timestamp": "2023-11-10 11:20:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 11:25:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 11:30:00", "fly_count": 1, "sensor": 2},
+            {"timestamp": "2023-11-10 11:35:00", "fly_count": 3, "sensor": 2},
+            {"timestamp": "2023-11-10 11:40:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 11:45:00", "fly_count": 1, "sensor": 2},
+            {"timestamp": "2023-11-10 11:50:00", "fly_count": 3, "sensor": 2},
+            {"timestamp": "2023-11-10 11:55:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 12:00:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 11:00:00", "fly_count": 3, "sensor": 3},
+            {"timestamp": "2023-11-10 11:05:00", "fly_count": 1, "sensor": 3},
+            {"timestamp": "2023-11-10 11:10:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:15:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:20:00", "fly_count": 1, "sensor": 3},
+            {"timestamp": "2023-11-10 11:25:00", "fly_count": 3, "sensor": 3},
+            {"timestamp": "2023-11-10 11:30:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:35:00", "fly_count": 1, "sensor": 3},
+            {"timestamp": "2023-11-10 11:40:00", "fly_count": 1, "sensor": 3},
+            {"timestamp": "2023-11-10 11:45:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:50:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:55:00", "fly_count": 3, "sensor": 3},
+            {"timestamp": "2023-11-10 12:00:00", "fly_count": 6, "sensor": 3},
+        ]
+    elif canteen == "Frontier":
+        # Sample fly situation time series data
+        fly_situation_history = [
+            {"timestamp": "2023-11-10 11:00:00", "fly_count": 2, "sensor":1},
+            {"timestamp": "2023-11-10 11:05:00", "fly_count": 5, "sensor": 1},
+            {"timestamp": "2023-11-10 11:10:00", "fly_count": 6, "sensor": 1},
+            {"timestamp": "2023-11-10 11:15:00", "fly_count": 4, "sensor": 1},
+            {"timestamp": "2023-11-10 11:20:00", "fly_count": 5, "sensor": 1},
+            {"timestamp": "2023-11-10 11:25:00", "fly_count": 2, "sensor": 1},
+            {"timestamp": "2023-11-10 11:30:00", "fly_count": 5, "sensor": 1},
+            {"timestamp": "2023-11-10 11:35:00", "fly_count": 6, "sensor": 1},
+            {"timestamp": "2023-11-10 11:40:00", "fly_count": 7, "sensor": 1},
+            {"timestamp": "2023-11-10 11:45:00", "fly_count": 8, "sensor": 1},
+            {"timestamp": "2023-11-10 11:50:00", "fly_count": 10, "sensor": 1},
+            {"timestamp": "2023-11-10 11:55:00", "fly_count": 9, "sensor": 1},
+            {"timestamp": "2023-11-10 12:00:00", "fly_count": 8, "sensor": 1},
+            {"timestamp": "2023-11-10 11:00:00", "fly_count": 1, "sensor": 2},
+            {"timestamp": "2023-11-10 11:05:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 11:10:00", "fly_count": 3, "sensor": 2},
+            {"timestamp": "2023-11-10 11:15:00", "fly_count": 2, "sensor": 2},
+            {"timestamp": "2023-11-10 11:20:00", "fly_count": 3, "sensor": 2},
+            {"timestamp": "2023-11-10 11:25:00", "fly_count": 4, "sensor": 2},
+            {"timestamp": "2023-11-10 11:30:00", "fly_count": 6, "sensor": 2},
+            {"timestamp": "2023-11-10 11:35:00", "fly_count": 7, "sensor": 2},
+            {"timestamp": "2023-11-10 11:40:00", "fly_count": 8, "sensor": 2},
+            {"timestamp": "2023-11-10 11:45:00", "fly_count": 10, "sensor": 2},
+            {"timestamp": "2023-11-10 11:50:00", "fly_count": 9, "sensor": 2},
+            {"timestamp": "2023-11-10 11:55:00", "fly_count": 8, "sensor": 2},
+            {"timestamp": "2023-11-10 12:00:00", "fly_count": 6, "sensor": 2},
+            {"timestamp": "2023-11-10 11:00:00", "fly_count": 3, "sensor": 3},
+            {"timestamp": "2023-11-10 11:05:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:10:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:15:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:20:00", "fly_count": 1, "sensor": 3},
+            {"timestamp": "2023-11-10 11:25:00", "fly_count": 3, "sensor": 3},
+            {"timestamp": "2023-11-10 11:30:00", "fly_count": 5, "sensor": 3},
+            {"timestamp": "2023-11-10 11:35:00", "fly_count": 7, "sensor": 3},
+            {"timestamp": "2023-11-10 11:40:00", "fly_count": 6, "sensor": 3},
+            {"timestamp": "2023-11-10 11:45:00", "fly_count": 3, "sensor": 3},
+            {"timestamp": "2023-11-10 11:50:00", "fly_count": 2, "sensor": 3},
+            {"timestamp": "2023-11-10 11:55:00", "fly_count": 1, "sensor": 3},
+            {"timestamp": "2023-11-10 12:00:00", "fly_count": 7, "sensor": 3},
+        ]
+
+    return fly_situation_history
+
+
+# # Streaming LLM output class
+# class StreamHandler(BaseCallbackHandler):
+#     # Referenced from: https://discuss.streamlit.io/t/langchain-stream/43782
+#     def __init__(self, container, initial_text='', display_method='markdown'):
+#         self.container = container
+#         self.text = initial_text
+#         self.display_method = display_method
+
+#     def on_llm_new_token(self, token: str, **kwargs) -> None:
+#         self.text += token
+#         display_function = getattr(self.container, self.display_method, None)
+#         if display_function is not None:
+#             display_function(self.text)
+#         else:
+#             raise ValueError(f'Invalid display_method: {self.display_method}')
+
+
+st.set_page_config(layout="centered")
+
+# Import configuration file for user authentication
+with open('credentials.yaml') as file:
+    config = yaml.load(file, Loader=SafeLoader)
+    
+# Create an authentication object.
+authenticator = stauth.Authenticate(
+    config['credentials'],
+    config['cookie']['name'],
+    config['cookie']['key'],
+    config['cookie']['expiry_days']
+    )
+# List of advanced users
+advanced_users = ['advanced']
+
+# Landing page if user not logged in
+if st.session_state['authentication_status'] is None:
+    # Landing page copy and banner
+    st.markdown('<h1 style="text-align: left;">Fly Situation Monitoring App</h1>', unsafe_allow_html=True)
+    st.markdown('<h4 style="text-align: left;">Keeping You Informed, Keeping Flies at Bay</h4>', unsafe_allow_html=True)
+    st.write('\n')
+    st.write('\n')
+    
+    # Loging log-in details
+    name, authentication_status, username = authenticator.login('', 'main')
+    
+    # If log-in failed
+    if st.session_state['authentication_status'] is False:
+        st.error('Username/password is incorrect.')
+    st.write('\n')
+    st.write('\n')
+
+    
+# App if user is logged in and authenticated
+if st.session_state['authentication_status']:
+    
+    # Streamlit app start
+    st.title("Fly Situation Monitoring App")
+    st.markdown("Keeping You Informed, Keeping Flies at Bay")
+    st.write('\n')
+    
+    # User selects a canteen
+    canteen = st.selectbox("Select a Canteen:", options=["Deck", "Frontier"])
+    st.write('\n')
+    
+    # If user is a student, show basic app layout
+    if not st.session_state['username'] in advanced_users:
+        
+        # Tabs
+        tab1, tab2 = st.tabs(["Current", "History"])
+        
+        # Tab 1: Fly Situation
+        with tab1:
+            st.header("Current Fly Situation")
+            
+            # Get data
+            fly_situation, delta1, delta2, delta3 = get_fly_situation(canteen)
+            
+            # Display key information using cards
+            col1_fly_curr, col2_fly_curr, col3_fly_curr = st.columns(3)
+            col1_fly_curr.metric("Temperature", str(fly_situation["temperature"]) + " °C", delta=delta1)
+            col2_fly_curr.metric("Humidty", str(fly_situation["humidity"]) + " %", delta=delta2)
+            col3_fly_curr.metric("Fly Count", str(fly_situation["fly_count"]), delta=delta3, delta_color="inverse")
+            st.caption("Last updated at " + fly_situation["last_updated"] + " (5 min intervals)")
+            
+            # Alert level
+            if fly_situation["fly_count"] > 20:
+                alert_level = "High"
+                alert_colour = "red"
+            elif fly_situation["fly_count"] > 10:
+                alert_level = "Moderate"
+                alert_colour = "orange"
+            else: 
+                alert_level = "Low"
+                alert_colour = "green"
+            st.markdown(f"<h2 style='color:{alert_colour}; text-align: left'>Alert Level: {alert_level}</h3>", unsafe_allow_html=True)
+            st.markdown('---')
+            
+            # Feedback
+            st.header("Feedback")
+            # Gather feedback
+            user_feedback = st.text_area("Provide Feedback on the Fly Situation:")
+            if st.button("Submit Feedback"):
+                st.success("Feedback submitted successfully!")
+            st.write('\n')
+            st.write('\n')
+            st.write('\n')
+                
+        # Tab 2: History
+        with tab2:
+            st.header("Fly Count Over Time")
+            
+            # Get history data
+            fly_situation_history = get_fly_situation_history(canteen)
+            
+            # Create a DataFrame for the time series data
+            df = pd.DataFrame(fly_situation_history)
+            sum_by_timestamp = df.groupby('timestamp')['fly_count'].sum().reset_index()
+            sum_by_timestamp["timestamp"] = pd.to_datetime(sum_by_timestamp["timestamp"])
+            
+            # Plot the time series using Plotly Express
+            fig = px.line(sum_by_timestamp, x="timestamp", y="fly_count", title="Fly Count Over Time", labels={"fly_count": "Fly Count", "timestamp": "Timestamp"})
+            st.plotly_chart(fig)
+
+            # Question-and-Answer
+            with st.form("form"):
+                prompt = st.text_input("Ask a Question:")
+                submit = st.form_submit_button("Submit")
+                if prompt:
+                    with st.spinner("Generating..."):
+                        pass
+            
+        # Logout
+        logout_col1, logout_col2 = st.columns([6,1])
+        with logout_col2:    
+            st.write('\n')
+            st.write('\n')
+            st.write('\n')
+            authenticator.logout('Logout', 'main')
+    
+    else:
+        # Tabs
+        tab1, tab2, tab3 = st.tabs(["Current", "History", "Control System"])
+        
+        # Tab 1: Fly Situation
+        with tab1:
+            st.header("Current Fly Situation")
+            # Get current data
+            fly_situation, delta1, delta2, delta3 = get_fly_situation(canteen)
+            # Display key information using cards
+            col1_fly_curr, col2_fly_curr, col3_fly_curr = st.columns(3)
+            col1_fly_curr.metric("Temperature", str(fly_situation["temperature"]) + " °C", delta=delta1)
+            col2_fly_curr.metric("Humidty", str(fly_situation["humidity"]) + " %", delta=delta2)
+            col3_fly_curr.metric("Fly Count", str(fly_situation["fly_count"]), delta=delta3, delta_color="inverse")
+            st.caption("Last updated at " + fly_situation["last_updated"] + " (5 min intervals)")
+            
+            # Alert 
+            if fly_situation["fly_count"] > 20:
+                alert_level = "High"
+                alert_colour = "red"
+            elif fly_situation["fly_count"] > 10:
+                alert_level = "Moderate"
+                alert_colour = "orange"
+            else: 
+                alert_level = "Low"
+                alert_colour = "green"
+            st.markdown(f"<h2 style='color:{alert_colour}; text-align: left'>Alert Level: {alert_level}</h3>", unsafe_allow_html=True)
+            st.markdown('---')
+            
+            # Feedback
+            st.header("Feedback")
+            # Gather feedback
+            user_feedback = st.text_area("Provide Feedback on the Fly Situation:")
+            if st.button("Submit Feedback"):
+                st.success("Feedback submitted successfully!")
+            st.write('\n')
+            st.write('\n')
+            st.write('\n')
+                
+
+        # Tab 2: History
+        with tab2:
+            st.header("Fly Count Over Time")
+            # Select sensor
+            selected_sensor = st.selectbox("Select Sensor:", ["All", "Sensor 1", "Sensor 2", "Sensor 3"])
+            # Get history data
+            fly_situation_history = get_fly_situation_history(canteen)
+            # Create a DataFrame for the time series data
+            df = pd.DataFrame(fly_situation_history)
+            if selected_sensor != "All":
+                df = df[df["sensor"]==int(selected_sensor[-1])]
+            sum_by_timestamp = df.groupby('timestamp')['fly_count'].sum().reset_index()
+            sum_by_timestamp["timestamp"] = pd.to_datetime(sum_by_timestamp["timestamp"])
+            # Plot the time series using Plotly Express
+            fig = px.line(sum_by_timestamp, x="timestamp", y="fly_count", title="Fly Count Over Time", labels={"fly_count": "Fly Count", "timestamp": "Timestamp"})
+            st.plotly_chart(fig)
+
+            # Question-and-Answer
+            with st.form("form"):
+                prompt = st.text_input("Ask a Question:")
+                submit = st.form_submit_button("Submit")
+                if prompt:
+                    with st.spinner("Generating..."):
+                        pass
+        
+        # Tab 3: Control System
+        with tab3:
+            st.write('\n')
+            
+            # Enable/disable automatic pest control system
+            automatic_control_enabled = st.checkbox("Enable Automatic Pest Control", value=True)
+            st.write('\n')
+            if not automatic_control_enabled:
+                disabled = False
+            else:
+                disabled = True
+            # Accoustic 
+            accoustic = st.selectbox("Accoustic Audio", ["Audio 1", "Audio 2", "Audio 3"], disabled=disabled)
+            st.write('\n')
+            
+            # Time interval for pheremones discharge in minutes)
+            pheremones_interval = st.slider("Pheremones Discharge Interval (minutes)", min_value=5, max_value=60, value=15, step=5, disabled=disabled)
+            st.write('\n')
+
+            # Pest activity threshold for alerts
+            pest_activity_threshold = st.slider("Fly Count Threshold to Send Out Alerts", min_value=0, max_value=100, value=30, step=5, disabled=disabled)
+            st.write('\n')
+
+        # Logout
+        logout_col1, logout_col2 = st.columns([6,1])
+        with logout_col2:    
+            st.write('\n')
+            st.write('\n')
+            st.write('\n')
+            authenticator.logout('Logout', 'main')
+                    
+

credentials.yaml

@@ -0,0 +1,22 @@
+credentials:
+  usernames:
+    student:
+      email: student
+      name: NUS Student
+      password: $2b$12$W8tHKNQBu4GBvCA246tr7O1vZdfDrYXFJFG2DmTcAGl4IxeGSdf3e
+    staff:
+      email: staff
+      name: NUS Staff
+      password: $2b$12$W8tHKNQBu4GBvCA246tr7O1vZdfDrYXFJFG2DmTcAGl4IxeGSdf3e
+    advanced:
+      email: advanced
+      name: Advanced User
+      password: $2b$12$W8tHKNQBu4GBvCA246tr7O1vZdfDrYXFJFG2DmTcAGl4IxeGSdf3e
+
+cookie:
+  expiry_days: 30
+  key: r3#dsd7k&SDFjd*393B
+  name: cookie
+preauthorized:
+  emails:
+  -