next test

README.md

@@ -4,7 +4,7 @@ emoji: 😻
 colorFrom: green
 colorTo: yellow
 sdk: gradio
-sdk_version: 5.4.0
+sdk_version: 4.39.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+import Stemmer  # from PyStemmer
+import bm25s
+import numpy as np
+import time
+import os
+
+examples = ["Can officers always order a passenger out of a car?","Find me briefs about credential searches", "Can police search an impounded car without a warrant?", "State is arguing State v. Carty is not good law"]
+
+def show_user_query(user_message, history):
+    return "", history + [[user_message, None]]
+
+def retrieve(history):
+    query = history[-1][0]
+    print ("query", query)
+    query_tokens = bm25s.tokenize(query, stemmer=stemmer)
+
+    results, scores = retriever.retrieve(query_tokens, k=10)
+    query_response = " ".join([str(i) for i in results])
+    #query_response = retrieve(history[-1][0])
+    #print (query_response)
+    history[-1][1] = ""
+    for character in query_response:
+        history[-1][1] += character
+        time.sleep(0.0001)
+        yield history
+
+
+
+with gr.Blocks() as demo:
+    #gr.LoginButton()
+    chatbot = gr.Chatbot()
+    query_textbox = gr.Textbox()
+    examples = gr.Examples(examples, query_textbox)
+    next_button = gr.ClearButton(chatbot, value="Clear", visible=True)
+
+    query_textbox.submit(show_user_query, [query_textbox, chatbot], [query_textbox, chatbot], queue=False).then(retrieve, chatbot, chatbot)
+
+
+def load_bm25():
+    stemmer = Stemmer.Stemmer("english")
+    retriever = bm25s.BM25.load("NJ_index_LLM_chunking", mmap=False)
+    return retriever, stemmer
+
+retriever, stemmer = load_bm25()
+#demo.launch(share=True, auth=(os.environ.get("username"), os.environ.get("password")))
+#demo.launch(auth=("admin", "pass1234"))
+
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+bm25s
+PyStemmer
+https://gradio-builds.s3.amazonaws.com/4485dd46a8e4b3f5b35e42d52f291b72fdc1a952/gradio-4.39.0-py3-none-any.whl