Update app.py

app.py

@@ -92,9 +92,10 @@ def cloud_button(label, key=None):
         }};
     </script>
     """
-    return cloud_button_html)
+    return cloud_button_html))
 
 def spaced_cloud_button(label):
+    button_id = f"cloud-button-{label}"
     cloud_button_html = f"""
     <div style="padding: 10px 10px 10px 0;">
     """
@@ -103,6 +104,10 @@ def spaced_cloud_button(label):
     </div>
     """
     st.markdown(cloud_button_html, unsafe_allow_html=True)
+    if st.session_state.get(button_id, False):
+        return True
+    return False
+
     
 
 
@@ -190,18 +195,14 @@ def main():
     if pdf_path is not None:
         query = st.text_input("Ask questions about your PDF file (in any preferred language):")
 
-        cloud_button_html = cloud_button("Was genau ist ein Belegarzt?"):
+
+        if spaced_cloud_button("Was genau ist ein Belegarzt?"):
             query = "Was genau ist ein Belegarzt?"
-        if spaced_cloud_button("Wofür wird die Alpha-ID verwendet?"):
-            query = "Wofür wird die Alpha-ID verwendet?"
-        if spaced_cloud_button("Was sind die Vorteile des ambulanten operierens?"):
-            query = "Was sind die Vorteile des ambulanten operierens?"
-        if spaced_cloud_button("Was kann ich mit dem Prognose-Analyse Toll machen?"):
-            query = "Was kann ich mit dem Prognose-Analyse Toll machen?"
-        if spaced_cloud_button("Was sagt mir die Farbe der Balken der Bevölkerungsentwicklung?"):
-            query = "Was sagt mir die Farbe der Balken der Bevölkerungsentwicklung?"
-        if spaced_cloud_button("Ich habe mein Meta Password vergessen, wie kann ich es zurücksetzen?"):
-            query = "Ich habe mein Meta Password vergessen, wie kann ich es zurücksetzen?"
+
+        if spaced_cloud_button("Test, test test ?"):
+            query = "Was genau ist ein Test test test ?"    
+
+        
 
         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"))