init

app.py

@@ -0,0 +1,57 @@
+import os
+import streamlit as st
+import time
+# from dotenv import load_dotenv
+from chatbot_function.chatbot import answer
+from huggingface_hub import Repository
+from huggingface_hub import login
+
+# load_dotenv()
+
+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()
+
+
+# Streamed response emulator
+def response_generator(query):
+    ans = answer(query)
+    
+    for word in ans.split(' '):
+            yield word + " "
+            time.sleep(0.05)
+
+st.title("Nirvana 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,62 @@
+altair==5.5.0
+annotated-types==0.7.0
+anyio==4.7.0
+attrs==24.3.0
+blinker==1.9.0
+cachetools==5.5.0
+certifi==2024.12.14
+charset-normalizer==3.4.1
+click==8.1.8
+distro==1.9.0
+filelock==3.16.1
+fsspec==2024.12.0
+gitdb==4.0.11
+GitPython==3.1.43
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.27.0
+idna==3.10
+Jinja2==3.1.5
+jiter==0.8.2
+joblib==1.4.2
+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.19.1
+numpy==2.2.1
+openai==1.58.1
+packaging==24.2
+pandas==2.2.3
+pillow==11.0.0
+protobuf==5.29.2
+pyarrow==18.1.0
+pydantic==2.10.4
+pydantic_core==2.27.2
+pydeck==0.9.1
+Pygments==2.18.0
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+pytz==2024.2
+PyYAML==6.0.2
+referencing==0.35.1
+requests==2.32.3
+rich==13.9.4
+rpds-py==0.22.3
+scikit-learn==1.6.0
+scipy==1.14.1
+six==1.17.0
+smmap==5.0.1
+sniffio==1.3.1
+streamlit==1.41.1
+tabulate==0.9.0
+tenacity==9.0.0
+threadpoolctl==3.5.0
+toml==0.10.2
+tornado==6.4.2
+tqdm==4.67.1
+typing_extensions==4.12.2
+tzdata==2024.2
+urllib3==2.3.0