Update app.py

app.py

@@ -261,7 +261,7 @@ def generate_response(db, query_text, previous_context):
     
     return full_response
 
-def streamlit_app():
+def streamlit_app(db):
     st.title("BioModelsRAG")
     
     search_str = st.text_input("Enter search query:")
@@ -304,7 +304,7 @@ def streamlit_app():
                     
     if db:
         @st.cache_resource
-        def get_messages():
+        def get_messages(db):
             if "messages" not in st.session_state:
                 st.session_state.messages = []
             return st.session_state.messages