Update app.py

app.py

@@ -64,6 +64,10 @@ from google.cloud import storage, exceptions  # Import exceptions for error hand
 from google.cloud.exceptions import NotFound
 from google.oauth2 import service_account
 
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+from reportlab.pdfbase.ttfonts import TTFont
+from reportlab.lib import colors
 
 import logging
 
@@ -366,8 +370,7 @@ def save_to_google_storage(bucket_name, file_path, destination_blob_name, expira
     )
     print(f"✅ File uploaded to Google Cloud Storage. Signed URL: {signed_url}")
     return signed_url
-
-# Function to generate and save a document
+# Function to generate and save a document using ReportLab
 def generate_document(task_description, md_content, user_name='jayw', bucket_name='curify'):
     logger.debug("Starting to generate document")
 
@@ -380,34 +383,47 @@ def generate_document(task_description, md_content, user_name='jayw', bucket_nam
 
     # Generate PDF file locally
     local_filename = f"{truncated_hash}.pdf"  # Use the truncated hash as the local file name
-    pdf = FPDF()
-    pdf.add_page()
+    c = canvas.Canvas(local_filename, pagesize=letter)
 
-    # Use a unified font that supports Unicode (e.g., NotoSansCJK)
-    font_path = 'NotoSansCJK-Regular.ttc'
+    # Load the .ttc font using ReportLab's TTFont and set it as the font for the canvas
+    font_path = 'NotoSansCJK-Regular.ttc'  # Path to your .ttc file
     try:
-        pdf.add_font('NotoSansCJK', '', font_path, uni=True)
-        pdf.set_font('NotoSansCJK', size=12)
+        # Register the font from the .ttc file
+        cttc_font = TTFont('NotoSansCJK', font_path)
+        c.ttfonts.addRegisteredFont('NotoSansCJK', cttc_font)
+        c.setFont('NotoSansCJK', 12)
     except Exception as e:
         logger.error(f"Error loading font from {font_path}: {e}")
         raise RuntimeError("Failed to load the unified font. Ensure the font file is accessible.")
 
+    # Set initial Y position for drawing text
+    y_position = 750  # Starting position for text
+
     # Process dictionary and render content
     for key, value in md_content.items():
         # Add key as a header
-        pdf.set_font('NotoSansCJK', style='B', size=12)  # Bold font for headers
-        pdf.multi_cell(0, 10, f"# {key}")
+        c.setFont('NotoSansCJK', 14)
+        c.drawString(100, y_position, f"# {key}")
+        y_position -= 20
 
         # Add value
-        pdf.set_font('NotoSansCJK', size=12)  # Regular font for content
+        c.setFont('NotoSansCJK', 12)  # Regular font for content
         if isinstance(value, list):  # Handle lists
             for item in value:
-                pdf.multi_cell(0, 10, f"- {item}")
+                c.drawString(100, y_position, f"- {item}")
+                y_position -= 15
         else:  # Handle single strings
-            pdf.multi_cell(0, 10, value)
+            c.drawString(100, y_position, value)
+            y_position -= 15
+
+        # Check if the page needs to be broken (if Y position is too low)
+        if y_position < 100:
+            c.showPage()  # Create a new page
+            c.setFont('NotoSansCJK', 12)  # Reset font
+            y_position = 750  # Reset the Y position for the new page
 
-    # Output the PDF to a local file
-    pdf.output(local_filename)
+    # Save the PDF
+    c.save()
 
     # Organize files into user-specific folders
     destination_blob_name = f"{user_name}/{truncated_hash}.pdf"
@@ -416,7 +432,7 @@ def generate_document(task_description, md_content, user_name='jayw', bucket_nam
     public_url = save_to_google_storage(bucket_name, local_filename, destination_blob_name)
     logger.debug("Finished generating document")
     return public_url
-
+    
 # In[10]: