SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 17:05:41 +01:00
Code Issues Projects Releases Packages Activity

Update ui.py

Browse source 
Change Source to hyperlink to local pdf with corresponding page
This commit is contained in:
slale-91 2024-08-05 08:31:37 +02:00 • committed by GitHub
parent e2847daf4c
commit 971c9780d0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
private_gpt/ui/ui.py
View file

@ -55,6 +55,7 @@ class Source(BaseModel):
file: str file: str
page: str page: str
text: str text: str
pdf_prefix: str = "C:/UsedFilesFolder/"
class Config: class Config:
frozen = True frozen = True
@ -76,7 +77,11 @@ class Source(BaseModel):
) # Unique sources only ) # Unique sources only
return curated_sources return curated_sources
def to_hyperlink(self) -> str:
encoded_file = self.file.replace(" ", "%20")
file_path = f"{self.pdf_prefix}{encoded_file}#page={self.page}"
return f'<a href="file:///{file_path}" target="_blank">{self.file} (page {self.page})</a>'
@singleton @singleton
class PrivateGptUi: class PrivateGptUi:
@ -125,7 +130,7 @@ class PrivateGptUi:
if f"{source.file}-{source.page}" not in used_files: if f"{source.file}-{source.page}" not in used_files:
sources_text = ( sources_text = (
sources_text sources_text
+ f"{index}. {source.file} (page {source.page}) \n\n" + f"{index}. {source.to_hyperlink()} \n\n"
) )
used_files.add(f"{source.file}-{source.page}") used_files.add(f"{source.file}-{source.page}")
sources_text += "<hr>\n\n" sources_text += "<hr>\n\n"