Update app.py

app.py

@@ -69,7 +69,7 @@ pdf_path3 = "Private_Book/Kosten_Strukturdaten_RAG_vorbereited.pdf"
 api_key = os.getenv("OPENAI_API_KEY")
 # Retrieve the API key from st.secrets
 
-@st.cache
+@st.cache_data
 def extract_text_from_pdf(pdf_path):
     text = ""
     reader = PdfReader(pdf_path)
@@ -519,25 +519,32 @@ def page3():
         # User query input
         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"))
             
             # Query the CromA collection
+            st.text("Querying CromA collection...")  # Diagnostic print
             results = collection.query(
                 query_texts=[full_query],
-                n_results=5  # Adjust the number of results as needed
+                n_results=5
             )
 
+            st.text(f"Query results: {results}")  # Diagnostic print
+
             # Process and display response from CromA results
             if results:
-                response = f"Top result: {results[0]['text']}"  # Example response using the first result
+                try:
+                    response = f"Top result: {results[0]['text']}"
+                except KeyError as ke:
+                    st.error(f"KeyError encountered: {ke}")
+                    response = "Error in processing the response."
             else:
                 response = "No results found for your query."
             
             st.session_state['chat_history_page3'].append(("Eve", response, "new"))
 
+
         # Display new messages at the bottom
         new_messages = st.session_state['chat_history_page3'][-2:]
         for chat in new_messages: