pdf download button code

src/streamlit_app.py

@@ -48,8 +48,7 @@ question = st.text_input(
 # ---------------- ANSWER ----------------
 if question:
     with st.spinner("Searching document..."):
-        query_embedding = embedder.encode([question]).astype("float32")
-        contexts = retrieve(query_embedding)
+        contexts = retrieve(question)
 
     context_text = "\n\n".join(c["text"] for c in contexts)
 
@@ -72,7 +71,20 @@ Answer clearly and concisely.
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": prompt},
             ],
+            max_tokens=800
         )
 
+    answer = response.choices[0].message.content
+    st.session_state["last_answer"] = answer
+
     st.subheader("Answer")
-    st.write(response.choices[0].message.content)
+    st.write(answer)
+
+    pdf_buffer = answer_to_pdf(question, answer)
+
+    st.download_button(
+        label="📄 Download Answer as PDF",
+        data=pdf_buffer,
+        file_name="OTT_Bot_Answer.pdf",
+        mime="application/pdf"
+    )