CAntoniadis commited on
Commit
258ff84
·
verified ·
1 Parent(s): b58a351

Update web_gui/streamlit.py

Browse files
Files changed (1) hide show
  1. web_gui/streamlit.py +5 -9
web_gui/streamlit.py CHANGED
@@ -529,14 +529,10 @@ if mode == "RAG Chat":
529
 
530
  ###### Added Search engine logic
531
  if mode == "Search engine":
532
-
533
- # 🔥 FORCE RESET OLD ENGINE (IMPORTANT)
534
- if "tfidf_engine" in st.session_state:
535
- del st.session_state["tfidf_engine"]
536
-
537
- engine = SimpleSearchEngine(corpus_dir="corpus")
538
- engine.build_index()
539
- st.session_state["tfidf_engine"] = engine
540
 
541
  st.title("Search Engine")
542
 
@@ -557,4 +553,4 @@ if mode == "Search engine":
557
  if r.get("url"):
558
  st.markdown(r["url"])
559
 
560
- st.divider()
 
529
 
530
  ###### Added Search engine logic
531
  if mode == "Search engine":
532
+ if "tfidf_engine" not in st.session_state:
533
+ engine = SimpleSearchEngine(corpus_dir="corpus")
534
+ engine.build_index()
535
+ st.session_state["tfidf_engine"] = engine
 
 
 
 
536
 
537
  st.title("Search Engine")
538
 
 
553
  if r.get("url"):
554
  st.markdown(r["url"])
555
 
556
+ st.divider()