momenaca commited on
Commit
cfaf8d8
·
1 Parent(s): ac7b044

fix missing source link for press

Browse files
spinoza_project/source/frontend/utils.py CHANGED
@@ -58,23 +58,37 @@ def get_source_link(metadata):
58
  return metadata["file_url"] + f"#page={metadata['content_page_number'] + 1}"
59
 
60
 
61
- def make_html_presse_source(source, i, score, config):
62
  meta = source.metadata
63
- return f"""
64
- <div class="card" id="doc{i}">
65
- <div class="card-content">
66
- <h2>Doc {i} - {meta['file_title']} - {meta['file_publisher']}</h2>
67
- <p>{source.page_content}</p>
 
 
 
 
 
 
 
 
 
68
  </div>
69
- <div class="card-footer">
70
- <span>{meta['file_source_type']}</span>
71
- <span>Relevance Score : {round(100*score,1)}%</span>
72
- <a href={meta['file_url']} target="_blank">
73
- <span role="img" aria-label="Open PDF">🔗</span>
74
- </a>
 
 
 
 
 
 
75
  </div>
76
- </div>
77
- """
78
 
79
 
80
  def make_html_source(source, i, score, config):
@@ -95,6 +109,7 @@ def make_html_source(source, i, score, config):
95
  </div>
96
  """
97
 
 
98
  def parse_output_llm_with_sources(output):
99
  content_parts = re.split(
100
  r"[\[(]?(Doc\s?\d+(?:,\s?Doc\s?\d+)*|doc\s?\d+(?:,\s?doc\s?\d+)*|Doc\s\d+)[\])?]",
 
58
  return metadata["file_url"] + f"#page={metadata['content_page_number'] + 1}"
59
 
60
 
61
+ def make_html_presse_source(source, i, score):
62
  meta = source.metadata
63
+ if meta["file_url"] != "none":
64
+ return f"""
65
+ <div class="card" id="doc{i}">
66
+ <div class="card-content">
67
+ <h2>Doc {i} - {meta['file_title']} - {meta['file_publisher']}</h2>
68
+ <p>{source.page_content}</p>
69
+ </div>
70
+ <div class="card-footer">
71
+ <span>{meta['file_source_type']}</span>
72
+ <span>Relevance Score : {round(100*score,1)}%</span>
73
+ <a href={meta['file_url']} target="_blank">
74
+ <span role="img" aria-label="Open PDF">🔗</span>
75
+ </a>
76
+ </div>
77
  </div>
78
+ """
79
+ else:
80
+ return f"""
81
+ <div class="card" id="doc{i}">
82
+ <div class="card-content">
83
+ <h2>Doc {i} - {meta['file_title']} - {meta['file_publisher']}</h2>
84
+ <p>{source.page_content}</p>
85
+ </div>
86
+ <div class="card-footer">
87
+ <span>{meta['file_source_type']}</span>
88
+ <span>Relevance Score : {round(100*score,1)}%</span>
89
+ </div>
90
  </div>
91
+ """
 
92
 
93
 
94
  def make_html_source(source, i, score, config):
 
109
  </div>
110
  """
111
 
112
+
113
  def parse_output_llm_with_sources(output):
114
  content_parts = re.split(
115
  r"[\[(]?(Doc\s?\d+(?:,\s?Doc\s?\d+)*|doc\s?\d+(?:,\s?doc\s?\d+)*|Doc\s\d+)[\])?]",