fixed bug in second tab

app.py

@@ -53,11 +53,10 @@ def process_input_passage_finder(message):
     return html_output
 
 # Chatbot functions
+
 def respond_chatbot(query, history):
     formatted_query = [HumanMessage(content=query)]
-    # Invoke the graph with properly formatted input
     result = chatbot.graph.invoke({"messages": formatted_query}, chatbot.config)
-    # Get the passages from the graph and append to history if documents exist
     state = chatbot.graph.get_state(config=chatbot.config).values
     documents = state.get("documents")
     passages = ''
@@ -65,12 +64,13 @@ def respond_chatbot(query, history):
         for d in documents:
             passages += f'<b>{d.metadata["publication_name"]} - {d.metadata["chapter_name"]}</b>\n{d.page_content}\n\n'
         history.append((f'Passages: {query}', passages))
-    # Extract the assistant's response and append to history
     response = result["messages"][-1].content
     system_message_dropdown = state.get("system_message_dropdown")
     history.append((query, f"<i>[{system_message_dropdown}]</i>\n" + response))
     return history
 
+
+
 # Generic Chatbot function
 def respond_genericchatbot(query, history):
     formatted_query = [HumanMessage(content=query)]
@@ -166,7 +166,6 @@ with gr.Blocks(css=css) as demo:
                         value=list(sp.system_prompt_templates.keys())[0],
                     )
                     
-                    # Chatbot description Textbox
                     chatbot_description = gr.Textbox(
                         value=sp.chatbot_descriptions[list(sp.system_prompt_templates.keys())[0]],
                         label="Chatbot Description",
@@ -192,12 +191,23 @@ with gr.Blocks(css=css) as demo:
                     </div>
                     """)
             
+            # system_prompt_dropdown.change(
+            #     fn=lambda x: (sp.chatbot_descriptions[x], sp.system_prompt_templates[x]),
+            #     inputs=[system_prompt_dropdown],
+            #     outputs=[chatbot_description, system_prompt_display]
+            # )
+            
+            def update_chatbot_info(selected_prompt):
+                chatbot.reset_system_prompt(selected_prompt)
+                return sp.chatbot_descriptions[selected_prompt], sp.system_prompt_templates[selected_prompt]
+
             system_prompt_dropdown.change(
-                fn=lambda x: (sp.chatbot_descriptions[x], sp.system_prompt_templates[x]),
+                fn=update_chatbot_info,
                 inputs=[system_prompt_dropdown],
                 outputs=[chatbot_description, system_prompt_display]
             )
-            
+
+
             submit_button_cb.click(
                 fn=respond_chatbot,
                 inputs=[user_input_cb, chatbot_output],