Update app.py

app.py

@@ -254,99 +254,97 @@ def page1():
             st.image(image, use_column_width='always')
 
  
-        # Start tracking user interactions
-        with streamlit_analytics.track():
-            if not os.path.exists(pdf_path):
-                st.error("File not found. Please check the file path.")
-                return
-    
-            VectorStore = load_vector_store(pdf_path, "KH_Reform_2301", force_reload=False)
+        if not os.path.exists(pdf_path):
+            st.error("File not found. Please check the file path.")
+            return
+
+        VectorStore = load_vector_store(pdf_path, "KH_Reform_2301", force_reload=False)
+
+        display_chat_history(st.session_state['chat_history_page1'])
+
+        st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
+        st.write("<div style='flex: 1;'></div>", unsafe_allow_html=True)
+        st.write("<!-- End Spacer -->", unsafe_allow_html=True)
+
+        new_messages_placeholder = st.empty()
+
+        query = st.text_input("Geben Sie hier Ihre Frage ein / Enter your question here:")
+
+        add_vertical_space(2)  # Adjust as per the desired spacing
+        
+        # Create two columns for the buttons
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            if st.button("Wie viele Ärzte benötigt eine Klinik in der Leistungsgruppe Stammzell-transplantation?"):
+                query = "Wie viele Ärzte benötigt eine Klinik in der Leistungsgruppe Stammzell-transplantation?"
+            if st.button("Wie viele Leistungsgruppen gibt es?"):
+                query = ("Wie viele Leistungsgruppen gibt es?")
+            if st.button("Was sind die hauptsächlichen Änderungsvorhaben der Krankenhausreform?"):
+                query = "Was sind die hauptsächlichen Änderungsvorhaben der Krankenhausreform?"
+
+        
+        with col2:
+            if st.button("Welche und wieviele Fachärzte benötige ich für die Leistungsgruppe Pädiatrie? "):
+                query = "Welche und wieviele Fachärzte benötige ich für die Leistungsgruppe Pädiatrie"
+            if st.button("Was soll die Reform der Notfallversorgung beinhalten?"):
+                query = "Was soll die Reform der Notfallversorgung beinhalten?"
+            if st.button("Was bedeutet die Vorhaltefinanzierung?"):
+                query = "Was bedeutet die Vorhaltefinanzierung?"
+
 
-            display_chat_history(st.session_state['chat_history_page1'])
-    
-            st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
-            st.write("<div style='flex: 1;'></div>", unsafe_allow_html=True)
-            st.write("<!-- End Spacer -->", unsafe_allow_html=True)
-    
-            new_messages_placeholder = st.empty()
     
-            query = st.text_input("Geben Sie hier Ihre Frage ein / Enter your question here:")
+        if query:
+            full_query = ask_bot(query)
+            st.session_state['chat_history_page1'].append(("User", query, "new"))
 
-            add_vertical_space(2)  # Adjust as per the desired spacing
+            # Start timing
+            start_time = time.time()
             
-            # Create two columns for the buttons
-            col1, col2 = st.columns(2)
-            
-            with col1:
-                if st.button("Wie viele Ärzte benötigt eine Klinik in der Leistungsgruppe Stammzell-transplantation?"):
-                    query = "Wie viele Ärzte benötigt eine Klinik in der Leistungsgruppe Stammzell-transplantation?"
-                if st.button("Wie viele Leistungsgruppen gibt es?"):
-                    query = ("Wie viele Leistungsgruppen gibt es?")
-                if st.button("Was sind die hauptsächlichen Änderungsvorhaben der Krankenhausreform?"):
-                    query = "Was sind die hauptsächlichen Änderungsvorhaben der Krankenhausreform?"
+            with st.spinner('Bot is thinking...'):
+                chain = load_chatbot()
+                docs = VectorStore.similarity_search(query=query, k=5)
+                with get_openai_callback() as cb:
+                    response = chain.run(input_documents=docs, question=full_query)
+                    response = handle_no_answer(response)  # Process the response through the new function
+
 
+                    
+            # Stop timing
+            end_time = time.time()
             
-            with col2:
-                if st.button("Welche und wieviele Fachärzte benötige ich für die Leistungsgruppe Pädiatrie? "):
-                    query = "Welche und wieviele Fachärzte benötige ich für die Leistungsgruppe Pädiatrie"
-                if st.button("Was soll die Reform der Notfallversorgung beinhalten?"):
-                    query = "Was soll die Reform der Notfallversorgung beinhalten?"
-                if st.button("Was bedeutet die Vorhaltefinanzierung?"):
-                    query = "Was bedeutet die Vorhaltefinanzierung?"
+            # Calculate duration
+            duration = end_time - start_time
 
+            # You can use Streamlit's text function to display the timing
+            st.text(f"Response time: {duration:.2f} seconds")
 
-        
-            if query:
-                full_query = ask_bot(query)
-                st.session_state['chat_history_page1'].append(("User", query, "new"))
+            # Display the current working directory after save_conversation
+            current_dir = os.getcwd()
+            st.text(f"Current working directory before save_conversation: {current_dir}")
 
-                # Start timing
-                start_time = time.time()
-                
-                with st.spinner('Bot is thinking...'):
-                    chain = load_chatbot()
-                    docs = VectorStore.similarity_search(query=query, k=5)
-                    with get_openai_callback() as cb:
-                        response = chain.run(input_documents=docs, question=full_query)
-                        response = handle_no_answer(response)  # Process the response through the new function
+            st.session_state['chat_history_page1'].append(("Bot", response, "new"))
 
 
-                        
-                # Stop timing
-                end_time = time.time()
-                
-                # Calculate duration
-                duration = end_time - start_time
+            # Display new messages at the bottom
+            new_messages = st.session_state['chat_history_page1'][-2:]
+            for chat in new_messages:
+                background_color = "#ffeecf" if chat[2] == "new" else "#ffeecf" if chat[0] == "User" else "#ffeecf"
+                new_messages_placeholder.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
 
-                # You can use Streamlit's text function to display the timing
-                st.text(f"Response time: {duration:.2f} seconds")
+            # Save conversation after chat interaction
+            save_conversation(st.session_state['chat_history_page1'], st.session_state['session_id'], 1)
 
-                # Display the current working directory after save_conversation
-                current_dir = os.getcwd()
-                st.text(f"Current working directory before save_conversation: {current_dir}")
-    
-                st.session_state['chat_history_page1'].append(("Bot", response, "new"))
-    
-    
-                # Display new messages at the bottom
-                new_messages = st.session_state['chat_history_page1'][-2:]
-                for chat in new_messages:
-                    background_color = "#ffeecf" if chat[2] == "new" else "#ffeecf" if chat[0] == "User" else "#ffeecf"
-                    new_messages_placeholder.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
-    
-                # Save conversation after chat interaction
-                save_conversation(st.session_state['chat_history_page1'], st.session_state['session_id'], 1)
+            
+            # Display the current working directory after save_conversation
+            current_dir = os.getcwd()
+            st.text(f"Current working directory after save_conversation: {current_dir}")
+            
+            # Clear the input field after the query is made
+            query = ""
 
-                
-                # Display the current working directory after save_conversation
-                current_dir = os.getcwd()
-                st.text(f"Current working directory after save_conversation: {current_dir}")
-                
-                # Clear the input field after the query is made
-                query = ""
-    
-            # Mark all messages as old after displaying
-            st.session_state['chat_history_page1'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history_page1']]
+        # Mark all messages as old after displaying
+        st.session_state['chat_history_page1'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history_page1']]
 
     except Exception as e:
         st.error(f"Upsi, an unexpected error occurred: {e}")