Update app.py

app.py

@@ -1,4 +1,3 @@
-##
 import streamlit as st
 import PyPDF2
 import openai
@@ -7,6 +6,9 @@ import os
 import numpy as np
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+from io import BytesIO
 
 # Function to extract text from a PDF file
 def extract_text_from_pdf(pdf_file):
@@ -27,6 +29,30 @@ def search_similar(query_embedding, index, stored_texts, top_k=3):
     results = [(stored_texts[i], distances[0][idx]) for idx, i in enumerate(indices[0])]
     return results
 
+# Function to create a PDF from the response
+def create_pdf(response_text):
+    buffer = BytesIO()
+    c = canvas.Canvas(buffer, pagesize=letter)
+    width, height = letter
+
+    # Add the response text to the PDF
+    c.drawString(30, height - 30, "Intelligent Reply:")
+    text_object = c.beginText(30, height - 50)
+    text_object.setFont("Helvetica", 10)
+    text_object.setTextOrigin(30, height - 50)
+
+    # Add the response text, line by line
+    lines = response_text.split("\n")
+    for line in lines:
+        text_object.textLine(line)
+
+    c.drawText(text_object)
+    c.showPage()
+    c.save()
+
+    buffer.seek(0)
+    return buffer
+
 # Streamlit app starts here
 st.title("Course Query Assistant")
 
@@ -86,6 +112,18 @@ if openai_api_key:
                 messages=[{"role": "user", "content": modified_prompt}]
             )
 
+            # Extract the response text
+            response_text = response['choices'][0]['message']['content']
+
             # Display the response
             st.write("### Intelligent Reply:")
-            st.write(response['choices'][0]['message']['content'])
+            st.write(response_text)
+
+            # Button to download the response as PDF
+            pdf_buffer = create_pdf(response_text)
+            st.download_button(
+                label="Download Intelligent Reply as PDF",
+                data=pdf_buffer,
+                file_name="intelligent_reply.pdf",
+                mime="application/pdf"
+            )