Update app.py

app.py

@@ -291,37 +291,40 @@ def streamlit_app(db):
                     convert_sbml_to_antimony(model_file_path, antimony_file_path)
                     
                     items = split_biomodels(antimony_file_path)
-                    if not final_items:
+                    if not items:  # Check if 'items' is empty, not 'final_items'
                         st.write("No content found in the biomodel.")
                         continue
-                    
+
                     final_items.extend(items)
                 
-    db = create_vector_db(final_items)
+                vector_db = create_vector_db(final_items)  # Renamed 'db' to avoid overwriting
+                
+                st.write("Models have been processed and added to the database.")
                 
-    st.write("Models have been processed and added to the database.")
-                    
-
     @st.cache_resource
     def get_messages(db):
         if "messages" not in st.session_state:
             st.session_state.messages = []
         return st.session_state.messages
-    st.session_state.messages = get_messages()
+
+    st.session_state.messages = get_messages(db)
 
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+    query_text = st.text_input("Enter your query:")  # Initialize 'query_text'
+    
     if prompt := st.chat_input(query_text):
         st.chat_message("user").markdown(prompt)
-        st.session_state.messages.append({"role": "user", "content":prompt})
+        st.session_state.messages.append({"role": "user", "content": prompt})
         response = generate_response(db, query_text, st.session_state)
 
         with st.chat_message("assistant"):
             st.markdown(response)
 
-        st.session_state.messages.append({"role":"assistant","content":response})
+        st.session_state.messages.append({"role": "assistant", "content": response})
+
 
 if __name__ == "__main__":
     streamlit_app(db)