Update app.py

app.py

@@ -124,30 +124,32 @@ def load_chatbot():
     return load_qa_chain(llm=OpenAI(), chain_type="stuff")
 
 def main():
-
+    # Hide Streamlit's menu and footer
     hide_streamlit_style = """
-            <style>
-            #MainMenu {visibility: hidden;}
-            footer {visibility: hidden;}
-            </style>
-            """
+        <style>
+        #MainMenu {visibility: hidden;}
+        footer {visibility: hidden;}
+        </style>
+    """
     st.markdown(hide_streamlit_style, unsafe_allow_html=True)
 
-
     # Main content
     st.title("Welcome to BinDocs ChatBot! 🤖")
-    
-    # Directly specifying the path to the PDF file
+
+    # Check if PDF file exists
     pdf_path = pdf_file_path
     if not os.path.exists(pdf_path):
         st.error("File not found. Please check the file path.")
         return
 
+    # Initialize chat history in session state
     if "chat_history" not in st.session_state:
         st.session_state['chat_history'] = []
 
+    # Display previous chat messages
     display_chat_history(st.session_state['chat_history'])
 
+    # Spacer
     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)
@@ -155,9 +157,11 @@ def main():
     new_messages_placeholder = st.empty()
 
     if pdf_path is not None:
+        # User input for questions
         query = st.text_input("Ask questions about your PDF file (in any preferred language):")
 
-       if st.button("Was genau ist ein Belegarzt?", key="button1"):
+        # Buttons for predefined queries
+        if st.button("Was genau ist ein Belegarzt?", key="button1"):
             query = "Was genau ist ein Belegarzt?"
 
         if st.button("Wofür wird die Alpha-ID verwendet?", key="button2"):
@@ -166,12 +170,8 @@ def main():
         if st.button("Was sind die Vorteile des ambulanten operierens?", key="button3"):
             query = "Was sind die Vorteile des ambulanten operierens?"
 
-
-
-    # For default styled buttons, use st.button
-        if st.button("Was sind die Vorteile des ambulanten operierens?"):
-            query = "Was sind die Vorteile des ambulanten operierens?"
-
+        # Processing the query
+        if query:
             loading_message = st.empty()
             loading_message.text('Bot is thinking...')
 
@@ -197,9 +197,8 @@ def main():
             # Clear the input field by setting the query variable to an empty string
             query = ""
 
-        # Mark all messages as old after displaying
-        st.session_state['chat_history'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history']]
-
+    # Mark all messages as old after displaying
+    st.session_state['chat_history'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history']]
 
 
 def display_chat_history(chat_history):