Update app.py

app.py

@@ -525,17 +525,18 @@ def page3():
             
             # Query the CromA collection
             st.text("Querying CromA collection...")  # Diagnostic print
+            # Query the CromA collection
             results = collection.query(
                 query_texts=[full_query],
                 n_results=5
             )
-
-            st.text(f"Query results: {results}")  # Diagnostic print
-
+    
             # Process and display response from CromA results
-            if results:
+            if results and results['documents']:
                 try:
-                    response = f"Top result: {results[0]['text']}"
+                    # Accessing the first document of the first result
+                    top_document = results['documents'][0][0]  # Adjusted access
+                    response = f"Top result: {top_document}"
                 except KeyError as ke:
                     st.error(f"KeyError encountered: {ke}")
                     response = "Error in processing the response."