From 90838792e10cf80689e91a50b32f628e0d308cbf Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 10 Apr 2024 00:08:14 +0200
Subject: Refactor ui

---
 rag/ui.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

(limited to 'rag')

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)
 
-- 
cgit v1.2.3-70-g09d2