Merge pull request #27 from GFA-DIU/fix-page-number-in-chatinterface

conversation/citation_utils.py

@@ -42,6 +42,13 @@ def format_artifacts_to_string(artifacts: List[dict]) -> str:
         )
     return "\n\n".join(parts)
 
+def fix_page_number(incorrect_page_number: str) -> str:
+    if incorrect_page_number == "No page number":
+        return incorrect_page_number
+    else:
+        correct_page_number = int(incorrect_page_number) + 1
+        return str(correct_page_number)
+
 def embed_references(result: dict) -> str:
     """
     Given an LLM result dict with keys:
@@ -61,9 +68,9 @@ def embed_references(result: dict) -> str:
     sources = sorted(result.sources, key=lambda s: s.source_number)
     
     # Build the references section
-    refs = ["", "**References:**"]
+    refs = ["", "**Referenzen:**"]
     for src in sources:
         idx = src.source_number
-        refs.append(f"[{idx}] {src.file}, page {src.page}")
+        refs.append(f"[{idx}] {src.file}, Seite {fix_page_number(src.page)}")
     
     return answer + "\n\n" + "\n\n".join(refs)