Update page2.py

page2.py

@@ -5,6 +5,11 @@ from streamlit_chat import message
 from PIL import Image
 import base64
 import io
+from IPython.display import display
+from IPython.display import Markdown
+
+import pathlib
+import textwrap
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
@@ -13,7 +18,6 @@ from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
 # Streamlit app
 def image():
 
-
     def process_image(uploaded_file):
         # Display the uploaded image
         image = Image.open(uploaded_file)
@@ -32,12 +36,13 @@ def image():
     llm = ChatGoogleGenerativeAI(model="gemini-pro-vision", google_api_key=apiKey)
 
     image_url = None  # Initialize image_url outside the if statement
+    with st.sidebar:
+        uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+        if uploaded_file is not None:
+            image_url = process_image(uploaded_file)
 
-    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-    if uploaded_file is not None:
-        image_url = process_image(uploaded_file)
-
-
+    with st.chat_message("assistant"):
+        st.write("Hello 👋, upload an image and ask questions related to it!")
     if 'messages' not in st.session_state:
         st.session_state['messages'] = []
 
@@ -63,8 +68,9 @@ def image():
                     "content": prompt
                 }
             )
-        response = llm.invoke([message])
-        text_output = response.content
+        with st.spinner('Generating...'):
+            response = llm.invoke([message])
+            text_output = response.content
 
         with st.chat_message("assistant").markdown(text_output):
             st.session_state.messages.append(