add the download PDF button

pages/Multimodal.py

@@ -4,6 +4,10 @@ from datetime import datetime
 from PIL import Image
 import google.generativeai as genai
 import os
+from reportlab.pdfgen import canvas
+from reportlab.lib.pagesizes import A4
+from io import BytesIO
+import textwrap
 
 MODEL_ID = "gemini-2.0-flash-exp" 
 api_key = os.getenv("GEMINI_API_KEY")
@@ -38,16 +42,50 @@ def save_user_prompt(username, prompt_time, prompt_type):
     conn.commit()
     conn.close()
 
+def create_pdf(data):
+    try:
+        pdf_file = BytesIO()
+        c = canvas.Canvas(pdf_file, pagesize=A4)
+        
+        # Set font and font size
+        c.setFont("Helvetica", 12)
+        
+        # Set margin
+        margin = 72  # 1 inch in points
+        x_position = margin
+        y_position = A4[1] - margin  # Start from the top of the page
+        
+        # Add title
+        c.drawString(x_position, y_position, "Exam Questions")
+        y_position -= 20
+        
+        # Wrap markdown data
+        lines = textwrap.wrap(data, width=60)  # Adjust width to fit the page
+        for line in lines:
+            if y_position < margin:  # Check if we need to start a new page
+                c.showPage()
+                y_position = A4[1] - margin  # Reset y-position to the top of the new page
+            
+            c.drawString(x_position, y_position, line)
+            y_position -= 20  # Adjust line spacing
+        
+        c.showPage()
+        c.save()
+        pdf_file.seek(0)
+        return pdf_file
+    
+    except Exception as e:
+        st.error(f"An error occurred while creating PDF file: {e}")
+        return None        
+
 def show_multimodal():
     st.subheader("Multimodal")
     username = st.session_state["username"]
+    st.write(f"Welcome, {username}! This page allows you to generate questions based on an image or PDF file.")
 
     # we dont use the system instruction for now
     #system_instruction = get_system_instruction(username)
 
-    # Streamlit app
-    st.title("Gemini Image Chat")
-
     # File uploader with allowed types
     uploaded_file = st.file_uploader("Choose an image or PDF...", type=["jpg", "jpeg", "png", "pdf"])
 
@@ -110,7 +148,7 @@ def show_multimodal():
                                 [1, 2, 3, 4, 5]) 
 
     # Combine user inputs into a prompt
-    prompt = f"""Refer to the provided Document. Generate {question_type} questions for a {year_level} {course} student
+    prompt = f"""Refer to the provided document. Generate {question_type} questions for a {year_level} {course} student
                     in {subject} on the topic of {topic} with a {difficulty} difficulty level.
                     The questions should require higher order thinking skills. 
                     """
@@ -206,8 +244,18 @@ def show_multimodal():
 
         #record the prompt for monitoring
         save_user_prompt(username, datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "Multimodal")
+        
+        # Create PDF
+        pdf_file = create_pdf(full_response)
+
+        # Generate a unique filename based on the current time
+        current_time = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+        pdf_filename = f"MC Type Exam_{current_time}.pdf"
 
-        st.success("Exam was generated succesfully.")
+        # Download button
+        if pdf_file:
+            st.download_button("Download PDF", pdf_file, pdf_filename, "application/pdf")
+            st.success("Exam was generated succesfully.")
 
 if st.session_state["authenticated"]:
     show_multimodal()   

requirements.txt

@@ -1,4 +1,5 @@
 streamlit
 passlib
 bcrypt==3.2.0
-google-generativeai
+google-generativeai
+reportlab