PDF_Generator / app.py
DataCracker's picture
Create app.py
4755734 verified
raw
history blame
1.38 kB
import streamlit as st
from fpdf import FPDF
import tempfile
if 'qa_pairs' not in st.session_state:
st.session_state.qa_pairs = []
def generate_pdf(qa_pairs):
pdf = FPDF()
pdf.add_page()
pdf.set_font("Arial", size=12)
for qa in qa_pairs:
pdf.multi_cell(0, 10, f"Q: {qa['question']}")
pdf.multi_cell(0, 10, f"A: {qa['answer']}")
pdf.ln()
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
pdf.output(temp_file.name)
return temp_file.name
st.title("Q&A Input Tool")
question = st.text_input("Enter a question:")
answer = st.text_area("Enter the answer:")
if st.button("Add Question & Answer"):
if question and answer:
st.session_state.qa_pairs.append({"question": question, "answer": answer})
st.success("Q&A added successfully!")
else:
st.error("Please enter both a question and an answer.")
if st.session_state.qa_pairs:
pdf_file_path = generate_pdf(st.session_state.qa_pairs)
with open(pdf_file_path, "rb") as pdf_file:
st.download_button(
label="Download PDF",
data=pdf_file,
file_name="QandA.pdf",
mime="application/pdf"
)
st.header("Current Q&A Pairs")
for qa in st.session_state.qa_pairs:
st.write(f"Q: {qa['question']}")
st.write(f"A: {qa['answer']}")
st.write("---")