Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from PIL import Image
+import random
 import time
 import streamlit_analytics
 from dotenv import load_dotenv
@@ -87,13 +88,52 @@ def load_chatbot():
     return load_qa_chain(llm=OpenAI(model_name="gpt-3.5-turbo-instruct"), chain_type="stuff")
 
 
-
 def display_chat_history(chat_history):
     for chat in 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):
+    no_answer_phrases = [
+        "ich weiß es nicht",
+        "ich weiß nicht",
+        "ich bin mir nicht sicher",
+        "es wird nicht erwähnt",
+        "Leider kann ich diese Frage nicht beantworten",
+        "kann ich diese Frage nicht beantworten",
+        "ich kann diese Frage nicht beantworten",
+        "ich kann diese Frage leider nicht beantworten",
+        "keine information",
+        "das ist unklar",
+        "da habe ich keine antwort",
+        "das kann ich nicht beantworten",
+        "i don't know",
+        "i am not sure",
+        "it is not mentioned",
+        "no information",
+        "that is unclear",
+        "i have no answer",
+        "i cannot answer that",
+        "unable to provide an answer",
+        "not enough context",
+    ]
+
+    alternative_responses = [
+        "Hmm, das ist eine knifflige Frage. Lass uns das gemeinsam erkunden. Kannst du mehr Details geben?",
+        "Interessante Frage! Ich bin mir nicht sicher, aber wir können es herausfinden. Hast du weitere Informationen?",
+        "Das ist eine gute Frage. Ich habe momentan keine Antwort darauf, aber vielleicht kannst du sie anders formulieren?",
+        "Da bin ich überfragt. Kannst du die Frage anders stellen oder mir mehr Kontext geben?",
+        "Ich stehe hier etwas auf dem Schlauch. Gibt es noch andere Aspekte der Frage, die wir betrachten könnten?",
+        # Add more alternative responses as needed
+    ]
+
+    # Check if response matches any phrase in no_answer_phrases
+    if any(phrase in response.lower() for phrase in no_answer_phrases):
+        return random.choice(alternative_responses)  # Randomly select a response
+    return response
+
+
 
 
 def page1():
@@ -166,11 +206,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
@@ -278,11 +319,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
@@ -336,4 +378,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()