Update app.py

app.py

@@ -16,6 +16,9 @@ os.environ["OPENAI_API_KEY"] = "sk-proj-Mb6oBgg51EUsImee2TMdT3BlbkFJjyQuPaCGgFNz
 
 st.title("PDF Question Answering")
 
+if 'translated_answer' not in st.session_state:
+    st.session_state.translated_answer = ""
+
 uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
 
 if uploaded_file is not None:
@@ -53,24 +56,38 @@ if uploaded_file is not None:
         answer = chain.run(input_documents=docs, question=prompt)
         st.write("Answer:", answer)
 
-        # Language selection and translation
-        translation_language = st.selectbox("Select translation language:", ["es", "fr", "de", "hi"])  # Add more languages as needed
-
-        if st.button("Translate Answer"):
-            translated_answer = GoogleTranslator(source='auto', target=translation_language).translate(answer)
-            st.write("Translated Answer:", translated_answer)
-        else:
-            translated_answer = ""
-
-        if st.button("Generate PDF"):
-            # Generate PDF
-            pdf_buffer = BytesIO()
-            c = canvas.Canvas(pdf_buffer, pagesize=letter)
-            c.drawString(100, 750, "Question:")
-            c.drawString(100, 735, query)
-            c.drawString(100, 715, "Answer:")
-            text_lines = answer.split('\n')
-            y = 700
+        st.session_state.answer = answer
+
+    # Language selection and translation
+    translation_language = st.selectbox("Select translation language:", ["es", "fr", "de", "hi"])  # Add more languages as needed
+
+    if st.button("Translate Answer"):
+        translated_answer = GoogleTranslator(source='auto', target=translation_language).translate(st.session_state.answer)
+        st.session_state.translated_answer = translated_answer
+
+    if st.session_state.translated_answer:
+        st.write("Translated Answer:", st.session_state.translated_answer)
+
+    if st.button("Generate PDF"):
+        # Generate PDF
+        pdf_buffer = BytesIO()
+        c = canvas.Canvas(pdf_buffer, pagesize=letter)
+        c.drawString(100, 750, "Question:")
+        c.drawString(100, 735, query)
+        c.drawString(100, 715, "Answer:")
+        text_lines = st.session_state.answer.split('\n')
+        y = 700
+        for line in text_lines:
+            if y < 50:
+                c.showPage()
+                y = 750
+            c.drawString(100, y, line)
+            y -= 15
+
+        if st.session_state.translated_answer:
+            c.drawString(100, y, "Translated Answer:")
+            y -= 15
+            text_lines = st.session_state.translated_answer.split('\n')
             for line in text_lines:
                 if y < 50:
                     c.showPage()
@@ -78,23 +95,12 @@ if uploaded_file is not None:
                 c.drawString(100, y, line)
                 y -= 15
 
-            if translated_answer:
-                c.drawString(100, y, "Translated Answer:")
-                y -= 15
-                text_lines = translated_answer.split('\n')
-                for line in text_lines:
-                    if y < 50:
-                        c.showPage()
-                        y = 750
-                    c.drawString(100, y, line)
-                    y -= 15
-
-            c.save()
-
-            pdf_buffer.seek(0)
-            st.download_button(
-                label="Download Answer as PDF",
-                data=pdf_buffer,
-                file_name="answer.pdf",
-                mime="application/pdf"
-            )
+        c.save()
+
+        pdf_buffer.seek(0)
+        st.download_button(
+            label="Download Answer as PDF",
+            data=pdf_buffer,
+            file_name="answer.pdf",
+            mime="application/pdf"
+        )