Update app.py

app.py

@@ -56,18 +56,41 @@ def process_image(uploaded_file):
         return None, None
 
 
-def generate_pdf(report_text):
-    """Generate a PDF report"""
+def generate_pdf(report_text, logo_b64):
+    """Generate a PDF report with logo"""
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(buffer, pagesize=letter)
     styles = getSampleStyleSheet()
-    story = [Paragraph("<b>Nutrition Analysis Report</b>", styles['Title']), Spacer(1, 12),
-             Paragraph(report_text.replace('\n', '<br/>'), styles['BodyText'])]
+    
+    # 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 this value to change the max width of the logo
+    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 = [
+        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'])
+    ]
+    
     doc.build(story)
     buffer.seek(0)
     return buffer
 
 
+
 def generate_analysis(uploaded_file, client):
     """Generate AI-powered food analysis"""
     base64_image, img_format = process_image(uploaded_file)