Initial commit

.gitignore

@@ -0,0 +1,132 @@
+venv/
+# 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/
+pip-wheel-metadata/
+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/
+
+# 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
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.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
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__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/
+
+.vscode/

.streamlit/config.toml

@@ -0,0 +1,10 @@
+[theme]
+base="dark"
+primaryColor="#e28198"
+backgroundColor="#061222"
+secondaryBackgroundColor="#8aafd8"
+textColor="#fee5d7"
+font="monospace"
+
+[server]
+enableStaticServing = true

Dockerfile

@@ -0,0 +1,27 @@
+FROM python:3.8.9
+
+WORKDIR /app
+
+COPY ./requirements.txt /app/requirements.txt
+COPY ./packages.txt /app/packages.txt
+
+RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
+RUN pip3 install --no-cache-dir -r /app/requirements.txt
+
+# User
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME /home/user
+ENV PATH $HOME/.local/bin:$PATH
+
+WORKDIR $HOME
+RUN mkdir app
+WORKDIR $HOME/app
+COPY . $HOME/app
+
+EXPOSE 8501
+CMD streamlit run app.py \
+    --server.headless true \
+    --server.enableCORS false \
+    --server.enableXsrfProtection false \
+    --server.fileWatcherType none

README.md

@@ -1,4 +1,5 @@
 ---
+<<<<<<< HEAD
 title: Naomi
 emoji: 🏆
 colorFrom: blue
@@ -9,3 +10,13 @@ license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+=======
+title: Naomi Chatbot
+sdk: docker
+app_port: 8501
+pinned: false
+
+---
+
+# naomi
+>>>>>>> 47fcbb2 (Initial commit)

app.py

@@ -0,0 +1,108 @@
+"""
+
+    app.py
+
+    Main UI script with streamlit.
+
+    Widget keys:
+        'FormSubmitter:intake-Submit': form button key returns value true if clicked (bool) FormSubmitter:{form-key}-{form-submission-key}
+"""
+
+import time
+import random
+import streamlit as st
+from datetime import datetime
+
+# Streamed response emulator
+def response_generator():
+    response = random.choice(
+        [
+            f"Hello there! {st.session_state['candidate_name']}",
+            f"Hi, {st.session_state['candidate_name']}! Is there anything I can help you with?",
+            f"Do you need help? {st.session_state['candidate_name']}",
+        ]
+    )
+    for word in response.split():
+        yield word + " "
+        time.sleep(0.05)
+
+# Title of the app
+st.title("Chatbot Naomi")
+
+print('Initial Session state', st.session_state)
+
+contact_options = ['Instagram', 'Email', 'Number']
+intake_form = [
+    'candidate_name',
+    'candidate_contact_type',
+    'candidate_contact',
+    'candidate_dob',
+    'candidate_location',
+    'intake_submission'
+]
+datetime_format = '%Y-%m-%d %H:%M:%S'
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+@st.dialog('Intake form', width='large')
+def open_intake_form():
+    st.markdown('Fill in your detaisl below to start chat session :)')
+    st.session_state.candidate_name = st.text_input("Enter your name", key='candidate_name')
+    contact_col_1, contact_col_2 = st.columns(spec=[0.3, 0.7], vertical_alignment='center')
+    contact_col_1.selectbox("Select contact option", contact_options, key='candidate_contact_type')
+    contact_col_2.text_input('Enter your username', key='candidate_contact')
+    st.session_state.candidate_dob = st.date_input("When is your birthday?", key='candidate_dob')
+    st.session_State.candidate_location = st.text_input('Enter your location', key='candidate_location')
+    #button = st.button('Submit', use_container_width=True, type='primary')
+    # after the button is clicked the page automatically reruns and the workflow starts from the beginning
+
+    if st.button('Submit', use_container_width=True, type='primary', key='intake_submission'):
+        print('Session state after submission for user input: ', st.session_state)
+        time.sleep(1)
+        st.rerun()
+
+@st.fragment()
+def open_chat_window(**kwargs):
+    # adds to current state (deletes if doesnt)
+    st.session_state.update(kwargs)
+
+    st.markdown('Welcome to the chat!')
+    msgbox = st.container(height=400, border=False)
+
+    # Display existing chat messages
+    for message in st.session_state.messages:
+        msgbox.chat_message(message["role"]).write(message['content'])
+
+    if user_input := st.chat_input('Enter your message'):
+        # Add user message to chat history
+        print(f'State: {st.session_state}\nUser inserted message: {user_input}')
+
+        st.session_state.messages.append({"role": "user", "content": user_input, "timestamp": datetime.now()})
+        # Display user message in chat message container
+        msgbox.chat_message("user").write(user_input)
+        response = msgbox.chat_message('assistant').write_stream(response_generator())
+
+        # Append assistant's response to the messages history
+        st.session_state.messages.append({"role": "assistant", "content": response, "timestamp": datetime.now()})
+
+    undo_button, reset_button = st.columns(2)
+    if undo_button.button('Undo message', use_container_width=True, type='secondary'):
+        st.session_state['undo_button'] = True
+    if reset_button.button('Reset chat', use_container_width=True, type='primary'):
+        st.session_state.clear()
+        time.sleep(1)
+        st.rerun()
+
+def main():
+    if 'intake_submission' not in st.session_state:
+        if st.button('Start chat session . . .', type='primary', key='open_intake'):
+            st.session_state.open_intake = True
+            open_intake_form()
+    else:
+        st.session_state['candidate_name'] = st.session_state['candidate_name'].lower().capitalize()
+        open_chat_window(**st.session_state)
+        #render_agent_mood()
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+requests

tests/test_app.py

@@ -0,0 +1,44 @@
+"""
+    test_app.py
+
+    Tests the main application.
+
+"""
+import warnings
+import pytest
+import streamlit as st
+from streamlit.testing.v1 import AppTest
+from app import open_intake_form
+
+warnings.filterwarnings("ignore", message="missing ScriptRunContext")
+warnings.filterwarnings("ignore", category=DeprecationWarning, message=".*custom tp_new.*")
+
+intake_form = [
+    'candidate_name',
+    'candidate_contact_type',
+    'candidate_contact',
+    'candidate_dob',
+    'candidate_location',
+    'intake_submission'
+]
+
+@pytest.fixture
+def app():
+    return AppTest.from_file('app.py').run()
+
+@pytest.fixture
+def form():
+    form = AppTest.from_function(open_intake_form).run()
+    form.session_state.open_intake = True
+
+    for key in intake_form:
+        form.session_state[key] = None
+    return form
+
+def test_new_session(app):
+    assert 'messages' in app.session_state, f"`messages` not found in session state"
+
+def test_open_intake_button(app):
+    # Simulate entering details
+    app.button(key='open_intake').click()
+    assert 'open_intake' in app.session_state