Update app.py

app.py

@@ -67,23 +67,26 @@ def clear_conversation():
     st.session_state['file_uploader_key'] = str(uuid.uuid4())
 
 def display_chat_history():
-    for entry in st.session_state['chat_history']:
-        role = entry["role"]
-        parts = entry["parts"][0]
-        if 'text' in parts:
-            st.markdown(f"**{role.title()}:** {parts['text']}")
-        elif 'data' in parts:
-            mime_type = parts.get('mime_type', '')
-            if mime_type.startswith('image'):
-                st.image(Image.open(io.BytesIO(base64.b64decode(parts['data']))), caption='Uploaded Image', use_column_width=True)
-            elif mime_type == 'application/pdf':
-                st.write("**PDF Content:**") 
-                pdf_reader = PyPDF2.PdfReader(io.BytesIO(base64.b64decode(parts['data'])))
-                for page_num in range(len(pdf_reader.pages)):
-                    page = pdf_reader.pages[page_num]
-                    st.write(page.extract_text())
-            elif mime_type.startswith('video'):
-                st.video(io.BytesIO(base64.b64decode(parts['data'])))
+    chat_container = st.empty()  # Create an empty container to hold the chat history
+    with chat_container.container():  # Use a container to update the chat history
+        for entry in st.session_state['chat_history']:
+            role = entry["role"]
+            parts = entry["parts"][0]
+            if 'text' in parts:
+                st.markdown(f"**{role.title()}:** {parts['text']}")
+            elif 'data' in parts:
+                mime_type = parts.get('mime_type', '')
+                if mime_type.startswith('image'):
+                    st.image(Image.open(io.BytesIO(base64.b64decode(parts['data']))), 
+                             caption='Uploaded Image', use_column_width=True)
+                elif mime_type == 'application/pdf':
+                    st.write("**PDF Content:**") 
+                    pdf_reader = PyPDF2.PdfReader(io.BytesIO(base64.b64decode(parts['data'])))
+                    for page_num in range(len(pdf_reader.pages)):
+                        page = pdf_reader.pages[page_num]
+                        st.write(page.extract_text())
+                elif mime_type.startswith('video'):
+                    st.video(io.BytesIO(base64.b64decode(parts['data'])))
 
 # --- Send Message Function ---
 def send_message():
@@ -126,6 +129,8 @@ def send_message():
     st.session_state.user_input = ''
     st.session_state.uploaded_files = []
     st.session_state.file_uploader_key = str(uuid.uuid4())
+
+    # Update the chat history display
     display_chat_history()
 
 # --- User Input Area ---
@@ -141,7 +146,7 @@ with col2:
     send_button = st.button(
         "Send",
         on_click=send_message,
-        type="primary" # Makes the Send button prominent
+        type="primary"
     )
 
 # --- File Uploader ---