Update app.py

app.py

@@ -47,14 +47,21 @@ cloud_button_style = """
 st.markdown(cloud_button_style, unsafe_allow_html=True)
 
 def cloud_button(label, key):
-    button_html = f'<button type="button" class="cloud-button">{label}</button>'
-    button_click = st.markdown(button_html, unsafe_allow_html=True)
-    if button_click:
+    # 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:
         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(
@@ -159,19 +166,16 @@ def main():
     if pdf_path is not None:
         query = st.text_input("Ask questions about your PDF file (in any preferred language):")
 
-
-        if st.markdown('<button class="cloud-button">Was genau ist ein Belegarzt?</button>', unsafe_allow_html=True):
+    # 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?"
-            
-        if cloud_button("Wofür wird die Alpha-ID verwendet?", "alpha_id"):
+
+        if cloud_button("Wofür wird die Alpha-ID verwendet?", "alpha_id_button"):
             query = "Wofür wird die Alpha-ID verwendet?"
-            
+
+    # For default styled buttons, use st.button
         if st.button("Was sind die Vorteile des ambulanten operierens?"):
             query = "Was sind die Vorteile des ambulanten operierens?"
-        
-            
-        if st.button("Ask") or (not st.session_state['chat_history'] and query) or (st.session_state['chat_history'] and query != st.session_state['chat_history'][-1][1]):
-            st.session_state['chat_history'].append(("User", query, "new"))
 
             loading_message = st.empty()
             loading_message.text('Bot is thinking...')