Refactor ui

1 files changed, 1 insertions, 5 deletions

diff --git a/rag/ui.py b/rag/ui.py
index 5a5e01a..82c6d47 100644
--- a/rag/ui.py
+++ b/rag/ui.py
@@ -60,11 +60,7 @@ if __name__ == "__main__":
 
     (result_column, context_column) = st.columns(2)
 
-    if submit:
-        if not query:
-            st.stop()
-
-        query = ss.get("query", "")
+    if submit and query:
         with st.spinner("Searching for documents..."):
             documents = retriever.retrieve(query)