Initial Draft

app.py

@@ -4,17 +4,11 @@ import streamlit as st
 ####### Generic functions ######
 ################################
 
-llm_action_type = None
-llm_button = None
-llm_input_text = None
-llm_output_text = None
-
-
 def option_change():
-    llm_input_text="Hello"
+    st.session_state.input_text="Hello"
 
 def button_clicked():
-    llm_output_text="Button Clicked"
+    st.session_state.output_text="Button Clicked"
 
 ################################
 ####### Display of data ########
@@ -29,12 +23,12 @@ col1, col2, col3 = st.columns(3)
 
 with col1:
     options = ["Sentiment Analysis", "Zero-shot Classification", "Text Generation","Mask Filling","Named Entity Recognition","Question & Answering","Summarization","English to French Translation"]
-    llm_action_type = st.selectbox("Select LLM Tasktype", options, on_change=option_change)
+    llm_action_type = st.selectbox("Select LLM Tasktype", options,key="action_type", on_change=option_change)
 with col3:
     llm_button = st.button("Generate Data",on_click=button_clicked)
 
 # Display Input in single Layout
-llm_input_text = st.text_area(label="Enter your message", height=400)
+llm_input_text = st.text_area(label="Enter your message",key="input_text", height=400)
 
 # Display Output in single Layout
-llm_output_text = st.text_area("Generate Output", height=400)
+llm_output_text = st.text_area("Generate Output",key="output_text", height=400)