Update app.py

app.py

@@ -192,16 +192,21 @@ def main():
     if pdf_path is not None:
         query = st.text_input("Ask questions about your PDF file (in any preferred language):")
 
-        cloud_button("Was genau ist ein Belegarzt?", "Was genau ist ein Belegarzt?", key="button1", color="1")
-        cloud_button("Wofür wird die Alpha-ID verwendet?", "Wofür wird die Alpha-ID verwendet?", key="button2", color="2")
-        cloud_button("Was sind die Vorteile des ambulanten operierens?", "Was sind die Vorteile des ambulanten operierens?", key="button3", color="3")
-        cloud_button("Was kann ich mit dem Prognose-Analyse Toll machen?", "Was kann ich mit dem Prognose-Analyse Toll machen?", key="button4", color="4")
-        cloud_button("Was sagt mir die Farbe der Balken der Bevölkerungsentwicklung?", "Was sagt mir die Farbe der Balken der Bevölkerungsentwicklung?", key="button5", color="5")
-        cloud_button("Ich habe mein Meta Password vergessen, wie kann ich es zurücksetzen?", "Ich habe mein Meta Password vergessen, wie kann ich es zurücksetzen?", key="button6", color="6")
-
+        # Define your cloud buttons with associated queries
+        cloud_buttons = [
+            ("Was genau ist ein Belegarzt?", "Was genau ist ein Belegarzt?", "1"),
+            ("Wofür wird die Alpha-ID verwendet?", "Wofür wird die Alpha-ID verwendet?", "2"),
+            # Add more buttons as needed
+        ]
+
+        # Create cloud buttons
+        for label, query, color in cloud_buttons:
+            cloud_button(label, query, color=color)
+
+        # Check if a button was clicked
         if st.session_state['button_clicked']:
-            label, query = st.session_state['button_clicked'].values()
-            st.session_state['chat_history'].append(("User", label, "new"))
+            query = st.session_state['button_clicked']['query']
+            st.session_state['chat_history'].append(("User", query, "new"))
             st.session_state['button_clicked'] = None
             
         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]):
@@ -230,6 +235,7 @@ def main():
 
         st.session_state['chat_history'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history']]
 
+
 def display_chat_history(chat_history):
     for chat in chat_history:
         background_color = "#FFA07A" if chat[2] == "new" else "#acf" if chat[0] == "User" else "#caf"