Update app.py

app.py

@@ -6,7 +6,7 @@ import io
 from dotenv import load_dotenv
 from groq import Groq
 from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image as ReportLabImage
 from reportlab.lib.styles import getSampleStyleSheet
 
 # ======================
@@ -57,22 +57,29 @@ def process_image(uploaded_file):
 
 
 def generate_pdf(report_text, logo_b64):
-    """Generate a PDF report"""
+    """Generate a PDF report with logo"""
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(buffer, pagesize=letter)
     styles = getSampleStyleSheet()
     
-    # Include logo at the beginning of the report
-    logo_image = Image.open(io.BytesIO(base64.b64decode(logo_b64)))
+    # Decode the base64 logo image
+    logo_data = base64.b64decode(logo_b64)
+    logo_image = Image.open(io.BytesIO(logo_data))
+    
+    # Resize the logo to fit the page width (you can adjust size if necessary)
     logo_width, logo_height = logo_image.size
     logo_aspect = logo_height / logo_width
     max_logo_width = 150  # Adjust as needed
     logo_width = min(logo_width, max_logo_width)
     logo_height = int(logo_width * logo_aspect)
     
+    # Create a ReportLab Image element to add the logo to the PDF
+    logo = ReportLabImage(io.BytesIO(logo_data), width=logo_width, height=logo_height)
+    
+    # Build the PDF content
     story = [
-        Paragraph(f'<img src="data:image/png;base64,{logo_b64}" width="{logo_width}" height="{logo_height}">', styles['Title']),
-        Spacer(1, 12),
+        logo,  # Add the logo at the top of the page
+        Spacer(1, 12),  # Space after the logo
         Paragraph("<b>Nutrition Analysis Report</b>", styles['Title']),
         Spacer(1, 12),
         Paragraph(report_text.replace('\n', '<br/>'), styles['BodyText'])