changed layout and added output text area for generated responses

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import torch
+from pandas import options
+from streamlit_tags import st_tags
 from transformers import BartForConditionalGeneration, BartTokenizer
 
 # initialize model + tok variables
@@ -17,6 +19,7 @@ examples = [
 # descriptions = "Interview question remake is a model that..."
 
 # pass in Strings of model choice and input text for context
+@st.cache
 def genQuestion(model_choice, context):
     # global descriptions
     if model_choice=="interview-question-remake":
@@ -42,12 +45,20 @@ def genQuestion(model_choice, context):
     return final_output
 
 
-    
+# Wide page layout (instead of having a narrower, one-column page layout)
+st.set_page_config(layout="wide")
+
 # Title
 st.title("Interview AI Test Website")
 
-# Input field
-input = st.text_input('Context')
+# Adding a Session State to store stateful variables and for saving user's labels/tags for generated questions
+if 'button_sent' not in st.session_state:
+    st.session_state.button_sent = False
+
+# Input fields
+input = st.text_input('Context')                                    # user inputs context to construct a response (str)
+
+maxl, minl = st.columns(2)
 
 option = st.selectbox(
                 'Please select a model.',
@@ -60,8 +71,23 @@ elif option == 'interview-length-tagged':
 elif option == 'reverse-interview-question':
     st.write("This model asks a question that would have resulted in the context you provide (a.k.a. it traverses backward through the interview)")
 
-if st.button('Submit'):
+
+# Column layout to display generated responses alongside tags
+col1, col2 = st.columns((3, 1))
+
+if st.button('Submit') or st.session_state.button_sent:
     with st.spinner('Generating a response...'):
         output = genQuestion(option, input)
         print(output)
-    st.write(output)
+    # st.write(output)
+    st.session_state.button_sent = True
+    col1.text_area(label="Generated Responses:", value=output, disabled=True, height=200)
+
+
+
+# TODO:
+#   - disable multiselect widget when responses are being generated AND when a question is not selected to be tagged
+#   - connect tags with an individual question
+#   - save session state so tags associated with their respective questions can also be saved
+#   - write/store the saved state data to some database for future use?
+#   - brainstorm good names for tags/labels OR allow users to enter their own tag names if possible