MVPilgrim commited on
Commit
0c173aa
·
1 Parent(s): c1cc1e0
Files changed (1) hide show
  1. app.py +27 -5
app.py CHANGED
@@ -406,6 +406,21 @@ try:
406
  #collection = client.collections.get("Chunks")
407
  logger.info("#### getRagData() exited.")
408
  return ragData
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
409
 
410
 
411
  # Display UI
@@ -493,9 +508,9 @@ try:
493
  return fullPrompt
494
 
495
 
496
- def on_submitButton_clicked():
497
  logger = st.session_state.logger
498
- logger.info("### on_submitButton_clicked entered.")
499
  st.session_state.sysTAtext = st.session_state.sysTA
500
  logger.info(f"sysTAtext: {st.session_state.sysTAtext}")
501
 
@@ -509,12 +524,19 @@ try:
509
  st.session_state.rspTA = rsp
510
  logger.info(f"rspTAtext: {st.session_state.rspTA}")
511
 
512
- logger.info("### on_submitButton_clicked exited.")
513
 
 
 
 
 
 
 
514
 
515
  with st.sidebar:
516
- st.selectRag = st.checkbox("Enable Query With RAG",value=False,key="selectRag",help=None,on_change=None,args=None,kwargs=None,disabled=False,label_visibility="visible")
517
- st.submitButton = st.button("Run LLM Query",key=None,help=None,on_click=on_submitButton_clicked,args=None,kwargs=None,type="secondary",disabled=False,use_container_width=False)
 
518
 
519
  logger.info("#### semsearch.py end of code.")
520
  except Exception as e:
 
406
  #collection = client.collections.get("Chunks")
407
  logger.info("#### getRagData() exited.")
408
  return ragData
409
+
410
+
411
+ def getAllRagData():
412
+ logger.info("#### getAllRagData() entered.")
413
+
414
+ chunksCollection = client.collections.get("Chunks")
415
+ response = chunksCollection.query.fetch_objects()
416
+
417
+ wstrObjs = str(response.objects)
418
+ logger.info(f"### response.objects: {wstrObjs}")
419
+ for o in response.objects:
420
+ wstr = o.properties
421
+ logger.info(f"### o.properties: {wstr}")
422
+
423
+ return wstrObjs
424
 
425
 
426
  # Display UI
 
508
  return fullPrompt
509
 
510
 
511
+ def on_runLLMButton_Clicked():
512
  logger = st.session_state.logger
513
+ logger.info("### on_runLLMButton_Clicked entered.")
514
  st.session_state.sysTAtext = st.session_state.sysTA
515
  logger.info(f"sysTAtext: {st.session_state.sysTAtext}")
516
 
 
524
  st.session_state.rspTA = rsp
525
  logger.info(f"rspTAtext: {st.session_state.rspTA}")
526
 
527
+ logger.info("### on_runLLMButton_Clicked exited.")
528
 
529
+ def on_getAllRagDataButton_Clicked():
530
+ logger = st.session_state.logger
531
+ logger.info("### on_getAllRagButton_Clicked entered.")
532
+ st.session_state.ragpTA = getAllRagData()
533
+
534
+ logger.info("### on_getAllRagButton_Clicked entered.")
535
 
536
  with st.sidebar:
537
+ st.selectRag = st.checkbox("Enable Query With RAG",value=False,key="selectRag",help=None,on_change=None,args=None,kwargs=None,disabled=False,label_visibility="visible")
538
+ st.runLLMButton = st.button("Run LLM Query",key=None,help=None,on_click=on_runLLMButton_Clicked,args=None,kwargs=None,type="secondary",disabled=False,use_container_width=False)
539
+ st.getAllRagDataButton = st.button("Get All Rag Data",key=None,help=None,on_click=on_getAllRagDataButton_Clicked,args=None,kwargs=None,type="secondary",disabled=False,use_container_width=False)
540
 
541
  logger.info("#### semsearch.py end of code.")
542
  except Exception as e: