rerank model

RAG/bedrock_agent.py

@@ -6,7 +6,6 @@ from io import BytesIO
 import uuid
 import pprint
 import logging
-print(boto3.__version__)
 from PIL import Image 
 import os
 import base64

pages/Multimodal_Conversational_Search.py

@@ -185,15 +185,17 @@ def handle_input():
 
 
 def write_user_message(md):
-    col1, col2 = st.columns([3,97])
-    
-    with col1:
-        st.image(USER_ICON, use_column_width='always')
-    with col2:
-        #st.warning(md['question'])
+    placeholder = st.empty()
+    with placeholder.container():
+        col1, col2 = st.columns([3,97])
+        
+        with col1:
+            st.image(USER_ICON, use_column_width='always')
+        with col2:
+            #st.warning(md['question'])
 
-        st.markdown("<div style='color:#e28743';font-size:18px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;font-style: italic;'>"+md['question']+"</div>", unsafe_allow_html = True)
-       
+            st.markdown("<div style='color:#e28743';font-size:18px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;font-style: italic;'>"+md['question']+"</div>", unsafe_allow_html = True)
+        
 
 
 def render_answer(question,answer,index,res_img):
@@ -280,31 +282,31 @@ def render_answer(question,answer,index,res_img):
                 st.write(answer["source"])
                
         
-    with col_3:
-        if(index == len(st.session_state.questions_)):
-
-            rdn_key = ''.join([random.choice(string.ascii_letters)
-                              for _ in range(10)])
-            currentValue = ''.join(st.session_state.input_rag_searchType)+str(st.session_state.input_is_rerank)+str(st.session_state.input_table_with_sql)+st.session_state.input_index
-            oldValue = ''.join(st.session_state.inputs_["rag_searchType"])+str(st.session_state.inputs_["is_rerank"])+str(st.session_state.inputs_["table_with_sql"])+str(st.session_state.inputs_["index"])
-            def on_button_click():
-                if(currentValue!=oldValue or 1==1): 
-                    st.session_state.input_query = st.session_state.questions_[-1]["question"]
-                    st.session_state.answers_.pop()
-                    st.session_state.questions_.pop()
+    # with col_3:
+    #     if(index == len(st.session_state.questions_)):
+
+    #         rdn_key = ''.join([random.choice(string.ascii_letters)
+    #                           for _ in range(10)])
+    #         currentValue = ''.join(st.session_state.input_rag_searchType)+str(st.session_state.input_is_rerank)+str(st.session_state.input_table_with_sql)+st.session_state.input_index
+    #         oldValue = ''.join(st.session_state.inputs_["rag_searchType"])+str(st.session_state.inputs_["is_rerank"])+str(st.session_state.inputs_["table_with_sql"])+str(st.session_state.inputs_["index"])
+    #         def on_button_click():
+    #             if(currentValue!=oldValue or 1==1): 
+    #                 st.session_state.input_query = st.session_state.questions_[-1]["question"]
+    #                 st.session_state.answers_.pop()
+    #                 st.session_state.questions_.pop()
                     
-                    handle_input()
-                    with placeholder.container():
-                        render_all()
-            if("currentValue"  in st.session_state):
-                del st.session_state["currentValue"]
-
-            try:
-                del regenerate
-            except:
-                pass  
-            placeholder__ = st.empty()
-            placeholder__.button("🔄",key=rdn_key,on_click=on_button_click)
+    #                 handle_input()
+    #                 with placeholder.container():
+    #                     render_all()
+    #         if("currentValue"  in st.session_state):
+    #             del st.session_state["currentValue"]
+
+    #         try:
+    #             del regenerate
+    #         except:
+    #             pass  
+    #         placeholder__ = st.empty()
+    #         placeholder__.button("🔄",key=rdn_key,on_click=on_button_click)
             
         
 #Each answer will have context of the question asked in order to associate the provided feedback with the respective question
@@ -335,7 +337,7 @@ with placeholder.container():
 st.markdown("")
 col_2, col_3 = st.columns([75, 20])
 with col_2:
-    st.text_input("Ask here", key="input_query", label_visibility="collapsed")
+    input = st.text_input("Ask here", key="input_query", label_visibility="collapsed")
 with col_3:
     play = st.button("Go",on_click=handle_input, key="play")