Update app.py

app.py

@@ -92,6 +92,11 @@ def display_chat_history(chat_history):
         background_color = "#ffeecf" if chat[2] == "new" else "#ffeecf" if chat[0] == "User" else "#ffeecf"
         st.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
 
+def handle_no_answer(response):
+    if response.strip().lower() in ["i don't know", "i am not sure", ""]:
+        # Return a more engaging response
+        return "Hmm, that's a tough one. I might not have all the answers, but let's try exploring this together. Can you provide more details or ask a different question?"
+    return response
 
 
 
@@ -165,11 +170,12 @@ def page1():
                 start_time = time.time()
                 
                 with st.spinner('Bot is thinking...'):
-                    # Use the VectorStore loaded at the start from the session state
                     chain = load_chatbot()
                     docs = VectorStore.similarity_search(query=query, k=3)
                     with get_openai_callback() as cb:
                         response = chain.run(input_documents=docs, question=query)
+                        response = handle_no_answer(response)  # Process the response through the new function
+
 
                         
                 # Stop timing
@@ -277,11 +283,12 @@ def page2():
                 start_time = time.time()
                 
                 with st.spinner('Bot is thinking...'):
-                    # Use the VectorStore loaded at the start from the session state
                     chain = load_chatbot()
                     docs = VectorStore.similarity_search(query=query, k=3)
                     with get_openai_callback() as cb:
                         response = chain.run(input_documents=docs, question=query)
+                        response = handle_no_answer(response)  # Process the response through the new function
+
 
                         
                 # Stop timing