ignore and some updates

.gitignore

@@ -0,0 +1,2 @@
+*.pyc
+__pycache__

app.py

@@ -6,6 +6,8 @@ from datasets import load_dataset
 from sentence_transformers import SentenceTransformer
 
 idx = 0
+index = None
+newdoc = None
 
 dataset = load_dataset("tollefj/rettsavgjoerelser_100samples_embeddings")
 model = SentenceTransformer("NbAiLab/nb-sbert-base")
@@ -27,18 +29,13 @@ def get_doc_embeddings(doc):
     return np.array(doc.embedding.tolist(), dtype="float32")
 
 
-def faiss_search(doc_url, query_str, K=5):
+def faiss_search(query_str, K=5):
     global idx
+    global index
+    global newdoc
     # find idx from url:
-    doc_idx = df[df.url == doc_url].index[0]
-    idx = int(doc_idx)
-    newdoc = build_doc_frame(df, idx)
-    embeddings = get_doc_embeddings(newdoc)
-
-    faiss.normalize_L2(embeddings)
-    index = faiss.IndexFlatIP(768)
-    index.add(embeddings)
-
+    # doc_idx = df[df.url == doc_url].index[0]
+    # idx = int(doc_idx)
     target_emb = model.encode([query_str])
     target_emb = np.array([target_emb.reshape(-1)])
     faiss.normalize_L2(target_emb)
@@ -65,25 +62,56 @@ def faiss_search(doc_url, query_str, K=5):
 dropdown_opts = [doc.url for idx, doc in df.iterrows()]
 
 with gr.Blocks() as demo:
-    gr.Label("Lovdata rettsavgjørelser - semantisk søk")
-
-    case_dropdown = gr.Dropdown(label="Velg en rettsavgjørelse", choices=dropdown_opts, default=dropdown_opts[0])
-    # when case_dropdown changes, update the summary dropdown:
-    # idx_label = gr.Label(f"Current index: {idx}")
-
-    query = gr.Textbox(
-        label="Søk etter setninger",
-        lines=1,
-        placeholder="Kollisjon mellom to kjøretøy.",
+    gr.HTML(
+        """
+        <h1>Lovdata rettsavgjørelser - semantisk søk</h1>
+        """
     )
-    k_slider = gr.Slider(minimum=1, maximum=10, label="Antall treff", value=5, step=1)
 
-    search_btn = gr.Button("Søk")
+    def on_selection_change(selected_case):
+        global idx
+        global index
+        global newdoc
+        idx = df[df.url == selected_case].index[0]
+        print("Selection changed!")
+        print(selected_case)
+
+        newdoc = build_doc_frame(df, idx)
+        embeddings = get_doc_embeddings(newdoc)
+        faiss.normalize_L2(embeddings)
+        index = faiss.IndexFlatIP(768)
+        index.add(embeddings)
+
+        summary = df.iloc[idx].summary.tolist()
+        # make a nice html-formatted ul-li list:
+        summary_html = "<ul>" + "".join([f"<li>{sent}</li>" for sent in summary]) + "</ul>"
+        # summary_dropdown.update(choices=summary, label="Velg fra oppsummeringene")
+        url_html = f"<a href='{selected_case}' target='_blank'>{selected_case}</a>"
+        return summary_html, url_html
+
+    with gr.Row():
+        with gr.Column():
+            case_dropdown = gr.Dropdown(label="Velg en rettsavgjørelse", choices=dropdown_opts)
+            summary_html = gr.HTML(label="Predefinert oppsummering", placeholder="<p>Velg en sak først<p>")
+            case_url = gr.HTML(label="URL til rettsavgjørelse", placeholder="https://lovdata.no/...")
+        with gr.Column():
+            query = gr.Textbox(
+                label="Søk etter setninger",
+                lines=1,
+                placeholder="Kollisjon mellom to kjøretøy.",
+            )
+            k_slider = gr.Slider(minimum=1, maximum=10, label="Antall treff", value=5, step=1)
+            search_btn = gr.Button("Søk")
 
     output = gr.Textbox(label="Resultater", lines=10)
 
-    # from the selected URL, find the index in the df:
-    search_btn.click(faiss_search, inputs=[case_dropdown, query, k_slider], outputs=[output])
+    case_dropdown.change(
+        on_selection_change,
+        inputs=[case_dropdown],
+        outputs=[summary_html, case_url],
+    )
+
+    search_btn.click(faiss_search, inputs=[query, k_slider], outputs=[output])
 
     # clear_btn.click(None, inputs=[None, None], outputs=None)
     # search_btn.click(faiss_search, inputs=[None, None, None], outputs=["text"])