Update app.py

app.py

@@ -10,7 +10,7 @@ import base64
 # Load environment variables
 load_dotenv()
 
-icons = {"assistant": "./chat-assistant.jpeg", "user": "./hum.jpeg"}
+icons = {"assistant": "👽", "user": "./hum.jpeg"}
 
 # Configure the Llama index settings
 Settings.llm = HuggingFaceInferenceAPI(
@@ -107,16 +107,10 @@ if user_prompt and uploaded_file:
     st.session_state.messages.append({'role': 'user', "content": user_prompt})
     with st.chat_message("user", avatar="./hum.jpeg"):
         st.write(user_prompt)
-    # response = handle_query(user_prompt)
-    # st.session_state.messages.append({'role': 'assistant', "content": response})
-
-# for message in st.session_state.messages:
-#     with st.chat_message(message['role']):
-#         st.write(message['content'])
 
 # Generate a new response if last message is not from assistant
 if st.session_state.messages[-1]["role"] != "assistant":
-    with st.chat_message("assistant",avatar="./chat-assistant.jpeg"):
+    with st.chat_message("assistant",avatar="👽"):
         response = handle_query(user_prompt)
         full_response = st.write_stream(response)
     message = {"role": "assistant", "content": full_response}