init

app.py

@@ -0,0 +1,55 @@
+import os
+import streamlit as st
+import time
+from huggingface_hub import Repository
+from huggingface_hub import login
+
+login(token = os.environ['HF_TOKEN'])
+
+repo = Repository(
+    local_dir="chatbot_function",
+    repo_type="dataset",
+    clone_from=os.environ['DATASET'],
+    token=True
+)
+repo.git_pull()
+
+from chatbot_function.agent import ask
+
+
+# Streamed response emulator
+def response_generator(query):
+    ans = ask(query)
+    
+    for word in ans.split(' '):
+            yield word + " "
+            time.sleep(0.05)
+
+st.title("HR Chatbot")
+
+if 'conversation_id' not in st.session_state:
+    st.session_state['conversation_id'] = ''
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# Accept user input
+if prompt := st.chat_input("What is up?"):
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    # Display user message in chat message container
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        response = st.write_stream(response_generator(prompt))
+
+    # Add assistant response to chat history
+    st.session_state.messages.append({"role": "assistant", "content": response})

requirements.txt

@@ -0,0 +1,45 @@
+altair==5.5.0
+attrs==25.1.0
+blinker==1.9.0
+cachetools==5.5.1
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+filelock==3.17.0
+fsspec==2025.2.0
+gitdb==4.0.12
+GitPython==3.1.44
+huggingface-hub==0.28.1
+idna==3.10
+Jinja2==3.1.5
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+mdurl==0.1.2
+narwhals==1.24.2
+numpy==2.2.2
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+protobuf==5.29.3
+pyarrow==19.0.0
+pydeck==0.9.1
+Pygments==2.19.1
+python-dateutil==2.9.0.post0
+pytz==2025.1
+PyYAML==6.0.2
+referencing==0.36.2
+requests==2.32.3
+rich==13.9.4
+rpds-py==0.22.3
+six==1.17.0
+smmap==5.0.2
+streamlit==1.41.1
+tenacity==9.0.0
+toml==0.10.2
+tornado==6.4.2
+tqdm==4.67.1
+typing_extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0