Update app.py

app.py

@@ -1,29 +1,27 @@
 import streamlit as st
 from ai_assistant import ai_doctor
 
-# Create layout with two containers
-left_container, right_container = st.columns([1, 3])
+# Create layout with two columns
+left_column, right_column = 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 image in the left column
+left_column.image("path_to_your_image.jpg", caption="Your AI Doctor Image", use_column_width=True)
 
-# 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)
+# Display title in the right column
+right_column.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')
+# Create a text input box for the OpenAI key
+openai_key = right_column.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="🚨")
+query = right_column.text_input('Enter your query', type='default')
+submit = right_column.button('Submit')
+if submit:
+    if query and openai_key:
+        try:
+            with right_column.spinner('Processing your query...'):
+                response = mh_assistant(openai_key, query)
+            right_column.write(response)
+        except Exception as e:
+            right_column.error(f'An error occurred: {e}', icon=':no_entry_sign:')
+    else:
+        right_column.error('Please enter your OpenAI key and Query both!', icon="🚨")