Update app.py

app.py

@@ -45,15 +45,9 @@ st.markdown(f"Nmber of documents: `{len(docs)}`.")
 st.markdown(f"From `{oldest_date.strftime('%Y-%m-%d')}` to `{newest_date.strftime('%Y-%m-%d')}`.")
 st.markdown("This app uses [BM25](https://en.wikipedia.org/wiki/Okapi_BM25), allowing you to search not only with keywords like \"machine learning\" \n  but also with documents like \"How to generate synthetic data using LLM.\"")
 
-col1, col2 = st.columns([4, 1])
+prompt = st.chat_input("Search anything...")
 
-with col1:
-    user_query = st.text_input("", placeholder="Search anything...")
-
-with col2:
-    search_button = st.button('→')
-
-if search_button:
+if prompt:
     results = retriever.invoke(user_query)
     st.text(f"Top n {len(results)} related papers")