Update app.py

app.py

@@ -21,7 +21,6 @@ login(token=hf_token, add_to_git_credential=True)
 openai_key = os.environ["OPENAI_API_KEY"]
 
 
-# Define the template
 template = PromptTemplate(
     prompt="""
 Answer the given question using the following documents. \
@@ -30,7 +29,7 @@ Provide example quotes and citations using extracted text from the documents. \
 Use facts and numbers from the documents in your answer. \
 Reference information used from documents at the end of each applicable sentence (ex: [source: document_name]), where 'document_name' is the text provided at the start of each document (demarcated by '- &&&' and '&&&:')'. \
 If no relevant information to answer the question is present in the documents, just say you don't have enough information to answer. \
-Context: {' - '.join(['&&& '+d.meta['document_name']+' ref. '+str(d.meta['ref_id'])+' &&&: '+d.content for d in documents])}; Question: {query}; Answer:""",
+Context: {' - '.join(['&&& '+d.meta['document']+' ref. '+str(d.meta['ref_id'])+' &&&: '+d.content for d in documents])}; Question: {query}; Answer:""",
 )
 
 # Create a list of options for the dropdown
@@ -74,7 +73,7 @@ def get_docs(input_query, country = None):
         # Create a Document object for each row
         doc = Document(
             row['content'],
-            meta={'country': row['country'],'document_name': row['document'], 'ref_id': row['ref_id'], 'score': row['score']}
+            meta={'country': row['country'],'document': row['document'], 'page': row['page'], 'file_name': row['file_name'], 'ref_id': row['ref_id'], 'score': row['score']}
         )
 
         # Append the Document object to the documents list
@@ -97,7 +96,7 @@ def get_refs(docs, res):
       doc = docs[i].to_dict()
       ref_id = doc['meta']['ref_id']
       if ref_id in ref_ids:
-            result_str += "**Ref. " + str(ref_id) + " [" + doc['meta']['country'] + " " + doc['meta']['document_name'] + "]:** " + "*'" + doc['content'] + "'*<br> <br>" # Add <br> for a line break
+            result_str += "**Ref. " + str(ref_id) + " [" + doc['meta']['country'] + " " + doc['meta']['document'] + ' p' + str(doc['meta']['page']) + "]:** " + "*'" + doc['content'] + "'*<br> <br>" # Add <br> for a line break
 
   return result_str