Update utils.py

utils.py

@@ -19,6 +19,7 @@ import json
 import base64
 from io import BytesIO 
 import urllib.parse
+from urllib.parse import quote
 import tempfile
 import uuid
 
@@ -747,8 +748,6 @@ def rag_chain_simpel( prompt,  retriever):
     #passend zum Prompt relevante Dokuemnte raussuchen
     relevant_docs = retriever.invoke(prompt)
 
-    print("relevant docs............."+str(relevant_docs))
-
     #zu jedem relevanten Dokument die wichtigen Informationen zusammenstellen (im Dict)
     extracted_docs = extract_document_info(relevant_docs)
     
@@ -869,6 +868,7 @@ def transfer_input(inputs):
 
 ########################################################
 ######## Hilfsfunktionen Datei-Upload ##################
+"""
 def download_link(doc):
     # URL für das Herunterladen der Datei
     # Check if doc is a dictionary and contains the key 'pfad'
@@ -878,6 +878,23 @@ def download_link(doc):
     else:
         file_url = f"https://huggingface.co/spaces/alexkueck/SucheRAG/resolve/main/{doc}?token=hf_token"
         return f'<b><a href="{file_url}" target="_blank" style="color: #BB70FC; font-weight: bold;">{doc}</a></b>'
+"""
+
+def download_link(doc):
+    # Basis-URL für das Hugging Face Repository
+    base_url = f"https://huggingface.co/spaces/{STORAGE_REPO_ID}/resolve/main"
+
+    # Check if doc is a dictionary and contains the key 'pfad'
+    if isinstance(doc, dict) and 'pfad' in doc:
+        # URL-encode the path to handle special characters
+        encoded_path = quote(doc['pfad'])
+        file_url = f"{base_url}/{encoded_path}?token={hf_token}"
+        return f'<b><a href="{file_url}" target="_blank" style="color: #BB70FC; font-weight: bold;">{doc["titel"]}</a></b>'
+    else:
+        # URL-encode the document name to handle special characters
+        encoded_doc = quote(doc)
+        file_url = f"{base_url}/{encoded_doc}?token={hf_token}"
+        return f'<b><a href="{file_url}" target="_blank" style="color: #BB70FC; font-weight: bold;">{doc}</a></b>'