draft

app.py

@@ -0,0 +1,65 @@
+import streamlit as st
+import requests
+import json
+import sqlite3
+import datetime
+import yaml
+
+chat_history = []
+#"st.session_state object", st.session_state
+
+with open('config.yaml', 'r') as file:
+    config = yaml.safe_load(file)
+
+docchat_engine_url = config['docchat-engine']['url']
+
+conn = sqlite3.connect('message.db')
+c = conn.cursor()
+
+
+#Get unique id from database and set the session unique id
+if 'session_unique_id' not in st.session_state:
+    #Insert timestamp of the request, the unique id is autogenerated at the table level. 
+    c.execute("INSERT INTO SESSION (TIMESTAMP) VALUES('{timestamp_value}')".format(timestamp_value=datetime.datetime.now()))
+    conn.commit()
+
+    #Fetch the autogenerated uniqueid from table (autoincremented number)
+    c.execute("SELECT UNIQUE_ID FROM SESSION ORDER BY UNIQUE_ID DESC LIMIT 1")
+    for tmp_select_return_msg in c.fetchone():
+        unique_id_from_db = tmp_select_return_msg
+    st.session_state['session_unique_id'] = unique_id_from_db
+
+
+def call_web_service(message):
+    response = requests.post('{value_url}/chat/'.format(value_url=docchat_engine_url), json={'text': message})
+    return response.json()
+
+st.title('Chat Interface')
+
+user_input = st.text_input("Type your message here:", key="user_input")
+if st.button("Send"):
+    if user_input:
+        response = call_web_service(user_input)
+        chat_history.append("You :" + " " + user_input)
+        chat_history.append("DocChat :" + " " + response["message"])
+        for message in chat_history:
+            format_sql_insert = 'INSERT INTO messages (message, unique_id) VALUES ("{value}","{uniqueid}")'.format(value=message,uniqueid=st.session_state['session_unique_id'])
+            print("DEBUG:  " + format_sql_insert)
+            c.execute(format_sql_insert)
+        conn.commit()
+
+        c.execute("Select message from messages where unique_id = {value_unique_id} order by id desc".format(value_unique_id=st.session_state['session_unique_id']))
+        msg_from_db = c.fetchall()
+
+        for op_message in msg_from_db:
+            st.write(op_message[0])
+
+if 'button_clicked' not in st.session_state:
+    st.session_state.button_clicked = False
+
+if st.sidebar.button('Load Chat'):
+    st.session_state.button_clicked = True
+
+if st.session_state.button_clicked:
+    response = requests.post('{value_url}/loadvector/'.format(value_url=docchat_engine_url), json={"text": "_chat"})
+    st.write('Context Loaded')

config.yaml

@@ -0,0 +1,2 @@
+docchat-engine:
+  url: 'https://docchat-u6rh.onrender.com'

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+datetime
+pyyaml