import streamlit as st
from ai_assistant import ai_doctor_chat

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

# Create layout with two columns
left_column, right_column = st.columns([1, 3])

# Display image in the left column
left_column.image("ai_doctor_img.jpg", width=200, use_column_width="auto")

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

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