Update app.py

app.py

@@ -109,13 +109,16 @@ def main():
 
         with st.spinner("Fetching response..."):
             try:
-                # Retrieve context from the vector index
                 context_results = vector_index.similarity_search(user_input, k=5)
-                context = "\n".join([f"Title: {doc['Title']}\nAbstract: {doc['Abstract']}\nPublication Date: {doc['Publication Date']}" 
+
+                if not context_results:
+                    st.warning("No relevant context found. Please refine your query.")
+                    response = "I'm sorry, I couldn't find any relevant information to answer your question."
+                else:
+                    context = "\n".join([f"Title: {doc['Title']}\nAbstract: {doc['Abstract']}\nPublication Date: {doc['Publication Date']}" 
                                      for doc in extract_data(context_results)])
+                    response = query_from_mistral(context.strip(), user_input)
 
-                # Get response from Mistral
-                response = query_from_mistral(context.strip(), user_input)
                 st.session_state.messages.append({"role": "bot", "content": response})
             except Exception as e:
                 st.error(f"Error: {e}")
@@ -128,4 +131,4 @@ def main():
             st.markdown(f"**Bot:** {message['content']}")
 
 if __name__ == "__main__":
-    main()
+    main()