Update app.py
Browse filesAdd complete case text
app.py
CHANGED
@@ -3,6 +3,7 @@ from huggingface_hub import InferenceClient, login, snapshot_download
|
|
3 |
from langchain_community.vectorstores import FAISS
|
4 |
from langchain_huggingface import HuggingFaceEmbeddings
|
5 |
import os
|
|
|
6 |
|
7 |
|
8 |
"""
|
@@ -21,6 +22,8 @@ embeddings = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-small")
|
|
21 |
|
22 |
vector_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
|
23 |
|
|
|
|
|
24 |
def respond(
|
25 |
message,
|
26 |
history: list[tuple[str, str]],
|
@@ -42,10 +45,12 @@ def respond(
|
|
42 |
context = ""
|
43 |
|
44 |
for doc in documents:
|
|
|
|
|
45 |
context += "Case number: " + doc.metadata["case_nb"] + spacer
|
46 |
context += "Case date: " + doc.metadata["case_date"] + spacer
|
47 |
context += "Case url: " + doc.metadata["case_url"] + spacer
|
48 |
-
context += "Case
|
49 |
|
50 |
message = f"""
|
51 |
A user is asking you the following question: {message}
|
|
|
3 |
from langchain_community.vectorstores import FAISS
|
4 |
from langchain_huggingface import HuggingFaceEmbeddings
|
5 |
import os
|
6 |
+
import pandas as pd
|
7 |
|
8 |
|
9 |
"""
|
|
|
22 |
|
23 |
vector_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
|
24 |
|
25 |
+
df = pd..read_csv("bger_cedh_db 1954-2024.csv")
|
26 |
+
|
27 |
def respond(
|
28 |
message,
|
29 |
history: list[tuple[str, str]],
|
|
|
45 |
context = ""
|
46 |
|
47 |
for doc in documents:
|
48 |
+
case_text = df[df["case_url"] == doc.metadata["case_url"]].case_text.values[0]
|
49 |
+
|
50 |
context += "Case number: " + doc.metadata["case_nb"] + spacer
|
51 |
context += "Case date: " + doc.metadata["case_date"] + spacer
|
52 |
context += "Case url: " + doc.metadata["case_url"] + spacer
|
53 |
+
context += "Case text: " + case_text + spacer
|
54 |
|
55 |
message = f"""
|
56 |
A user is asking you the following question: {message}
|