fix: st.write

app.py

@@ -102,17 +102,19 @@ st.write("This app searches through scientific papers to answer your questions a
 query = st.text_input("Please ask me anything about autism ✨")
 
 if query:
-    with st.status("Searching for answers..."):
+    with st.status("Searching for answers...") as status:
         # Load dataset
         df = load_dataset(query)
+        st.write("Searching for data in PubMed and arXiv...")
         # Get relevant context
         context = "\n".join([
             f"{text[:1000]}" for text in df['text'].head(3)
         ])
-        
+        st.write("Data found!")
         # Generate answer
         answer = generate_answer(query, context)
-        st.status.update(
+        st.write("Generating answer...")
+        status.update(
             label="Search complete!", state="complete", expanded=False
         )
     if answer and not answer.isspace():