Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import torch
 from datetime import datetime
 from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image as PDFImage
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.lib import colors
 from simple_salesforce import Salesforce
@@ -52,7 +52,7 @@ def generate_captions_from_image(image):
     return caption
 
 # Function to save DPR text to a PDF file
-def save_dpr_to_pdf(dpr_text, filename):
+def save_dpr_to_pdf(dpr_text, image_paths, filename):
     try:
         # Create a PDF document
         doc = SimpleDocTemplate(filename, pagesize=letter)
@@ -92,6 +92,14 @@ def save_dpr_to_pdf(dpr_text, filename):
             else:
                 flowables.append(Spacer(1, 12))
         
+        # Add images to the report
+        for img_path in image_paths:
+            try:
+                img = PDFImage(img_path, width=200, height=150)  # Adjust image size if needed
+                flowables.append(img)
+            except Exception as e:
+                flowables.append(Paragraph(f"Error loading image: {str(e)}", body_style))
+        
         # Build the PDF
         doc.build(flowables)
         return f"PDF saved successfully as {filename}", filename
@@ -135,6 +143,7 @@ def upload_file_to_salesforce(file_path, filename, sf_connection, file_type):
 def generate_dpr(files):
     dpr_text = []
     captions = []
+    image_paths = []
     current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     
     # Add header to the DPR
@@ -151,6 +160,9 @@ def generate_dpr(files):
         # Generate DPR section for this image with dynamic caption
         dpr_section = f"\nImage: {file.name}\nDescription: {caption}\n"
         dpr_text.append(dpr_section)
+        
+        # Save image path for embedding in the report
+        image_paths.append(file.name)
     
     # Combine DPR text
     dpr_output = "\n".join(dpr_text)
@@ -159,7 +171,7 @@ def generate_dpr(files):
     pdf_filename = f"DPR_{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.pdf"
     
     # Save DPR text to PDF
-    pdf_result, pdf_filepath = save_dpr_to_pdf(dpr_output, pdf_filename)
+    pdf_result, pdf_filepath = save_dpr_to_pdf(dpr_output, image_paths, pdf_filename)
     
     # Salesforce upload
     salesforce_result = ""
@@ -171,7 +183,6 @@ def generate_dpr(files):
         try:
             # Create Daily_Progress_Reports__c record
             report_description = "; ".join(captions)[:255]  # Concatenate captions, limit to 255 chars
-            # The fix: Assign the report_description to 'Detected_Activities__c' instead of 'Report_Description__c'
             dpr_record = sf.Daily_Progress_Reports__c.create({
                 'Detected_Activities__c': report_description  # Store in Detected_Activities__c field
             })
@@ -199,7 +210,7 @@ def generate_dpr(files):
                 })
                 salesforce_result += f"Updated PDF URL for record ID {dpr_record_id}\n"
             
-            # Upload images to Salesforce
+            # Upload images to Salesforce and create Site_Images__c records
             for file in files:
                 image_filename = os.path.basename(file.name)
                 image_content_document_id, image_upload_result = upload_file_to_salesforce(
@@ -207,6 +218,12 @@ def generate_dpr(files):
                 )
                 if image_content_document_id:
                     image_content_document_ids.append(image_content_document_id)
+                
+                # Create Site_Images__c record
+                site_image_record = sf.Site_Images__c.create({
+                    'Image__c': image_content_document_id,
+                    'Related_Report__c': dpr_record_id  # Link image to DPR record
+                })
                 salesforce_result += image_upload_result + "\n"
                 
                 # Link image to DPR record