Update app.py

app.py

@@ -88,51 +88,36 @@ def page1():
     st.sidebar.markdown("**Session-ID:**")
     st.sidebar.markdown(st.session_state.session_id)
     st.sidebar.divider()
-    st.title("Assitent-Name")
-    st.text("Beschreibung")
 
-    if "assistant" not in st.session_state:
-        openai.api_key = api_key
-        # Load the previously created assistant
-        st.session_state.assistant = openai.beta.assistants.retrieve(assistant_id)
-
-        # Create a new thread for this session
-        st.session_state.thread = client.beta.threads.create(
-            metadata={'session_id': st.session_state.session_id}
-        )
-
-    # If the run is completed, display the messages
-    elif hasattr(st.session_state.run, 'status') and st.session_state.run.status == "completed":
-        # Retrieve the list of messages
-        st.session_state.messages = client.beta.threads.messages.list(
-            thread_id=st.session_state.thread.id
-        )
+    # Create columns for layout
+    col1, col2 = st.columns([3, 1])  # Adjust the ratio to your liking
 
+    with col1:
+        st.title('Welcome to MediMetrics!')
+        st.subheader('powered by :orange[BinDoc GmbH] ')
 
-        # Display messages
-        for message in reversed(st.session_state.messages.data):
-            if message.role in ["user", "assistant"]:
-                with st.chat_message(message.role):
-                    for content_part in message.content:
-                        if content_part.type == 'text':  # For text responses
-                            message_text = content_part.text.value
-                            st.markdown(message_text)
-                        elif hasattr(content_part, 'image_file') and content_part.image_file:  
-                            image = handle_and_display_image(content_part.image_file.file_id)
-                            if image:
-                                st.image(image, caption="Generated Image")
+    with col2:
+        # Load and display the image in the right column
+        image = Image.open('BinDoc Logo (Quadratisch).png')
+        st.image(image, use_column_width='always')
 
-    
-
-    prompt = st.text_input("Wie kann ich dir helfen?")
-    if st.button("Submit"):
+    # 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=prompt
+            content=question
         )
 
         # Do a run to process the messages in the thread
@@ -140,10 +125,71 @@ def page1():
             thread_id=st.session_state.thread.id,
             assistant_id=st.session_state.assistant.id,
         )
+
+        # Wait and rerun logic
         if st.session_state.retry_error < 3:
-            time.sleep(1) # Wait 1 second before checking run status
+            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."):
+            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."):
+            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?"):
+            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?"):
+            submit_question("Wie hat sich die Fremdkapitalquote des Malteser Krankenhaus St. Franziskus-Hospital verändert?")
+
+    prompt = st.text_input("Wie kann ich dir helfen?")
+
+    if st.button("Submit"):
+        submit_question(prompt)
+
+    if "assistant" not in st.session_state:
+        openai.api_key = api_key
+        # Load the previously created assistant
+        st.session_state.assistant = openai.beta.assistants.retrieve(assistant_id)
+
+        # Create a new thread for this session
+        st.session_state.thread = client.beta.threads.create(
+            metadata={'session_id': st.session_state.session_id}
+        )
+
+    # Check if the run is completed and display the last assistant message
+    if hasattr(st.session_state.run, 'status') and st.session_state.run.status == "completed":
+        # Retrieve the list of messages
+        st.session_state.messages = client.beta.threads.messages.list(
+            thread_id=st.session_state.thread.id
+        )
+
+        # Find the last assistant message
+        last_assistant_message = next((message for message in reversed(st.session_state.messages.data) if message.role == "assistant"), None)
+        if last_assistant_message:
+            with st.chat_message('assistant'):
+                for content_part in last_assistant_message.content:
+                    if content_part.type == 'text':
+                        st.markdown(content_part.text.value)
+                    elif hasattr(content_part, 'image_file') and content_part.image_file:
+                        image = handle_and_display_image(content_part.image_file.file_id)
+                        if image:
+                            st.image(image, caption="Generated Image")
+
+            # Add assistant's response to chat history
+            st.session_state.chat_history.append(("Assistant", last_assistant_message.content[0].text.value))
+
+    # 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'):
         # Handle the 'running' status
@@ -176,5 +222,6 @@ def page1():
                 time.sleep(3)
                 st.rerun()
 
+
 if __name__ == "__main__":
-    main()
+    main()