Update app.py

app.py

@@ -261,7 +261,7 @@ def generate_response(db, query_text, previous_context):
     
     return full_response
 
-def streamlit_app(db):
+def streamlit_app():
     st.title("BioModelsRAG")
     
     search_str = st.text_input("Enter search query:")
@@ -298,7 +298,7 @@ def streamlit_app(db):
     
     # Cache the chat messages
     @st.cache_resource
-    def get_messages(_db):
+    def get_messages():
         if "messages" not in st.session_state:
             st.session_state.messages = []
         return st.session_state.messages
@@ -327,4 +327,4 @@ def streamlit_app(db):
         st.session_state.messages.append({"role": "assistant", "content": response})
         
 if __name__ == "__main__":
-    streamlit_app(db)
+    streamlit_app()