Update app.py

app.py

@@ -354,7 +354,9 @@ def page2():
                     query = "Wie viele Patienten eines Grund- und Regelversorgers kommen aus welcher Fahrzeitzone?"
 
         
+
             if query:
+                full_query = ask_bot(query)
                 st.session_state['chat_history_page2'].append(("User", query, "new"))
 
                 # Start timing
@@ -364,10 +366,8 @@ def page2():
                     chain = load_chatbot()
                     docs = VectorStore.similarity_search(query=query, k=5)
                     with get_openai_callback() as cb:
-                        response = chain.run(input_documents=docs, question=query)
+                        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()
@@ -466,7 +466,9 @@ def page3():
                     query = "Wie hoch sind die Gesamtkosten der Krankenhäuser pro Jahr?"
 
         
+
             if query:
+                full_query = ask_bot(query)
                 st.session_state['chat_history_page3'].append(("User", query, "new"))
 
                 # Start timing
@@ -476,10 +478,8 @@ def page3():
                     chain = load_chatbot()
                     docs = VectorStore.similarity_search(query=query, k=5)
                     with get_openai_callback() as cb:
-                        response = chain.run(input_documents=docs, question=query)
+                        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()