Update app.py

app.py

@@ -138,10 +138,9 @@ def process_text(user_name, text_input):
 
 # Function to process image
 def process_image(user_name, image_input, user_prompt):
-    if isinstance(image_input, str):
-        with open(image_input, "rb") as image_file:
-            image_input = image_file.read()
+    image = Image.open(BytesIO(image_input))
     base64_image = base64.b64encode(image_input).decode("utf-8")
+    
     response = client.chat.completions.create(
         model=MODEL,
         messages=[
@@ -154,11 +153,19 @@ def process_image(user_name, image_input, user_prompt):
         temperature=0.0,
     )
     image_response = response.choices[0].message.content
-    st.markdown(image_response)
+    
     timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
+    st.session_state.messages.append({"user": user_name, "message": image_response, "timestamp": timestamp})
+    
+    with st.chat_message(user_name):
+        st.image(image)
+        st.markdown(f"{user_name} ({timestamp}): {user_prompt}")
+    
+    with st.chat_message("Assistant"):
+        st.markdown(image_response)
+    
     filename_md = generate_filename(user_prompt, "md")
     create_file(filename_md, user_prompt, image_response, user_name, timestamp)
-    st.session_state.messages.append({"user": user_name, "message": image_response, "timestamp": timestamp})
     save_data()
     return image_response