Update app.py

app.py

@@ -1,14 +1,14 @@
 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")
-
-# Display title in the right column
-right_column.markdown("<h1 style='text-align: center;'>Your AI Doctor &#129302;</h1>", unsafe_allow_html=True)
+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')
@@ -18,7 +18,7 @@ submit = right_column.button('Submit')
 if submit:
     if query and openai_key:
         try:
-            with right_column.spinner('Processing your query...'):
+            with st.spinner('Processing your query...'):
                 response = ai_doctor_chat(openai_key, query)
             right_column.write(response)
         except Exception as e: