Update app.py

app.py

@@ -76,6 +76,22 @@ st.title("RAG Extractor")
 if 'messages' not in st.session_state:
     st.session_state.messages = [{'role': 'assistant', "content": 'Hello I am Pingu! Upload a PDF and ask me anything about its content.'}]
 
+# Define your custom icon
+custom_icon_url = "./custom_icon.png"  # Adjust this to your icon's file path
+
+# CSS to customize the chatbot icon
+st.markdown(f"""
+    <style>
+        .chat-container .stChatMessage.st-chat-bot {
+            background-image: url({custom_icon_url});
+            background-size: contain;
+            background-repeat: no-repeat;
+            background-position: left center;
+            padding-left: 60px; /* Adjust this value according to the icon size */
+        }
+    </style>
+    """, unsafe_allow_html=True)
+
 with st.sidebar:
     st.title("Input")
     uploaded_file = st.file_uploader("Upload your PDF Files and then click on the Submit & Process Button")
@@ -84,7 +100,7 @@ with st.sidebar:
             filepath = "data/saved_pdf.pdf"
             with open(filepath, "wb") as f:
                 f.write(uploaded_file.getbuffer())
-    displayPDF(filepath)  
+            displayPDF(filepath)  
             data_ingestion()  # Process PDF every time new file is uploaded
             st.success("PDF is ready!")