Update app.py

app.py

@@ -41,7 +41,14 @@ def store_embeddings(embeddings, metadata):
 def retrieve_relevant_chunks(query, k=5):
     query_embedding = embedding_model.encode([query])
     distances, indices = faiss_index.search(query_embedding, k)
-    return [(metadata_store[i], distances[i]) for i in indices[0]]
+    
+    # Safeguard: Ensure indices are within bounds of metadata_store
+    valid_results = [
+        (metadata_store[i], distances[0][j])
+        for j, i in enumerate(indices[0])
+        if i < len(metadata_store)
+    ]
+    return valid_results
 
 def ask_groq_api(question, context):
     chat_completion = client.chat.completions.create(
@@ -79,9 +86,12 @@ if uploaded_files:
     user_question = st.text_input("Ask a question about the uploaded papers:")
     if user_question:
         relevant_chunks = retrieve_relevant_chunks(user_question)
-        context = "\n\n".join([chunk['chunk'] for chunk, _ in relevant_chunks])
-        answer = ask_groq_api(user_question, context)
-        st.write("**Answer:**", answer)
+        if relevant_chunks:
+            context = "\n\n".join([chunk['chunk'] for chunk, _ in relevant_chunks])
+            answer = ask_groq_api(user_question, context)
+            st.write("**Answer:**", answer)
+        else:
+            st.write("No relevant sections found for your question.")
 
     if st.button("Generate Scatter Plot"):
         st.write("Generating scatter plot for methods vs. results...")
@@ -94,4 +104,3 @@ if uploaded_files:
         st.pyplot(plt)
 
     st.text_area("Annotate Your Insights:", height=100, key="annotations")
-