components/body/prompt.py

import streamlit as st


def prompt_box():
    prompt = st.text_area("Enter your question here", height=100)
    if st.session_state.get('generate_answer_button', None):
        if prompt == "" or prompt is None:
            st.caption(":red[Please enter a prompt]")
    
    
    if prompt is not None: # If prompt is not empty
        st.session_state['prompt'] = prompt
    

if __name__ == "__main__":
    prompt_box()