init

main.py

@@ -0,0 +1,68 @@
+import json
+import os
+import streamlit as st
+import random
+import time
+import requests
+from dotenv import load_dotenv
+
+load_dotenv()
+
+conversation_id = ''
+
+# Streamed response emulator
+def response_generator(query):
+    global conversation_id
+    word = ''
+
+    res =requests.post(
+        f'{os.environ['api_url']}',
+        headers={
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {os.environ['api_key']}'
+        },
+        json={
+            "inputs": {},
+            "query": query,
+            "response_mode": "blocking",
+            "conversation_id": conversation_id,
+            "user": "abc-123",
+        }
+    )
+    if res.status_code == 200:
+        data = json.loads(res.text)
+
+        conversation_id = data['conversation_id']
+
+        answer = data.get('answer','Unable to answer at the moment')
+    else:
+        answer =  'Error'
+    
+    for word in answer.split():
+            yield word + " "
+            time.sleep(0.05)
+
+st.title("Ask Amy")
+
+# 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,41 @@
+altair==5.4.1
+attrs==24.2.0
+blinker==1.8.2
+cachetools==5.5.0
+certifi==2024.8.30
+charset-normalizer==3.3.2
+click==8.1.7
+gitdb==4.0.11
+GitPython==3.1.43
+idna==3.8
+Jinja2==3.1.4
+jsonschema==4.23.0
+jsonschema-specifications==2023.12.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+narwhals==1.6.0
+numpy==2.1.0
+packaging==24.1
+pandas==2.2.2
+pillow==10.4.0
+protobuf==5.28.0
+pyarrow==17.0.0
+pydeck==0.9.1
+Pygments==2.18.0
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+pytz==2024.1
+referencing==0.35.1
+requests==2.32.3
+rich==13.8.0
+rpds-py==0.20.0
+six==1.16.0
+smmap==5.0.1
+streamlit==1.38.0
+tenacity==8.5.0
+toml==0.10.2
+tornado==6.4.1
+typing_extensions==4.12.2
+tzdata==2024.1
+urllib3==2.2.2