add sidebar buttons

app.py

@@ -260,6 +260,73 @@ def main():
         st.session_state.auto_submit = True
         st.session_state.auto_submit_text = text
 
+    # Quick input buttons - place these above the chat input
+    st.sidebar.subheader("Quick Inputs")
+    
+    # Create buttons in the sidebar
+    if st.sidebar.button("Example 1 C"):
+        send_preset_message("The density of cytokine receptor bearing cells has no effect on the distance over which cytokines act.")
+    
+    if st.sidebar.button("Example 2 C"):
+        send_preset_message("A total of 1,000 people in the UK are asymptomatic carriers of vCJD infection.")
+    
+    if st.sidebar.button("Example 3 C"):
+        send_preset_message("Sepsis related mortality has risen from 2009 to 2014.")
+    
+    if st.sidebar.button("Example 4 S"):
+        send_preset_message("IL-6 signaling plays a major role in atherosclerotic cardiovascular disease.")
+    
+    if st.sidebar.button("Example 5 S"):
+        send_preset_message("The severity of cardiac involvement in amyloidosis can be described by the degree of transmurality of late gadolinium enhancement in MRI.")
+
+    if st.sidebar.button("Example 6 S"):
+        send_preset_message("There was an estimated 30 million cases of pneumonia in young children worldwide in 2010.")
+
+    # Handle auto-submission if a button was clicked
+    if "auto_submit" in st.session_state and st.session_state.auto_submit:
+        prompt = st.session_state.auto_submit_text
+
+        # Process the message - in this example we just echo it back
+        # In a real app, you would have more complex logic here
+        # st.session_state.messages.append({"role": "assistant", "content": f"You said: {prompt}"})
+
+        display_message = prompt + "  \n"+ "  \n"+ f"Retriever: {selected_retriever}, Reasoner: {selected_reasoner}"
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        st.session_state.messages.append({"role": "summary", "content": display_message})
+
+        # Display user message in chat message container
+        with st.chat_message("user"):
+            st.markdown(display_message)
+
+        retrieved_documents = retriever(prompt, selected_retriever)
+        reasoning, decision = reasoner(prompt, retrieved_documents, selected_reasoner, llm_client)
+
+        # Display assistant response in chat message container
+        with st.chat_message("assistant"):
+            message_placeholder = st.empty()
+            full_response = ""
+            if decision.lower() == 'support':
+                assistant_response = f'The claim is CORRECT because {reasoning}'
+            elif decision.lower() == 'contradict':
+                assistant_response = f'The claim is INCORRECT because {reasoning}'
+            else:
+                assistant_response = f'Sorry, the query failed due to an issue with connecting to the LLM service.'
+
+            # Simulate stream of response with milliseconds delay
+            for chunk in assistant_response.split():
+                full_response += chunk + " "
+                time.sleep(0.05)
+                # Add a blinking cursor to simulate typing
+                message_placeholder.markdown(full_response + "▌")
+            message_placeholder.markdown(full_response)
+        # Add assistant response to chat history
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
+
+        # Reset the auto_submit flag
+        st.session_state.auto_submit = False
+        # Rerun to update the UI immediately - may not be needed
+        # st.rerun()
+
     # Accept user input
     prompt = st.chat_input("Type your claim here")
     if prompt: