Update app.py

app.py

@@ -12,50 +12,6 @@ from langchain.chains.question_answering import load_qa_chain
 from langchain.callbacks import get_openai_callback
 import os
 
-def cloud_button(label, key):
-    button_id = f"button_{key}"
-    button_html = f"""
-        <style>
-            #{button_id} {{
-                display: inline-flex;
-                align-items: center;
-                justify-content: center;
-                padding: 8px 15px;
-                font-size: 14px;
-                cursor: pointer;
-                text-align: center;
-                text-decoration: none;
-                outline: none;
-                color: #fff;
-                background-color: #add8e6;
-                border: none;
-                border-radius: 20px;
-                box-shadow: 0 5px 10px rgba(0,0,0,0.1);
-                margin: 5px;
-            }}
-            #{button_id}:hover {{
-                background-color: #9fced1;
-            }}
-            #{button_id}:active {{
-                background-color: #87b6a7;
-                box-shadow: 0 2px 5px rgba(0,0,0,0.1);
-                transform: translateY(2px);
-            }}
-        </style>
-        <button id="{button_id}" class="cloud-button">{label}</button>
-        <script>
-            document.getElementById("{button_id}").onclick = function() {{
-                window.parent.postMessage({{
-                    'setSessionState': {{ '{key}': true }}
-                }}, "*");
-            }};
-        </script>
-    """
-    
-    st.markdown(button_html, unsafe_allow_html=True)
-    return st.session_state.get(key, False)
-
-        
 # Step 1: Clone the Dataset Repository
 repo = Repository(
     local_dir="Private_Book",  # Local directory to clone the repository
@@ -76,9 +32,7 @@ with st.sidebar:
     
     add_vertical_space(1)  # Adjust as per the desired spacing
     
-    st.markdown(""")
-
-
+    st.markdown("""
     Hello! I’m here to assist you with:<br><br>
     📘 **Glossary Inquiries:**<br>
     I can clarify terms like "DiGA", "AOP", or "BfArM", providing clear and concise explanations to help you understand our content better.<br><br>
@@ -128,57 +82,70 @@ 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! 🤖")
-
-    # Check if PDF file exists
+    
+    # Directly specifying the path to the PDF file
     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)
 
     new_messages_placeholder = st.empty()
 
+    # Add cloud-like button styles
+    cloud_button_style = """
+        <style>
+            .stButton > button {
+                border: none;
+                padding: 10px 20px;
+                border-radius: 25px;
+                font-size: 16px;
+                transition-duration: 0.4s;
+                cursor: pointer;
+                background-color: white; 
+                color: black; 
+                border: 2px solid #008CBA;
+            }
+            .stButton > button:hover {
+                background-color: #008CBA;
+                color: white;
+            }
+        </style>
+    """
+    st.markdown(cloud_button_style, unsafe_allow_html=True)
+
     if pdf_path is not None:
-        # User input for questions
         query = st.text_input("Ask questions about your PDF file (in any preferred language):")
 
-        # Buttons for predefined queries
-        if st.button("Was genau ist ein Belegarzt?", key="button1"):
-            st.session_state['belegarzt_button'] = True
-        if st.button("Wofür wird die Alpha-ID verwendet?", key="button2"):
-            query = "Wofür wird die Alpha-ID verwendet?"
-        if st.button("Was sind die Vorteile des ambulanten operierens?", key="button3"):
-            query = "Was sind die Vorteile des ambulanten operierens?"
-
-        if st.session_state.get('belegarzt_button'):
+        if st.button("Was genau ist ein Belegarzt?"):
             query = "Was genau ist ein Belegarzt?"
-            st.session_state['belegarzt_button'] = False  # Reset the button state
-
+        if st.button("Wofür wird die Alpha-ID verwendet?"):
+            query = "Wofür wird die Alpha-ID verwendet?"
+    
+            
+        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]):
+            st.session_state['chat_history'].append(("User", query, "new"))
 
-        # Processing the query
-        if query:
             loading_message = st.empty()
             loading_message.text('Bot is thinking...')
 
@@ -204,8 +171,9 @@ 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):