Update app.py

app.py

@@ -527,11 +527,6 @@ def page3():
             st.error("File not found. Please check the file path.")
             return
 
-        display_chat_history(st.session_state['chat_history_page3'])
-        new_messages_placeholder = st.empty()
-        query = st.text_input("Geben Sie hier Ihre Frage ein / Enter your question here:")
-
-        
         # Initialize CromA client
         chroma_client = chromadb.Client()
 
@@ -554,6 +549,38 @@ def page3():
             )
             st.session_state["documents_added"] = True
 
+
+        display_chat_history(st.session_state['chat_history_page3'])
+        new_messages_placeholder = st.empty()
+        query = st.text_input("Geben Sie hier Ihre Frage ein / Enter your question here:")
+
+        # Handling query input
+        if query:
+            full_query = ask_bot(query)
+            st.session_state['chat_history_page3'].append(("User", query, "new"))
+        
+            # Start timing for response
+            start_time = time.time()
+        
+            # Querying the CromA collection
+            results = collection.query(
+                query_texts=[full_query],
+                n_results=5  # Adjust the number of results as needed
+            )
+                    
+            # Calculate the response duration
+            end_time = time.time()
+            duration = end_time - start_time
+    
+            # Process and display response from CromA results
+            if results:
+                # TODO: Adjust the following logic based on CromA's actual result structure
+                response = f"Top result: {results[0]['text']}"  # Example response using the first result
+            else:
+                response = "No results found for your query."
+    
+            st.session_state['chat_history_page3'].append(("Eve", response, "new"))
+
         # Simple interaction
         if st.button("Test Button"):
             st.write("Button clicked.")