Spaces:
Build error
Build error
| 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("DocChat :" + " " + response["message"]) | |
| chat_history.append("You :" + " " + user_input) | |
| 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') | |