Corrected code for download button

src/streamlit_app.py

@@ -28,7 +28,6 @@ client = Groq(api_key=GROQ_API_KEY)
 
 # ---------------- UI ----------------
 st.title("📚 OTT Bot")
-
 st.markdown("Upload PDFs via HF Dataset and query them using AI")
 
 # ---------------- INGESTION ----------------
@@ -48,11 +47,12 @@ question = st.text_input(
 # ---------------- ANSWER ----------------
 if question:
     with st.spinner("Searching document..."):
-        contexts = retrieve(question)
+        query_embedding = embedder.encode([question]).astype("float32")
+        contexts = retrieve(query_embedding)
 
-    context_text = "\n\n".join(c["text"] for c in contexts)
+        context_text = "\n\n".join(c["text"] for c in contexts)
 
-    prompt = f"""
+        prompt = f"""
 Answer the question based on the context below.
 
 Context:
@@ -71,20 +71,19 @@ 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
+    answer_text = response.choices[0].message.content
 
     st.subheader("Answer")
-    st.write(answer)
+    st.write(answer_text)
 
-    pdf_buffer = answer_to_pdf(question, answer)
+    # ---------------- PDF DOWNLOAD ----------------
+    pdf_buffer = answer_to_pdf(question, answer_text)
 
     st.download_button(
         label="📄 Download Answer as PDF",
         data=pdf_buffer,
         file_name="OTT_Bot_Answer.pdf",
         mime="application/pdf"
-    )
+    )