Update app.py

app.py

@@ -298,10 +298,11 @@ def streamlit_app(db):
 
                     final_items.extend(items)
                 
-                db = create_vector_db(final_items)  # Renamed 'db' to avoid overwriting
+                db = create_vector_db(final_items)  # Create or update the database with final items
                 
                 st.write("Models have been processed and added to the database.")
-                
+    
+    # Cache the chat messages
     @st.cache_resource
     def get_messages(db):
         if "messages" not in st.session_state:
@@ -310,20 +311,26 @@ def streamlit_app(db):
 
     st.session_state.messages = get_messages(db)
 
+    # Display chat history
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
     
-    if prompt := st.chat_input(query_text):
+    # Chat input will act as the query input for the model
+    if prompt := st.chat_input("Ask a question about the models:"):
+        # Add user input to chat
         st.chat_message("user").markdown(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
-        response = generate_response(db, query_text, st.session_state)
-
+        
+        # Generate the response from the model
+        response = generate_response(db, prompt, st.session_state.messages)  # Pass the prompt to generate_response
+        
+        # Display assistant response
         with st.chat_message("assistant"):
             st.markdown(response)
 
+        # Add the assistant response to the chat history
         st.session_state.messages.append({"role": "assistant", "content": response})
-
-
+        
 if __name__ == "__main__":
     streamlit_app(db)