import streamlit as st

st.set_page_config(layout='wide')

# Title
st.title("LLM Basics Demo - with Huggingface Pipeline()")

# 3 Column Layout
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 an option", options)
with col3:
    llm_button = st.button("Generate Data")

# Display Input in single Layout
col4 = st.columns(1)
with col4:
    llm_input_text = st.text_area("Enter your message", "")

# Display Output in single Layout
col5 = st.columns(1)
with col5:
    llm_output_text = st.text_area("Generate Output", "")