Update app.py

app.py

@@ -1,4 +1,8 @@
 import json
+import base64
+import os
+import shutil
+import time
 import gradio as gr
 
 from rag.search import search
@@ -7,12 +11,6 @@ from llm.reasoning import generate_reasoning
 from llm.reasoning import generate_reasoning_from_prompt
 from rag.ingest import ingest_pdfs_and_web, save_chunks
 
-import base64
-
-import os
-
-import time
-
 print("🔄 Startar RAG-ingest")
 DATA_DIR = "rag/data"
 start_time = time.perf_counter()
@@ -38,6 +36,16 @@ with open("content.json", encoding="utf-8") as f:
 
 DOC_INDEX = {d["id"]: d for d in DOCUMENTS}
 
+PUBLIC_DIR = "/tmp/gradio/public_pdfs"
+os.makedirs(PUBLIC_DIR, exist_ok=True)
+
+for file in os.listdir("rag/files"):
+    if file.lower().endswith(".pdf"):
+        shutil.copy(
+            os.path.join("rag/files", file),
+            os.path.join(PUBLIC_DIR, file)
+        )
+
 # =====================================================
 # FUNKTIONER
 # =====================================================
@@ -111,15 +119,13 @@ def clear_all():
 
 def format_source_link(chunk: dict) -> str:
     source = chunk.get("source", "Okänd källa")
-    source_type = chunk.get("source_type")
 
-    if source_type == "pdf":
-        return f"[{source}](/file=rag/files/{source})"
+    if chunk.get("source_type") == "pdf":
+        return f"[{source}](/file={PUBLIC_DIR}/{source})"
 
-    if source_type == "web":
+    if chunk.get("source_type") == "web":
         return f"[{source}]({source})"
 
-    # Fallback – visa åtminstone namnet
     return source