Update app.py

app.py

@@ -9,6 +9,7 @@ from langchain.llms import OpenAI
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from io import BytesIO
+from googletrans import Translator
 
 # Set your OpenAI API key here
 os.environ["OPENAI_API_KEY"] = "sk-proj-Mb6oBgg51EUsImee2TMdT3BlbkFJjyQuPaCGgFNzYwt1c52m"
@@ -52,6 +53,12 @@ if uploaded_file is not None:
         answer = chain.run(input_documents=docs, question=prompt)
         st.write("Answer:", answer)
 
+        # Translate answer
+        translator = Translator()
+        translation_language = st.selectbox("Select translation language:", ["es", "fr", "de", "hi"])  # Add more languages as needed
+        translated_answer = translator.translate(answer, dest=translation_language).text
+        st.write("Translated Answer:", translated_answer)
+
         if st.button("Generate PDF"):
             # Generate PDF
             pdf_buffer = BytesIO()
@@ -67,6 +74,17 @@ if uploaded_file is not None:
                     y = 750
                 c.drawString(100, y, line)
                 y -= 15
+
+            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)