Update app.py

app.py

@@ -101,29 +101,55 @@ def page1():
         image = Image.open('BinDoc Logo (Quadratisch).png')
         st.image(image, use_column_width='always')
 
+    # Initialize chat history in session state if not exists
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+
+    # Function to handle question submission
+    def submit_question(question):
+        # Add question to chat history
+        st.session_state.chat_history.append(("User", question))
+        process_question(question)
+
+    # Function to process the question
+    def process_question(question):
+        # Add message to the thread
+        st.session_state.messages = client.beta.threads.messages.create(
+            thread_id=st.session_state.thread.id,
+            role="user",
+            content=question
+        )
+
+        # Do a run to process the messages in the thread
+        st.session_state.run = client.beta.threads.runs.create(
+            thread_id=st.session_state.thread.id,
+            assistant_id=st.session_state.assistant.id,
+        )
 
-    # Initialize prompt in session state if not exists
-    if 'prompt' not in st.session_state:
-        st.session_state.prompt = ''
+        # Wait and rerun logic
+        if st.session_state.retry_error < 3:
+            time.sleep(1)  # Wait 1 second before checking run status
+            st.rerun()
 
     # Example query buttons
     col3, col4 = st.columns(2)  # Create two columns for buttons
 
     with col3:
         if st.button("Plotte mir die Anzahl der Kliniken von 2010 bis 2020."):
-            st.session_state.prompt = "Plotte mir die Anzahl der Kliniken von 2010 bis 2020."
+            submit_question("Plotte mir die Anzahl der Kliniken von 2010 bis 2020.")
         if st.button("Erstelle mir ein Bar Chart über die Bettenanzahl zwischen 1994 und 2002."):
-            st.session_state.prompt = "Erstelle mir ein Bar Chart über die Bettenanzahl zwischen 1994 und 2002."
+            submit_question("Erstelle mir ein Bar Chart über die Bettenanzahl zwischen 1994 und 2002.")
 
     with col4:
         if st.button("Hat die Bettenanzahl zwischen 2018 und 2021 zu- oder abgenommen?"):
-            st.session_state.prompt = "Hat die Bettenanzahl zwischen 2018 und 2021 zu- oder abgenommen?"
+            submit_question("Hat die Bettenanzahl zwischen 2018 und 2021 zu- oder abgenommen?")
         if st.button("Wie hat sich die Fremdkapitalquote des Malteser Krankenhaus St. Franziskus-Hospital verändert?"):
-            st.session_state.prompt = "Wie hat sich die Fremdkapitalquote des Malteser Krankenhaus St. Franziskus-Hospital verändert?"
+            submit_question("Wie hat sich die Fremdkapitalquote des Malteser Krankenhaus St. Franziskus-Hospital verändert?")
 
+    prompt = st.text_input("Wie kann ich dir helfen?")
 
-
-    prompt = st.text_input("Wie kann ich dir helfen?", value=st.session_state.prompt)
+    if st.button("Submit"):
+        submit_question(prompt)
 
     if "assistant" not in st.session_state:
         openai.api_key = api_key
@@ -154,22 +180,15 @@ def page1():
                         if image:
                             st.image(image, caption="Generated Image")
 
-    if st.button("Submit"):
-        # Add message to the thread
-        st.session_state.messages = client.beta.threads.messages.create(
-            thread_id=st.session_state.thread.id,
-            role="user",
-            content=st.session_state.prompt  # Use the prompt from session state
-        )
+            # Add assistant's response to chat history
+            st.session_state.chat_history.append(("Assistant", last_assistant_message.content[0].text.value))
 
-        # Do a run to process the messages in the thread
-        st.session_state.run = client.beta.threads.runs.create(
-            thread_id=st.session_state.thread.id,
-            assistant_id=st.session_state.assistant.id,
-        )
-        if st.session_state.retry_error < 3:
-            time.sleep(1)  # Wait 1 second before checking run status
-            st.rerun()
+    # Display chat history
+    for role, message in st.session_state.chat_history:
+        if role == "User":
+            st.text(f"User: {message}")
+        elif role == "Assistant":
+            st.text(f"Assistant: {message}")
 
     # Check if 'run' object has 'status' attribute
     if hasattr(st.session_state.run, 'status'):
@@ -190,8 +209,7 @@ def page1():
                     time.sleep(3)  # Longer delay before retrying
                     st.rerun()
                 else:
-                    st.error("FAILED: The OpenAI API is currently processing too many requests. Please try again later")
-
+                    st.error("FAILED: The OpenAI API is currently processing too many requests. Please try again later ......")
 
         # Handle any status that is not 'completed'
         elif st.session_state.run.status != "completed":
@@ -204,5 +222,6 @@ def page1():
                 time.sleep(3)
                 st.rerun()
 
+
 if __name__ == "__main__":
     main()