Update app.py

app.py

@@ -136,13 +136,13 @@ st.caption("Using GPT-4.1, IntFloat-e5-basev2 as Embedding model and Pinecone as
 for msg in st.session_state.messages:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
-        if "images" in msg and msg["images"]:
+        #if "images" in msg and msg["images"]:
             # Display images in a row
-            cols = st.columns(len(msg["images"]))
-            for idx, img_b64 in enumerate(msg["images"]):
-                with cols[idx]:
-                    if "," in img_b64: img_b64 = img_b64.split(",")[1]
-                    st.image(base64.b64decode(img_b64), use_container_width=True)
+            #cols = st.columns(len(msg["images"]))
+            #for idx, img_b64 in enumerate(msg["images"]):
+                #with cols[idx]:
+                    #if "," in img_b64: img_b64 = img_b64.split(",")[1]
+                    #st.image(base64.b64decode(img_b64), use_container_width=True)
 
 # Chat Input
 if prompt := st.chat_input("Ask about your document..."):
@@ -167,7 +167,7 @@ if prompt := st.chat_input("Ask about your document..."):
     with st.chat_message("assistant"):
         with st.spinner("Thinking (retrieving the context)..."):
             try:
-                answer, images = run_advanced_rag(
+                answer = run_advanced_rag(
                     prompt,
                     st.session_state.session_id,
                     st.session_state.bm25,
@@ -178,20 +178,20 @@ if prompt := st.chat_input("Ask about your document..."):
                 st.markdown(answer)
                 
                 # Display Images if found
-                if images:
-                    st.write("---")
-                    st.caption("📸 Retrieved Visual Context:")
-                    cols = st.columns(min(3, len(images)))
-                    for idx, img_b64 in enumerate(images[:3]):
-                        with cols[idx]:
-                            if "," in img_b64: img_b64 = img_b64.split(",")[1]
-                            st.image(base64.b64decode(img_b64), use_container_width=True)
+                #if images:
+                    #st.write("---")
+                    #st.caption("📸 Retrieved Visual Context:")
+                    #cols = st.columns(min(3, len(images)))
+                    #for idx, img_b64 in enumerate(images[:3]):
+                        #with cols[idx]:
+                            #if "," in img_b64: img_b64 = img_b64.split(",")[1]
+                            #st.image(base64.b64decode(img_b64), use_container_width=True)
                 
                 # Save to history
                 st.session_state.messages.append({
                     "role": "assistant", 
-                    "content": answer, 
-                    "images": images[:3] if images else []
+                    "content": answer
+                    #"images": images[:3] if images else []
                 })
                 
             except Exception as e: