import streamlit as st
from ai_assistant import ai_doctor

# Create layout with two containers
left_container, right_container = st.columns([1, 3])

# Display image in the left container
with left_container:
    st.image("ai_doctor_img.jpg", caption="Your AI Doctor Image", width=200)

# Display title in the right container
with right_container:
    st.markdown("<h1 style='text-align: center;'>Your AI Doctor &#129302;</h1>", unsafe_allow_html=True)

    # Create a text input box for the OpenAI key
    openai_key = st.text_input('Enter your OpenAI Key', type='password')

    query = st.text_input('Enter your query', type='default')
    submit = st.button('Submit')
    if submit:
        if query and openai_key:
            try:
                with st.spinner('Processing your query...'):
                    response = mh_assistant(openai_key, query)
                st.write(response)
            except Exception as e:
                st.error(f'An error occurred: {e}', icon=':no_entry_sign:')
        else:
            st.error('Please enter your OpenAI key and Query both!', icon="🚨")