v0.0.6 sidebar

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 import os
 import re
 import sys
+import base64
 import logging
 logging.basicConfig(stream=sys.stdout, level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -10,7 +11,7 @@ logger = logging.getLogger(__name__)
 from dotenv import load_dotenv
 load_dotenv()
 
-os.environ['AWS_DEFAULT_REGION'] = 'us-west-2'
+#os.environ['AWS_DEFAULT_REGION'] = 'us-west-2'
 
 for key in st.session_state.keys():
     #del st.session_state[key]
@@ -40,6 +41,39 @@ def set_baseten_key(bs_api_key):
 
 set_baseten_key(bs_api_key)
 
+def autoplay_video(video_path):
+    with open(video_path, "rb") as f:
+        video_content = f.read()
+
+    video_str = f"data:video/mp4;base64,{base64.b64encode(video_content).decode()}"
+    st.markdown(f"""
+        <video style="display: block; margin: auto; width: 140px;" controls loop autoplay width="140" height="180">
+            <source src="{video_str}" type="video/mp4">
+        </video>
+        """, unsafe_allow_html=True)
+
+# sidebar
+with st.sidebar:
+    st.header('KG Questions')
+    video, text = st.columns([2, 2])
+    with video:
+        autoplay_video('docs/images/kg_construction.mp4')
+    with text:
+        st.write(
+f'''
+###### The construction of a Knowledge Graph is mesmerizing.
+###### Concepts in the middle are what most are doing. Are we considering anything different? Why? Why not?
+###### Concepts on the edge are what few are doing. Are we considering that? Why? Why not?
+'''
+)
+    st.write(
+f'''
+#### How can <what most are doing> help with <what few are doing>?
+''')
+
+
+
+
 from llama_index import StorageContext
 from llama_index import ServiceContext
 from llama_index import load_index_from_storage 
@@ -188,11 +222,11 @@ def clear_question(query_model):
         st.session_state.question_input = ''
         st.session_state.question_answered = False
         st.session_state.answer = ''
+        st.session_state.answer_rating = 3
         st.session_state.prev_model = query_model
 
 
 initial_query = ''
-#st.session_state.prev_model = None
 
 if 'question' not in st.session_state:
     st.session_state.question = ''
@@ -249,15 +283,10 @@ def submit():
     st.session_state.question_input = ''
     st.session_state.question_answered = False
 
-#def submit_rating(query_model, req, resp):
-#    print(f'query model {query_model}')
-#    if 'answer_rating' in st.session_state:
-#        print(f'rating {st.session_state.answer_rating}')
 
 st.write(f'Model, question, answer and rating are logged to help with the improvement of this application.')
 question = st.text_input("Enter a question, e.g. What benchmarks can we use for QA?", key='question_input',  on_change=submit )
  
-# answer_str = None
 if(st.session_state.question):
     col1, col2 = st.columns([2, 2])
     with col1:
@@ -277,25 +306,16 @@ if(st.session_state.question):
                 st.write(f' Please rate this answer.')
         with col2:
             from streamlit_star_rating import st_star_rating
-            stars = st_star_rating("", maxValue=5, defaultValue=3, key="answer_rating", 
-                               # customCSS = "div {background-color: red;}"
-                               # on_change = submit_rating(query_model, st.session_state.question, answer_str)
-                               )
-        print(f"------stars {stars}")
+            stars = st_star_rating("", maxValue=5, defaultValue=3, key="answer_rating")
+        #print(f"------stars {stars}")
     except Exception as e:
-        print(e)
-        answer_str = str(e)
+        #print(f'{type(e)}, {e}')
+        answer_str = f'{type(e)}, {e}'
         st.session_state.answer_rating = -1
+        st.write(f'An error occured, please try again. \n{answer_str}')
     finally:
         if 'question' in st.session_state:
             req = st.session_state.question
-            #st.session_state.question = ''
             if(__spaces__):
-                #request_log = get_request_log()
                 st.session_state.request_log.add_request_log_entry(query_model, req, answer_str, st.session_state.answer_rating)
  
-   # if "answer_rating" in st.session_state:
-   #     if(__spaces__):
-   #         print('time to log the rating')
-   #         #request_log = get_request_log()
-   #         st.session_state.request_log.add_request_log_entry(query_model, req, answer_str, st.session_state.answer_rating)

kron/llm_predictor/KronBasetenCamelLLM.py

@@ -27,6 +27,7 @@ class KronBasetenCamelLLM(Baseten):
             model = baseten.deployed_model_id(self.model)
             response = model.predict({"instruction": prompt, **kwargs})
 
+        print(f'baseten response: {response}')
         response_txt = response['completion']
         #print(f'\n********{response_txt}')
         return response_txt

kron/persistence/dynamodb_request_log.py

@@ -12,7 +12,7 @@ session = boto3.Session(
 #    aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
 )
 
-logger.info(f'region name {session.region_name}')
+logger.debug(f'region name {session.region_name}')
 dynamodb = session.resource('dynamodb')
 
 class RequestLog: