Added download button for the answer

requirements.txt

@@ -9,4 +9,5 @@ datasets
 scikit-learn
 pypdf
 pdfplumber
-torch
+torch
+reportlab

src/pdf_utils.py

@@ -0,0 +1,31 @@
+from io import BytesIO
+from reportlab.lib.pagesizes import A4
+from reportlab.pdfgen import canvas
+from reportlab.lib.units import inch
+
+def answer_to_pdf(question: str, answer: str) -> BytesIO:
+    buffer = BytesIO()
+    c = canvas.Canvas(buffer, pagesize=A4)
+    width, height = A4
+
+    textobject = c.beginText(1 * inch, height - 1 * inch)
+    textobject.setFont("Helvetica", 11)
+
+    textobject.textLine("OTT Bot – Answer")
+    textobject.textLine("-" * 50)
+    textobject.textLine("")
+    textobject.textLine(f"Question:")
+    textobject.textLine(question)
+    textobject.textLine("")
+    textobject.textLine("Answer:")
+    textobject.textLine("")
+
+    for line in answer.split("\n"):
+        textobject.textLine(line)
+
+    c.drawText(textobject)
+    c.showPage()
+    c.save()
+
+    buffer.seek(0)
+    return buffer

src/streamlit_app.py

@@ -6,6 +6,7 @@ from ingest import build_index
 from rag import retrieve
 from groq import Groq
 from config import EMBEDDING_MODEL
+from pdf_utils import answer_to_pdf
 
 # ---------------- CONFIG ----------------
 st.set_page_config(page_title="OTT Bot", layout="wide")