Update app.py

app.py

@@ -47,21 +47,12 @@ cloud_button_style = """
 st.markdown(cloud_button_style, unsafe_allow_html=True)
 
 def cloud_button(label, key):
-    # Create a placeholder for the button
-    button_placeholder = st.empty()
-    
-    # Check if the button was previously clicked
-    was_clicked = button_placeholder.button(label, key=key)
-    
-    # If clicked, update the session state
-    if was_clicked:
+    button_html = f'<button type="button" class="cloud-button">{label}</button>'
+    button_click = st.button(label, key=key, use_container_width=True)
+    if button_click:
         st.session_state[key] = not st.session_state.get(key, False)
-        
     return st.session_state.get(key, False)
 
-
-
-
         
 # Step 1: Clone the Dataset Repository
 repo = Repository(
@@ -166,12 +157,18 @@ def main():
     if pdf_path is not None:
         query = st.text_input("Ask questions about your PDF file (in any preferred language):")
 
-    # Use the cloud_button function for the custom styled buttons
         if cloud_button("Was genau ist ein Belegarzt?", "belegarzt_button"):
             query = "Was genau ist ein Belegarzt?"
+            st.session_state["belegarzt_button"] = False  # Reset the button state
 
         if cloud_button("Wofür wird die Alpha-ID verwendet?", "alpha_id_button"):
             query = "Wofür wird die Alpha-ID verwendet?"
+            st.session_state["alpha_id_button"] = False  # Reset the button state
+
+        if st.button("Was sind die Vorteile des ambulanten operierens?"):
+            query = "Was sind die Vorteile des ambulanten operierens?"
+    
+
 
     # For default styled buttons, use st.button
         if st.button("Was sind die Vorteile des ambulanten operierens?"):