umaiku commited on
Commit
d5c54ef
·
verified ·
1 Parent(s): 9797856

Update app.py

Browse files

Add complete case text

Files changed (1) hide show
  1. app.py +6 -1
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 chunk: " + doc.page_content + spacer
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}