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}
|